From bac084b1d0b1072195248c6660f7d5fd2f1cd683 Mon Sep 17 00:00:00 2001
From: w1ndyb0y <windyboy@163.com>
Date: Wed, 23 Dec 2020 14:30:09 +0800
Subject: [PATCH] add client ready flag

---
 utils/socket.go | 2 ++
 utils/sqlite.go | 4 +++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/utils/socket.go b/utils/socket.go
index efd96ec..2d01956 100644
--- a/utils/socket.go
+++ b/utils/socket.go
@@ -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 {
diff --git a/utils/sqlite.go b/utils/sqlite.go
index c0c7029..a8d9ef2 100644
--- a/utils/sqlite.go
+++ b/utils/sqlite.go
@@ -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 {