func getTestDB() *bolt.DB {
database, err := bolt.Open(getDBPath(), 0600, &bolt.Options{Timeout: 1 * time.Second})
if err != nil {
log.Fatal(err)
}
db.GenerateDefaultBucket(db.GetBucketName(), database)
return database
}
func main() {
address := flag.String("a", ":8080", "address to listen")
boltLocation := flag.String("d", "bolt.db", "location of the database file")
flag.Parse()
// Open the DB connection
database, err := bolt.Open(*boltLocation, 0600, &bolt.Options{Timeout: 1 * time.Second})
if err != nil {
log.Fatal(err)
}
// Close the DB connection on exit
defer database.Close()
// Generate the default bucket
db.GenerateDefaultBucket(db.GetBucketName(), database)
api := h.APIHandler{FeatureService: s.FeatureService{DB: database}}
// Create and listen for the HTTP server
router := h.NewRouter(api)
log.Fatal(http.ListenAndServe(*address, router))
}