package service

import (
	"encoding/json"
	"fmt"

	"ems.agt/src/framework/logger"
	"ems.agt/src/framework/vo/result"
	"ems.agt/src/modules/ws/model"
	"ems.agt/src/modules/ws/processor"
)

// 实例化服务层 WSReceiveImpl 结构体
var NewWSReceiveImpl = &WSReceiveImpl{}

// WSReceiveImpl WebSocket消息接收处理 服务层处理
type WSReceiveImpl struct{}

// Receive 接收处理
func (s *WSReceiveImpl) Receive(client *model.WSClient, reqMsg model.WSRequest) {
	if reqMsg.RequestID == "" {
		msg := "message requestId is required"
		logger.Warnf("ws ReceiveMessage UID %s err: %s", client.BindUid, msg)
		msgByte, _ := json.Marshal(result.ErrMsg(msg))
		client.MsgChan <- msgByte
		return
	}

	var resByte []byte
	var err error

	switch reqMsg.Type {
	case "ps":
		resByte, err = processor.GetProcessData(reqMsg.RequestID, reqMsg.Data)
	case "net":
		resByte, err = processor.GetNetConnections(reqMsg.RequestID, reqMsg.Data)
	case "cdr":
		resByte, err = processor.GetCDRConnect(reqMsg.RequestID, reqMsg.Data)
	case "ue":
		resByte, err = processor.GetUEConnect(reqMsg.RequestID, reqMsg.Data)
	case "upf_tf":
		resByte, err = processor.GetUPFTotalFlow(reqMsg.RequestID, reqMsg.Data)
	case "ne_state":
		resByte, err = processor.GetNeState(reqMsg.RequestID, reqMsg.Data)
	default:
		err = fmt.Errorf("message type not supported")
	}

	if err != nil {
		logger.Warnf("ws ReceiveMessage UID %s err: %s", client.BindUid, err.Error())
		msgByte, _ := json.Marshal(result.ErrMsg(err.Error()))
		client.MsgChan <- msgByte
		return
	}
	client.MsgChan <- resByte
}