Files
PandaX/apps/visual/api/visual_screen.go
2023-04-15 13:03:57 +08:00

104 lines
2.9 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
package api
// ==========================================================================
// 生成日期2023-04-10 11:26:49 +0800 CST
// 生成路径: apps/visual/api/visual_screen.go
// 生成人panda
// ==========================================================================
import (
"github.com/XM-GO/PandaKit/model"
"github.com/XM-GO/PandaKit/restfulx"
"github.com/emicklei/go-restful/v3"
"github.com/kakuilan/kgo"
"strings"
"pandax/apps/visual/entity"
"pandax/apps/visual/services"
pxSocket "pandax/pkg/websocket"
)
type VisualScreenApi struct {
VisualScreenApp services.VisualScreenModel
}
// GetVisualScreenList Screen列表数据
func (p *VisualScreenApi) GetVisualScreenList(rc *restfulx.ReqCtx) {
data := entity.VisualScreen{}
pageNum := restfulx.QueryInt(rc, "pageNum", 1)
pageSize := restfulx.QueryInt(rc, "pageSize", 10)
data.ScreenName = restfulx.QueryParam(rc, "screenName")
data.Status = restfulx.QueryParam(rc, "status")
data.GroupId = int64(restfulx.QueryInt(rc, "groupId", 0))
list, total := p.VisualScreenApp.FindListPage(pageNum, pageSize, data)
rc.ResData = model.ResultPage{
Total: total,
PageNum: int64(pageNum),
PageSize: int64(pageNum),
Data: list,
}
}
// GetVisualScreen 获取Screen
func (p *VisualScreenApi) GetVisualScreen(rc *restfulx.ReqCtx) {
screenId := restfulx.PathParam(rc, "screenId")
rc.ResData = p.VisualScreenApp.FindOne(screenId)
}
// InsertVisualScreen 添加Screen
func (p *VisualScreenApi) InsertVisualScreen(rc *restfulx.ReqCtx) {
var data entity.VisualScreen
restfulx.BindQuery(rc, &data)
data.UserId = rc.LoginAccount.UserId
data.ScreenId = kgo.KStr.Uniqid("px")
data.Creator = rc.LoginAccount.UserName
p.VisualScreenApp.Insert(data)
}
// UpdateVisualScreen 修改Screen
func (p *VisualScreenApi) UpdateVisualScreen(rc *restfulx.ReqCtx) {
var data entity.VisualScreen
restfulx.BindQuery(rc, &data)
p.VisualScreenApp.Update(data)
}
// DeleteVisualScreen 删除Screen
func (p *VisualScreenApi) DeleteVisualScreen(rc *restfulx.ReqCtx) {
screenId := restfulx.PathParam(rc, "screenId")
screenIds := strings.Split(screenId, ",")
p.VisualScreenApp.Delete(screenIds)
}
// UpdateScreenStatus 修改状态
func (p *VisualScreenApi) UpdateScreenStatus(rc *restfulx.ReqCtx) {
var screen entity.VisualScreen
restfulx.BindQuery(rc, &screen)
p.VisualScreenApp.Update(screen)
}
func (p *VisualScreenApi) ScreenTwin(request *restful.Request, response *restful.Response) {
screenId := request.QueryParameter("screenId")
if screenId == "" {
restfulx.ErrorRes(response, "请传组态Id")
return
}
newWebsocket, err := pxSocket.NewWebsocket(response.ResponseWriter, request.Request, nil)
if err != nil {
restfulx.ErrorRes(response, "创建Websocket失败")
return
}
pxSocket.AddWebSocketByScreenId(screenId, newWebsocket)
go func() {
for {
_, message, err := newWebsocket.Conn.ReadMessage()
if err != nil {
return
}
pxSocket.OnMessage(newWebsocket, string(message))
}
}()
}