fix run, add connection check

This commit is contained in:
zhiqiang feng
2020-12-22 11:45:25 +08:00
parent 6f00ed6029
commit 13ff3f09e8
3 changed files with 54 additions and 9 deletions
+11 -3
View File
@@ -2,10 +2,11 @@ package utils
import (
"database/sql"
_ "github.com/mattn/go-sqlite3"
"log"
"os"
"time"
_ "github.com/mattn/go-sqlite3"
)
const (
@@ -18,7 +19,8 @@ const (
[tele_text] TEXT NOT NULL
)
`
TableInsert = "insert into telegram (tele_text) values (?)"
InsertNew = "insert into telegram (tele_text) values (?)"
InsertOld = "insert into telegram (tele_text, tele_processed) values (?, 1)"
)
var (
@@ -69,7 +71,13 @@ func InsertTelegram(teleString string) {
isWriting = true
db := getDb()
defer db.Close()
stmt, err := db.Prepare(TableInsert)
var insertSQL string
if IsClientConnected() && WriteToClient(teleString) == nil {
insertSQL = InsertOld
} else {
insertSQL = InsertNew
}
stmt, err := db.Prepare(insertSQL)
if err != nil {
log.Fatal(err)
}