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 client net.Conn
var server net.Listener
var ClientReady = false
func Listen(address string) {
var err error
@@ -29,6 +30,7 @@ func Listen(address string) {
}
if client == nil {
client = conn
ClientReady = false
log.Println("client connected ", client.RemoteAddr())
LoadUnprocessed()
} else {
+3 -1
View File
@@ -70,7 +70,7 @@ func InsertTelegram(teleString string) {
db := getDb()
defer db.Close()
var insertSQL string
if IsClientConnected() && WriteToClient(teleString) == nil {
if ClientReady && WriteToClient(teleString) == nil {
insertSQL = InsertOld
} else {
insertSQL = InsertNew
@@ -104,6 +104,8 @@ func LoadUnprocessed() {
}
}
}
ClientReady = true
log.Println("client is ready for new telegram")
}
func getTelegram(db *sql.DB) []telegram {