x

log/mq/nsq.go

@@ -1,9 +1,11 @@
 package mq
 
 import (
+	"context"
 	"encoding/json"
 	"fmt"
 	"git.diulo.com/mogfee/protoc-gen-kit/log"
+	"git.diulo.com/mogfee/protoc-gen-kit/pkg/xgo"
 	"github.com/nsqio/go-nsq"
 	"sync"
 	"sync/atomic"
@@ -11,28 +13,34 @@ import (
 
 type nsqLogger struct {
 	topic  string
-	host  string
+	client *nsq.Producer
 	list   chan []byte
 	wg     sync.WaitGroup
 	stop   atomic.Bool
 }
 
-func (s *nsqLogger) Run(client *nsq.Producer) {
-	s.wg.Add(1)
-	defer s.wg.Done()
+func (s *nsqLogger) Start() error {
+	xgo.Go(func() {
 		for {
 			if s.stop.Load() {
 				break
 			}
 			for v := range s.list {
-			if err := client.Publish(s.topic, v); err != nil {
+				fmt.Println(string(v))
+				func(v []byte) {
+					s.wg.Add(1)
+					defer s.wg.Done()
+					if err := s.client.Publish(s.topic, v); err != nil {
 						fmt.Println(err)
-				continue
 					}
+				}(v)
 			}
 		}
+	})
+	return nil
 }
-func (s *nsqLogger) Shutdown() error {
+
+func (s *nsqLogger) Shutdown(ctx context.Context) error {
 	if s.stop.Load() {
 		return nil
 	}
@@ -40,6 +48,11 @@ func (s *nsqLogger) Shutdown() error {
 	s.wg.Wait()
 	return nil
 }
+
+func (s *nsqLogger) Name() string {
+	return "app_log"
+}
+
 func NewNsqLogger(host string, topic string) log.Logger {
 	client, err := nsq.NewProducer(host, nsq.NewConfig())
 	if err != nil {
@@ -48,13 +61,11 @@ func NewNsqLogger(host string, topic string) log.Logger {
 
 	std := &nsqLogger{
 		topic:  topic,
-		host:  host,
+		client: client,
 		list:   make(chan []byte, 1000),
 		wg:     sync.WaitGroup{},
 	}
-	go std.Run(client)
 	return std
-
 }
 
 func (l *nsqLogger) Log(level log.Level, keyvals ...any) error {

pkg/captcha/captcha_test.go

@@ -8,8 +8,8 @@ import (
 
 func TestGenerateKey(t *testing.T) {
 	client := redis.NewClient(&redis.Options{})
-	store := NewRedisStore("user", client)
-	id, _, err := Generate(store)
+	store := NewRedisStore(client)
+	id, _, err := Generate()
 	fmt.Println(id, err)
 	fmt.Println(store.Verify(id, "aa", false))
 }

pkg/xgo/xgo.go

@@ -0,0 +1,12 @@
+package xgo
+
+import "fmt"
+
+func Go(fun func()) {
+	go func() {
+		if err := recover(); err != nil {
+			fmt.Println(err)
+		}
+		fun()
+	}()
+}