From 018ab8ae8997c977db08baaa132e5918d6e2bacd Mon Sep 17 00:00:00 2001 From: Evan Langlais Date: Mon, 29 Apr 2019 20:11:40 -0400 Subject: [PATCH] Implementing generator_state online row with TTL --- gateway.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/gateway.go b/gateway.go index fd9ccc2..d938ece 100644 --- a/gateway.go +++ b/gateway.go @@ -98,6 +98,9 @@ func connectionHandler(w http.ResponseWriter, r *http.Request) { DBIDMap.add(g.databaseId, g.serverSn) // Add to map of database IDs to SNs for { // Do forever + if _, err := dbConn.Exec(fmt.Sprintf("INSERT INTO `sd5-state`.generator_state (gen_id, gen_ip, ttl) VALUES (%d, '%s', DATE_ADD(CURRENT_TIME(), INTERVAL 15 MINUTE )) ON DUPLICATE KEY UPDATE gen_id = VALUES(gen_id), gen_ip = VALUES(gen_ip), ttl = VALUES(ttl)", g.databaseId, ws.conn.RemoteAddr().String())); err != nil { + fmt.Printf("Error updating generator state: %v\n", err) + } var packet Packet err := conn.ReadJSON(&packet) // Get our next packet if err != nil { @@ -206,7 +209,6 @@ func (ws *socketConn) handleStatus(updates Packet) { var table = "" - // TODO: Database updating and emailing for statusCode, value := range msg.Updates { if strings.Contains(statusCode, "status.") { // We have a status update @@ -224,8 +226,6 @@ func (ws *socketConn) handleStatus(updates Packet) { query = fmt.Sprintf("INSERT INTO `sd5-state`.%s (gen_id, status_code) VALUES (%d, '%s') ON DUPLICATE KEY UPDATE gen_id = VALUES(gen_id), status_code = VALUES(status_code)", table, generatorId, statusCode) } - fmt.Println(query) - if _, err := dbConn.Exec(query); err != nil { fmt.Printf("Error processing statuses/faults: %v\n", err) }