From 133c80088c6dd782020c7f9821f43875c2ad731f Mon Sep 17 00:00:00 2001
From: w1ndyb0y <windyboy@163.com>
Date: Mon, 1 Feb 2021 17:29:01 +0800
Subject: [PATCH] check tcp when insert telegram to db

---
 utils/sqlite.go | 40 +++++++++++++++++++++-------------------
 1 file changed, 21 insertions(+), 19 deletions(-)

diff --git a/utils/sqlite.go b/utils/sqlite.go
index 50e97dd..9eed6bd 100644
--- a/utils/sqlite.go
+++ b/utils/sqlite.go
@@ -66,26 +66,28 @@ func InitDb(file string, init bool) error {
 }
 
 func InsertTelegram(teleString string) {
-	getWriteLock()
-	db := getDb()
-	defer db.Close()
-	var insertSQL string
-	if ClientReady && WriteToClient(teleString) == nil {
-		insertSQL = InsertOld
-	} else {
-		insertSQL = InsertNew
-	}
-	stmt, _ := db.Prepare(insertSQL)
+	if Tcp {
+		getWriteLock()
+		db := getDb()
+		defer db.Close()
+		var insertSQL string
+		if ClientReady && WriteToClient(teleString) == nil {
+			insertSQL = InsertOld
+		} else {
+			insertSQL = InsertNew
+		}
+		stmt, _ := db.Prepare(insertSQL)
 
-	defer stmt.Close()
-	result, err := stmt.Exec(teleString)
-	id, _ := result.LastInsertId()
-	checkDbErr(err, "error in insert telegram ")
-	isWriting = false
-	if insertSQL == InsertOld {
-		Log.Info("telegram processed ", zap.Int64("id", id))
-	} else {
-		Log.Info("telegram saved ", zap.Int64("id", id))
+		defer stmt.Close()
+		result, err := stmt.Exec(teleString)
+		id, _ := result.LastInsertId()
+		checkDbErr(err, "error in insert telegram ")
+		isWriting = false
+		if insertSQL == InsertOld {
+			Log.Info("telegram processed ", zap.Int64("id", id))
+		} else {
+			Log.Info("telegram saved ", zap.Int64("id", id))
+		}
 	}
 }