wsjtx-exporter/cmd/http-exporter/mysql.go

package main

import (
  log "github.com/sirupsen/logrus"
  "database/sql"
  _ "github.com/go-sql-driver/mysql"
  "github.com/denzs/wsjtx-exporter/shared/wsjtx"
  "errors"
  "fmt"
)

func getLast(station string, identifier string) (LastTs, error) {
  var result LastTs

  db, dbDown := dbConn()
  if dbDown {
    log.Fatal("cant reach database..")
  }
  defer db.Close()

  log.Info("starting query...")
  err := db.QueryRow("SELECT UNIX_TIMESTAMP(ts) FROM wsjtx_all_txt WHERE station = ? AND identifier = ? ORDER BY ts DESC LIMIT 1", station, identifier).Scan(&result.Last)
  log.Info("finished query...")
  if err != nil {
    // FIXME 
    if err.Error() == "sql: no rows in result set" {
      log.Infof("uuuh yeah! this is a nice one! ;)")
      return result, errors.New("not found")
    } else {
      log.Infof("well.. shit happend while doing ts lookup: %s", err.Error())
      return result, errors.New(fmt.Sprintf("some other weird shit happened: %s", err.Error()))
    }
  }
  
  log.Infof("last timestamp in db is: %d",result.Last)
  return result, nil
}

func handleMysql(result wsjtx.Result) {
  log.Info("starting mysq stuff..")
  db, dbDown := dbConn()
  if dbDown {
    log.Fatal("cant reach database..")
  }
  defer db.Close()

  stmt, err := db.Prepare("INSERT INTO wsjtx_all_txt(ts, station, identifier, callsign, band, mode, report, dxcc, geohash, continent, cqzone, ituzone, rx) VALUES(FROM_UNIXTIME(?),?,?,?,?,?,?,?,?,?,?,?,?)")
  defer stmt.Close()

  _, err = stmt.Exec(result.Timestamp.Unix(), result.Station, result.Identifier, 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{"warning":err.Error()}).Warn("error while executing prepared statement..")
  } else {
    log.Info("insert done! :)")
  }
}


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 '%" + mysqlTable + "%';")

  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 " + mysqlTable + " (" +
                        "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_" + mysqlTable + "(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 + "?parseTime=true&time_zone=%27UTC%27")
  db, er := sql.Open("mysql", mysqlUser + ":" + mysqlPass + "@tcp(" + mysqlHost + ")/" + mysqlDb)
  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
}