You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.7 KiB
88 lines
2.7 KiB
package main
|
|
|
|
import (
|
|
log "github.com/sirupsen/logrus"
|
|
"database/sql"
|
|
_ "github.com/go-sql-driver/mysql"
|
|
"github.com/denzs/wsjtx_dashboards/shared/wsjtx"
|
|
)
|
|
|
|
func handleMysql(result wsjtx.Result) {
|
|
db, dbDown := dbConn()
|
|
if dbDown {
|
|
log.Fatal("cant reach database..")
|
|
}
|
|
defer db.Close()
|
|
|
|
stmt, err := db.Prepare("INSERT INTO wsjtx_all_txt(ts, station, callsign, band, mode, report, dxcc, geohash, continent, cqzone, ituzone, rx) VALUES(?,?,?,?,?,?,?,?,?,?,?,?)")
|
|
defer stmt.Close()
|
|
|
|
_, err = stmt.Exec(result.Timestamp, station, result.Call, result.Band, result.Mode, result.Signal, result.Ent.Entity, result.GeoHash, result.Ent.Continent, result.Ent.CQZone, result.Ent.ITUZone, result.Rx)
|
|
|
|
if err != nil {
|
|
log.WithFields(log.Fields{"err":err.Error()}).Warn("error while executing prepared statement..")
|
|
}
|
|
}
|
|
|
|
|
|
func init_db() {
|
|
var cnt int
|
|
log.Debug("init_db()")
|
|
db, _ := dbConn()
|
|
|
|
row := db.QueryRow("SELECT COUNT(*) FROM information_schema.TABLES WHERE TABLE_NAME LIKE '%" + mysql_table + "%';")
|
|
|
|
switch err := row.Scan(&cnt); err {
|
|
case sql.ErrNoRows:
|
|
log.Error("something went wrong while scanning for existing table structure..")
|
|
panic(err.Error())
|
|
case nil:
|
|
default:
|
|
panic(err)
|
|
}
|
|
|
|
if cnt < 1 {
|
|
qry := "CREATE TABLE IF NOT EXISTS " + mysql_table + " (" +
|
|
"ts timestamp NOT NULL," +
|
|
"station VARCHAR(16) NOT NULL," +
|
|
"callsign VARCHAR(16) NOT NULL," +
|
|
"band VARCHAR(10) NOT NULL," +
|
|
"continent VARCHAR(32) NOT NULL," +
|
|
"mode VARCHAR(16) NOT NULL," +
|
|
"dxcc VARCHAR(128) NOT NULL," +
|
|
"geohash VARCHAR(16) NOT NULL," +
|
|
"report TINYINT NOT NULL," +
|
|
"cqzone INT NOT NULL," +
|
|
"ituzone INT NOT NULL," +
|
|
"rx TINYINT NOT NULL," +
|
|
"PRIMARY KEY UC_" + mysql_table + "(ts, station, callsign)," +
|
|
"INDEX idx_dxcc (dxcc));"
|
|
log.WithFields(log.Fields{"query":qry}).Debug("creating database..")
|
|
_, err := db.Exec(qry)
|
|
if err != nil {
|
|
log.Error("something went wrong while creating table structure..")
|
|
panic(err)
|
|
}
|
|
} else {
|
|
log.Info("found existing table..")
|
|
}
|
|
}
|
|
|
|
|
|
func dbConn() (db *sql.DB, err bool){
|
|
db, er := sql.Open("mysql", mysql_user+":"+mysql_pass+"@tcp("+mysql_host+")/"+mysql_db)
|
|
if er != nil {
|
|
log.Error("db not reachable: %s",err)
|
|
return nil, true
|
|
}
|
|
|
|
pingerr := db.Ping()
|
|
if pingerr != nil {
|
|
log.Error("db not pingable..")
|
|
return nil, true
|
|
}
|
|
|
|
return db, false
|
|
}
|
|
|
|
|
|
|