Implementing generator_state online row with TTL

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)
 		}