From b9a826faf821e5072dd75c6f1e465d955a1cf2c5 Mon Sep 17 00:00:00 2001
From: w1ndyb0y <windyboy@163.com>
Date: Mon, 1 Feb 2021 17:03:24 +0800
Subject: [PATCH] pulsar write

---
 utils/socket.go | 5 +----
 utils/sqlite.go | 6 ++++++
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/utils/socket.go b/utils/socket.go
index 791a956..d1753c4 100644
--- a/utils/socket.go
+++ b/utils/socket.go
@@ -47,7 +47,7 @@ func Listen(address string) {
 }
 
 func WriteToClient(data string) error {
-	if Tcp {
+	if client != nil {
 		_, err := client.Write([]byte(data + "\r\n"))
 		if err != nil {
 			Log.Error("error in write data to client socket ", zap.Error(err))
@@ -55,9 +55,6 @@ func WriteToClient(data string) error {
 		}
 		return err
 	}
-	if Pulsar {
-		return PulsarSend(data)
-	}
 	return nil
 }
 
diff --git a/utils/sqlite.go b/utils/sqlite.go
index 50e97dd..88f2473 100644
--- a/utils/sqlite.go
+++ b/utils/sqlite.go
@@ -70,6 +70,12 @@ func InsertTelegram(teleString string) {
 	db := getDb()
 	defer db.Close()
 	var insertSQL string
+	if Pulsar {
+		err := PulsarSend(teleString)
+		if err != nil {
+			Log.Error("error in send to pulsar", zap.Error(err))
+		}
+	}
 	if ClientReady && WriteToClient(teleString) == nil {
 		insertSQL = InsertOld
 	} else {