Merge pull request #126 from taosdata/fix/TS-4379-3.1

fix loop auto restart when authentication failed

db/connector.go

@@ -4,8 +4,11 @@ import (
 	"context"
 	"database/sql"
 	"fmt"
+	"os"
+	"strings"
 
 	_ "github.com/taosdata/driver-go/v3/taosRestful"
+	"github.com/taosdata/taoskeeper/infrastructure/log"
 )
 
 type Connector struct {
@@ -17,6 +20,8 @@ type Data struct {
 	Data [][]interface{} `json:"data"`
 }
 
+var dbLogger = log.GetLogger("db")
+
 func NewConnector(username, password, host string, port int, usessl bool) (*Connector, error) {
 	var protocol string
 	if usessl {
@@ -48,6 +53,10 @@ func NewConnectorWithDb(username, password, host string, port int, dbname string
 func (c *Connector) Exec(ctx context.Context, sql string) (int64, error) {
 	res, err := c.db.ExecContext(ctx, sql)
 	if err != nil {
+		if strings.Contains(err.Error(), "Authentication failure") {
+			dbLogger.Error("Authentication failure")
+			os.Exit(1)
+		}
 		return 0, err
 	}
 	return res.RowsAffected()
@@ -56,6 +65,10 @@ func (c *Connector) Exec(ctx context.Context, sql string) (int64, error) {
 func (c *Connector) Query(ctx context.Context, sql string) (*Data, error) {
 	rows, err := c.db.QueryContext(ctx, sql)
 	if err != nil {
+		if strings.Contains(err.Error(), "Authentication failure") {
+			dbLogger.Error("Authentication failure")
+			os.Exit(1)
+		}
 		return nil, err
 	}
 	data := &Data{}

taoskeeper.service

@@ -9,7 +9,11 @@ Type=simple
 ExecStart=/usr/bin/taoskeeper
 TimeoutSec=0
 RestartSec=2
+StandardOutput=null
+StandardError=journal
 Restart=always
+StartLimitBurst=3
+StartLimitInterval=60s
 
 [Install]
 WantedBy=multi-user.target