mirror of
https://gitee.com/XM-GO/PandaX.git
synced 2026-04-23 02:48:34 +08:00
105 lines
2.9 KiB
Go
105 lines
2.9 KiB
Go
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"
|
||
"pandax/pkg/global"
|
||
"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
|
||
}
|
||
go func() {
|
||
for {
|
||
_, message, err := newWebsocket.Conn.ReadMessage()
|
||
if err != nil {
|
||
return
|
||
}
|
||
pxSocket.OnMessage(newWebsocket, string(message))
|
||
}
|
||
}()
|
||
global.Log.Info("Websocket连接成功")
|
||
}
|