ref: change database path

SharkEzz · Dec 31, 2021 · 86ff5e9 · 86ff5e9
1 parent ac34966
commit 86ff5e9
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,2 +1,2 @@
-database/local.db
+local.db
 sattrack
diff --git a/main.go b/main.go
@@ -27,14 +27,16 @@ func main() {
 	log.Println("Starting", appName, version)
 	flag.Parse()
 
-	db := database.Init("database/local.db")
+	db := database.Init("./local.db")
 
 	if *shouldUpdate {
 		services.UpdateDatabase(*tleListURL, db)
 	} else {
 		log.Println("Not updating TLE database (flag -update not given)")
 	}
 
+	log.Println("Loaded", services.GetTLECount(db), "TLEs")
+
 	validator := validator.New()
 
 	app := fiber.New(fiber.Config{

diff --git a/services/TLEService.go b/services/TLEService.go
@@ -69,3 +69,11 @@ func GetTLEFromDatabase(catNbr int, db *gorm.DB) (*sgp4.TLE, error) {
 
 	return tle, nil
 }
+
+func GetTLECount(db *gorm.DB) int64 {
+	var count int64
+
+	db.Model(&models.TLE{}).Count(&count)
+
+	return count
+}