完善一部分websocket的代码

utils/auth/confirmation_link_test.go

@@ -25,7 +25,7 @@ func BenchmarkConfirmationLink(b *testing.B) {
 	cl := NewConfirmationLink[Register](key, "http://localhost:9900/api/auth/oauth2/register")
 	for i := 0; i < b.N; i++ {
 
-		uri, _ := cl.Generate(&Register{Id: 39, Password: "21dsadsad", platform: "google", Expired: time.Now()})
+		uri, _ := cl.Generate(&Register{Id: 39, Password: "21dsadsad", platform: "google", Expired: time.Now().UTC()})
 		u, _ := url.Parse(uri)
 		token := u.Query()["token"]
 		cl.Decrypt(token[0])
@@ -125,7 +125,7 @@ func TestConfirmationLink(t *testing.T) {
 	key := "21321321"
 
 	cl := NewConfirmationLink[Register](key, "http://localhost:9900/api/auth/oauth2/register")
-	uri, _ := cl.Generate(&Register{Id: 39, Password: "21dsadsad", platform: "google", Expired: time.Now()})
+	uri, _ := cl.Generate(&Register{Id: 39, Password: "21dsadsad", platform: "google", Expired: time.Now().UTC()})
 	log.Println(uri)
 
 	u, _ := url.Parse(uri)

utils/auth/user_test.go

@@ -11,7 +11,7 @@ import (
 
 // TestGenJwt 测试jwt序列化
 func TestGenJwt(t *testing.T) {
-	now := time.Now().Unix()
+	now := time.Now().UTC().Unix()
 	secret := "fusen123"
 	a, err := GenerateJwtToken(&secret, 3600, now, 123, 1234)
 	if err != nil {
@@ -34,7 +34,7 @@ func TestGenJwt(t *testing.T) {
 }
 
 func TestGenBackendJwt(t *testing.T) {
-	now := time.Now().Unix()
+	now := time.Now().UTC().Unix()
 	secret := "fusen_backend_2023"
 	a, err := GenerateBackendJwtToken(&secret, 3600*24*7, now, 1, 1)
 	if err != nil {

utils/id_generator/unique_number.go

@@ -8,6 +8,6 @@ import (
 )
 
 func GenSnNum() string {
-	a := fmt.Sprintf("%s%.8d", time.Now().Format("20060102150405.000"), rand.Intn(1000000))
+	a := fmt.Sprintf("%s%.8d", time.Now().UTC().Format("20060102150405.000"), rand.Intn(1000000))
 	return strings.ReplaceAll(a, ".", "")
 }

utils/image/image_size.go

@@ -60,7 +60,7 @@ func ThousandFaceImageFormat(req *ThousandFaceImageFormatReq) {
 	req.Cover = ""
 	req.CoverDefault = req.CoverImg
 	if req.Size >= 200 && len(coverSlice) >= 2 && len(coverImgSlice) >= 2 {
-		req.CoverImg = fmt.Sprintf("%s/test/%d/%d_%d.png?%d", constants.DOMAIN_RENDER_IMG_NAME, req.UserId, req.UserId, req.ProductId, time.Now().Unix())
+		req.CoverImg = fmt.Sprintf("%s/test/%d/%d_%d.png?%d", constants.DOMAIN_RENDER_IMG_NAME, req.UserId, req.UserId, req.ProductId, time.Now().UTC().Unix())
 		req.CoverDefault = fmt.Sprintf("%s_%d.%s", coverImgSlice[0], req.Size, coverImgSlice[1])
 	}
 }

utils/websocket_data/base_event.go

@@ -0,0 +1,59 @@
+package websocket_data
+
+import (
+	"time"
+
+	"github.com/google/uuid"
+)
+
+// 和前端交流的事件机制
+type EventType string
+
+const (
+	UserEmailRegister EventType = "E_USER_EMAIL_REGISTER" // 用户注册
+)
+
+// WebsocketEvent 所有事件都必须继承这个结构体
+type WebsocketEvent struct {
+	Type       EventType `json:"event_type"`  // 事件
+	SenderTime time.Time `json:"sender_time"` // 发送的时间, 可能用来统计时间
+	TraceId    string    `json:"trace_id"`    // 链路ID
+	Code       int       `json:"code"`        // 状态码
+	Message    string    `json:"msg"`         // 描述
+	Data       any       `json:"data"`        // 关注的数据
+}
+
+// NewWebsocketEvent 创建一个Websocket事件
+func NewWebsocketEvent(etype EventType) *WebsocketEvent {
+	uid := uuid.NewString()
+	return &WebsocketEvent{
+		Type:       etype,
+		SenderTime: time.Now().UTC(),
+		TraceId:    uid,
+	}
+}
+
+func (event *WebsocketEvent) WithData(data any) *WebsocketEvent {
+	event.Data = data
+	return event
+}
+
+func (event *WebsocketEvent) WithMessgae(msg string) *WebsocketEvent {
+	event.Message = msg
+	return event
+}
+
+func (event *WebsocketEvent) WithCode(code int) *WebsocketEvent {
+	event.Code = code
+	return event
+}
+
+func (event *WebsocketEvent) WithTraceId(traceId string) *WebsocketEvent {
+	event.TraceId = traceId
+	return event
+}
+
+func (event *WebsocketEvent) WithSenderTime(senderTime time.Time) *WebsocketEvent {
+	event.SenderTime = senderTime
+	return event
+}

utils/websocket_data/event.go

@@ -0,0 +1,6 @@
+package websocket_data
+
+// 用户注册成功的事件关注的数据
+type DataEmailRegister struct {
+	JwtToken string `json:"token"` // 注册成功的事件码
+}

utils/websocket_data/event_test.go

@@ -0,0 +1,13 @@
+package websocket_data
+
+import (
+	"encoding/json"
+	"log"
+	"testing"
+)
+
+func TestEvent(t *testing.T) {
+
+	data, _ := json.Marshal(NewWebsocketEvent(UserEmailRegister).WithCode(200))
+	log.Println(string(data))
+}