PandaX/apps/visual/api/visual_screen.go

package api

// ==========================================================================
// 生成日期：2023-04-10 11:26:49 +0800 CST
// 生成路径: apps/visual/api/visual_screen.go
// 生成人：panda
// ==========================================================================
import (
	"encoding/json"
	"github.com/XM-GO/PandaKit/model"
	"github.com/XM-GO/PandaKit/restfulx"
	"github.com/emicklei/go-restful/v3"
	"github.com/kakuilan/kgo"
	"log"
	"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) ScreenTwinData(rc *restfulx.ReqCtx) {
	twin := `[{"key":"1001","name":"监测站001","attrs":[{"key":"wd","type":"int64","name":"温度"},{"key":"sd","type":"float64","name":"湿度"}]},{"key":"2001","name":"控制器001","attrs":[{"key":"q","type":"struct","name":"灯光强度1"},{"key":"open","type":"bool","name":"灯光开关"}]}]`

	data := make([]map[string]interface{}, 0)
	json.Unmarshal([]byte(twin), &data)
	rc.ResData = data
}

func (p *VisualScreenApi) ScreenTwinData1(rc *restfulx.ReqCtx) {
	products := `[{"classId": "p313123","name": "微型环境监测站"},{"classId": "p313124","name": "温湿度传感器"}]`
	twin := `[{"twinId":"1001","name":"监测站001","attrs":[{"key":"wd","type":"int64","name":"温度"},{"key":"sd","type":"float64","name":"湿度"}]},{"twinId":"2001","name":"控制器001","attrs":[{"key":"q","type":"struct","name":"灯光强度1"},{"key":"open","type":"bool","name":"灯光开关"}]}]`
	pageNum := restfulx.QueryInt(rc, "pageNum", 1)
	pageSize := restfulx.QueryInt(rc, "pageSize", 10)
	classId := restfulx.QueryParam(rc, "classId")
	if classId == "" {
		data := make([]map[string]interface{}, 0)
		json.Unmarshal([]byte(products), &data)
		rc.ResData = data
	} else {
		//todo 分页查询孪生体
		log.Println(pageNum, pageSize)
		data := make([]map[string]interface{}, 0)
		json.Unmarshal([]byte(twin), &data)
		rc.ResData = data
	}
}

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连接成功")
}