30 lines
673 B
Go
30 lines
673 B
Go
|
// Code scaffolded by goctl. Safe to edit.
|
||
|
|
// goctl 1.10.1
|
||
|
|
|
||
|
|
package handler
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/zeromicro/go-zero/rest/httpx"
|
||
|
|
"img_infinite_canvas/internal/logic"
|
||
|
|
"img_infinite_canvas/internal/svc"
|
||
|
|
"img_infinite_canvas/internal/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
func agentWebsocketHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
|
||
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
var req types.ProjectIdRequest
|
||
|
|
if err := httpx.Parse(r, &req); err != nil {
|
||
|
|
httpx.ErrorCtx(r.Context(), w, err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
l := logic.NewAgentWebsocketLogic(r.Context(), svcCtx)
|
||
|
|
err := l.Stream(w, r, &req)
|
||
|
|
if err != nil {
|
||
|
|
httpx.ErrorCtx(r.Context(), w, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|