Make server out of SS script
This commit is contained in:
@@ -6,6 +6,7 @@ import (
|
||||
"encoding/csv"
|
||||
"flag"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
|
||||
_ "github.com/go-sql-driver/mysql"
|
||||
@@ -19,13 +20,19 @@ var (
|
||||
)
|
||||
|
||||
func main() {
|
||||
err := server()
|
||||
assertNoErr(err)
|
||||
|
||||
flag.Parse()
|
||||
file, err := os.Open(*csvPath)
|
||||
assertNoErr(err)
|
||||
records, err := csv.NewReader(file).ReadAll()
|
||||
assertNoErr(err)
|
||||
|
||||
db := openDB(*mysqlConnString)
|
||||
db, err := openDB(*mysqlConnString)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
bobDB := bob.NewDB(db)
|
||||
|
||||
err = os.RemoveAll("aips-report.csv")
|
||||
@@ -37,7 +44,7 @@ func main() {
|
||||
reportData := make([][]string, len(records))
|
||||
for idx, record := range records {
|
||||
q := models.LocationsPackages.Query()
|
||||
like := fmt.Sprintf("%%%s%%", record[0])
|
||||
like := fmt.Sprintf("%%%s%%", record)
|
||||
q.Apply(models.SelectWhere.LocationsPackages.CurrentPath.Like(like))
|
||||
res, err := q.All(context.Background(), bobDB)
|
||||
assertNoErr(err)
|
||||
@@ -62,10 +69,10 @@ func assertNoErr(err error) {
|
||||
}
|
||||
}
|
||||
|
||||
func openDB(connStr string) *sql.DB {
|
||||
func openDB(connStr string) (*sql.DB, error) {
|
||||
db, err := sql.Open("mysql", connStr)
|
||||
assertNoErr(err)
|
||||
err = db.Ping()
|
||||
assertNoErr(err)
|
||||
return db
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return db, db.Ping()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user