init: initial commit

main/db.go

@@ -0,0 +1,55 @@
+package main
+
+import (
+	"database/sql"
+	"fmt"
+
+	"git.dvdrw.dev/nsmarter/scraper/scraper"
+	_ "github.com/mattn/go-sqlite3"
+)
+
+func readDb() ([]scraper.Station, error) {
+	db, err := sql.Open("sqlite3", stationDbPath)
+	if err != nil {
+		return nil, err
+	}
+	defer db.Close()
+
+	queryString := "SELECT stationId FROM stations"
+	if innerCityOnly {
+		queryString += " WHERE cityId=72 "
+	}
+
+	if(limitQuerySize > 0) {
+		queryString += fmt.Sprintf(" LIMIT %d ", limitQuerySize)
+	}
+
+	queryString += ";"
+
+	rows, err := db.Query(queryString)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var stations []scraper.Station = make([]scraper.Station, 0, 1000)
+
+	for i := 0; rows.Next(); i++ {
+		station := scraper.Station{}
+		err = rows.Scan(&station.Id)
+
+		if err != nil {
+			return nil, err
+		}
+
+		stations = append(stations, station)
+		Log.Printf("Imported station with ID=%v\n", station.Id)
+	}
+
+	err = rows.Err()
+	if err != nil {
+		return nil, err
+	}
+
+	return stations, nil
+}