123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869 |
- package main
- import (
- "database/sql"
- "flag"
- "log"
- "net/http"
- "sync"
- kyoketsu "git.aetherial.dev/aeth/kyoketsu/pkg"
- )
- const dbfile = "sqlite.db"
- func main() {
- port := flag.Int("port", 8080, "Select the port to run the server on")
- debug := flag.Bool("debug", false, "Pass this to start the pprof server")
- flag.Parse()
- db, err := sql.Open("sqlite3", dbfile)
- if err != nil {
- log.Fatal(err)
- }
- if *debug {
- go func() {
- log.Println(http.ListenAndServe("localhost:6060", nil))
- }()
- }
- hostsRepo := kyoketsu.NewSQLiteRepo(db)
- var wg sync.WaitGroup
- wg.Add(1)
- go kyoketsu.RunHttpServer(*port, hostsRepo, kyoketsu.RetrieveScanDirectives())
- if err = hostsRepo.Migrate(); err != nil {
- log.Fatal(err)
- }
- log.Println("SUCCESS ::: SQLite database initiated, and open for writing.")
- wg.Wait()
- }
|