Merge pull request #132 from taosdata/fix/TD-30968

add flush log
taosdata · Jul 16, 2024 · 12e4b2a · 12e4b2a
2 parents 4ac3a44 + f4a0bcb
commit 12e4b2a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/db/connector.go b/db/connector.go
@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"os"
 	"strings"
+	"time"
 
 	_ "github.com/taosdata/driver-go/v3/taosRestful"
 	"github.com/taosdata/taoskeeper/infrastructure/log"
@@ -55,6 +56,9 @@ func (c *Connector) Exec(ctx context.Context, sql string) (int64, error) {
 	if err != nil {
 		if strings.Contains(err.Error(), "Authentication failure") {
 			dbLogger.Error("Authentication failure")
+			ctxLog, cancelLog := context.WithTimeout(context.Background(), 3*time.Second)
+			defer cancelLog()
+			log.Close(ctxLog)
 			os.Exit(1)
 		}
 		return 0, err
@@ -67,6 +71,9 @@ func (c *Connector) Query(ctx context.Context, sql string) (*Data, error) {
 	if err != nil {
 		if strings.Contains(err.Error(), "Authentication failure") {
 			dbLogger.Error("Authentication failure")
+			ctxLog, cancelLog := context.WithTimeout(context.Background(), 3*time.Second)
+			defer cancelLog()
+			log.Close(ctxLog)
 			os.Exit(1)
 		}
 		return nil, err

diff --git a/system/program.go b/system/program.go
@@ -122,5 +122,11 @@ func (p *program) Stop(s service.Service) error {
 	}
 
 	logger.Println("Server exiting")
+
+	ctxLog, cancelLog := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancelLog()
+	logger.Println("Flushing Log")
+	log.Close(ctxLog)
+
 	return nil
 }