add client ready flag

This commit is contained in:
w1ndyb0y
2020-12-23 14:30:09 +08:00
parent 7a21e09048
commit bac084b1d0
2 changed files with 5 additions and 1 deletions
+2
View File
@@ -12,6 +12,7 @@ const ServerType = "tcp"
var ServerRunning = false var ServerRunning = false
var client net.Conn var client net.Conn
var server net.Listener var server net.Listener
var ClientReady = false
func Listen(address string) { func Listen(address string) {
var err error var err error
@@ -29,6 +30,7 @@ func Listen(address string) {
} }
if client == nil { if client == nil {
client = conn client = conn
ClientReady = false
log.Println("client connected ", client.RemoteAddr()) log.Println("client connected ", client.RemoteAddr())
LoadUnprocessed() LoadUnprocessed()
} else { } else {
+3 -1
View File
@@ -70,7 +70,7 @@ func InsertTelegram(teleString string) {
db := getDb() db := getDb()
defer db.Close() defer db.Close()
var insertSQL string var insertSQL string
if IsClientConnected() && WriteToClient(teleString) == nil { if ClientReady && WriteToClient(teleString) == nil {
insertSQL = InsertOld insertSQL = InsertOld
} else { } else {
insertSQL = InsertNew insertSQL = InsertNew
@@ -104,6 +104,8 @@ func LoadUnprocessed() {
} }
} }
} }
ClientReady = true
log.Println("client is ready for new telegram")
} }
func getTelegram(db *sql.DB) []telegram { func getTelegram(db *sql.DB) []telegram {