合并

server/websocket/internal/logic/datatransferlogic.go

@@ -8,14 +8,16 @@ import (
 	"fusenapi/utils/auth"
 	"fusenapi/utils/id_generator"
 	"fusenapi/utils/websocket_data"
-	"github.com/gorilla/websocket"
 	"net/http"
 	"sync"
 	"time"
 
+	"github.com/gorilla/websocket"
+
 	"context"
 
 	"fusenapi/server/websocket/internal/svc"
+
 	"github.com/zeromicro/go-zero/core/logx"
 )
 

server/websocket/internal/logic/mq_consumer.go

@@ -5,6 +5,7 @@ import (
 	"encoding/json"
 	"fusenapi/constants"
 	"fusenapi/utils/websocket_data"
+
 	"github.com/zeromicro/go-zero/core/logx"
 )
 

server/websocket/internal/logic/ws_render_image_logic.go

@@ -5,6 +5,7 @@ import (
 	"fusenapi/constants"
 	"fusenapi/utils/hash"
 	"fusenapi/utils/websocket_data"
+
 	"github.com/zeromicro/go-zero/core/logx"
 )
 

server/websocket/internal/types/types.go

@@ -5,6 +5,23 @@ import (
 	"fusenapi/utils/basic"
 )
 
+type RequestToUnity struct {
+}
+
+type RequestReadImages struct {
+}
+
+type RenderNotifyReq struct {
+	Sign string     `json:"sign"`
+	Time int64      `json:"time"`
+	Info NotifyInfo `json:"info"`
+}
+
+type NotifyInfo struct {
+	TaskId string `json:"task_id"` //任务id
+	Image  string `json:"image"`
+}
+
 type Request struct {
 }
 
@@ -72,4 +89,4 @@ func (resp *Response) SetStatusWithMessage(sr *basic.StatusResponse, msg string,
 		newResp.Data = data[0]
 	}
 	return newResp
-}
+}