Make fortoprt handling more robust

fortoprt/fetch.go

@@ -8,6 +8,7 @@ import (
 
 	"git.fuyu.moe/5GPowerQuality/parser/data"
 	"git.fuyu.moe/5GPowerQuality/parser/shared"
+	"git.fuyu.moe/Fuyu/flog"
 	"golang.org/x/net/websocket"
 )
 
@@ -43,6 +44,18 @@ func FetchData() {
 	requestData(conn, 2)
 
 	for {
+		readData(conn)
+	}
+}
+
+func readData(conn *websocket.Conn) {
+	defer func() {
+		v := recover()
+		if v != nil {
+			flog.Critical(v)
+		}
+	}()
+
 	b, n := make([]byte, 16384), 0
 
 	msg := message{}
@@ -62,7 +75,6 @@ func FetchData() {
 		break
 	}
 
-		// sets := data.Sets{}
 	mID := data.GetMeterID(data.SourceFortop, host+`-`+strconv.Itoa(msg.Data["U1"].DeviceID))
 
 	sets := data.Sets{}
@@ -85,7 +97,6 @@ func FetchData() {
 	}
 
 	data.InsertSets(sets)
-	}
 }
 
 func requestData(conn *websocket.Conn, meter int) {