check tcp when insert telegram to db

This commit is contained in:
w1ndyb0y
2021-02-01 17:29:01 +08:00
parent fe1a8169e6
commit 133c80088c
+21 -19
View File
@@ -66,26 +66,28 @@ func InitDb(file string, init bool) error {
} }
func InsertTelegram(teleString string) { func InsertTelegram(teleString string) {
getWriteLock() if Tcp {
db := getDb() getWriteLock()
defer db.Close() db := getDb()
var insertSQL string defer db.Close()
if ClientReady && WriteToClient(teleString) == nil { var insertSQL string
insertSQL = InsertOld if ClientReady && WriteToClient(teleString) == nil {
} else { insertSQL = InsertOld
insertSQL = InsertNew } else {
} insertSQL = InsertNew
stmt, _ := db.Prepare(insertSQL) }
stmt, _ := db.Prepare(insertSQL)
defer stmt.Close() defer stmt.Close()
result, err := stmt.Exec(teleString) result, err := stmt.Exec(teleString)
id, _ := result.LastInsertId() id, _ := result.LastInsertId()
checkDbErr(err, "error in insert telegram ") checkDbErr(err, "error in insert telegram ")
isWriting = false isWriting = false
if insertSQL == InsertOld { if insertSQL == InsertOld {
Log.Info("telegram processed ", zap.Int64("id", id)) Log.Info("telegram processed ", zap.Int64("id", id))
} else { } else {
Log.Info("telegram saved ", zap.Int64("id", id)) Log.Info("telegram saved ", zap.Int64("id", id))
}
} }
} }