func SetupDB() *sqlx.DB { db := persistence.ConnectDb() db.Exec("DELETE from climbs") db.Exec("DELETE from rocks") return db }
func main() { flag.Parse() db := persistence.ConnectDb() // clear database fmt.Println("Clearing Database") factory.ClearDB(db) // create records r := factory.Rock{} c := factory.Climb{} fmt.Println(fmt.Sprintf("Generating %v records...", *n)) for i := 0; i < *n; i++ { rm := rock.Model{ Latitude: float32(rand.Intn(60-20)+20) + (float32(rand.Intn(10)) * .1), Longitude: -float32(rand.Intn(130-60)+60) + (float32(rand.Intn(10)) * .1), } r.Create(db, &rm) cm := climb.Model{ RockID: rm.ID, Name: fake.Words(), Rating: rand.Intn(14), Description: fake.Sentences(), } c.Create(db, &cm) fmt.Println(fmt.Sprintf("%v generated", i+1)) } fmt.Println("DONE") }
func main() { port := os.Getenv("PORT") if port == "" { log.Fatal("$PORT must be set") } db := persistence.ConnectDb() router := mux.NewRouter() router.HandleFunc("/hello", hello) location.AddRoutes(router, db) climb.AddRoutes(router, db) rock.AddRoutes(router, db) http.ListenAndServe(":"+port, router) }
package sql_test import ( "github.com/jmoiron/sqlx" "github.com/nicholasray/sandstone/lib/persistence" . "github.com/onsi/ginkgo" . "github.com/onsi/gomega" "testing" ) var db *sqlx.DB func TestSql(t *testing.T) { RegisterFailHandler(Fail) RunSpecs(t, "Sql Suite") } var _ = BeforeSuite(func() { db = persistence.ConnectDb() }) var _ = AfterSuite(func() { db.Close() })