122 lines
3.1 KiB
Go
122 lines
3.1 KiB
Go
package tools
|
|
|
|
import (
|
|
"ai_scheduler/internal/config"
|
|
"ai_scheduler/internal/entitys"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"gitea.cdlsxd.cn/self-tools/l_request"
|
|
"github.com/gofiber/websocket/v2"
|
|
)
|
|
|
|
type ZltxOrderLogTool struct {
|
|
config config.ToolConfig
|
|
}
|
|
|
|
func (t *ZltxOrderLogTool) Name() string {
|
|
return "zltxOrderDirectLog"
|
|
}
|
|
|
|
func (t *ZltxOrderLogTool) Description() string {
|
|
return "查询订单日志"
|
|
}
|
|
|
|
func (t *ZltxOrderLogTool) Definition() entitys.ToolDefinition {
|
|
return entitys.ToolDefinition{
|
|
Type: "function",
|
|
Function: entitys.FunctionDef{
|
|
Name: t.Name(),
|
|
Description: t.Description(),
|
|
Parameters: map[string]interface{}{
|
|
"type": "object",
|
|
"properties": map[string]interface{}{
|
|
"order_number": map[string]interface{}{
|
|
"type": "string",
|
|
"description": "订单编号",
|
|
},
|
|
"serial_number": map[string]interface{}{
|
|
"type": "string",
|
|
"description": "流水号",
|
|
},
|
|
},
|
|
"required": []string{"order_number", "serial_number"},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
// ZltxOrderDetailRequest 直连天下订单详情请求参数
|
|
type ZltxOrderLogRequest struct {
|
|
OrderNumber string `json:"order_number"`
|
|
SerialNumber string `json:"serial_number"`
|
|
}
|
|
|
|
// ZltxOrderDetailResponse 直连天下订单详情响应
|
|
type ZltxOrderDirectLogResponse struct {
|
|
Code int `json:"code"`
|
|
Error string `json:"error"`
|
|
Data []ZltxOrderDirectLogData `json:"data"`
|
|
}
|
|
|
|
// ZltxOrderLogData 直连天下订单详情数据
|
|
type ZltxOrderDirectLogData struct {
|
|
Datetime string `json:"datetime"`
|
|
ServerID string `json:"serverId"`
|
|
Mes string `json:"mes"`
|
|
Data map[string]interface{} `json:"data"`
|
|
}
|
|
|
|
func (t *ZltxOrderLogTool) Execute(channel chan entitys.Response, c *websocket.Conn, args json.RawMessage) error {
|
|
var req ZltxOrderLogRequest
|
|
if err := json.Unmarshal(args, &req); err != nil {
|
|
return fmt.Errorf("invalid zltxOrderLog request: %w", err)
|
|
}
|
|
if req.OrderNumber == "" || req.SerialNumber == "" {
|
|
return fmt.Errorf("orderNumber and serialNumber is required")
|
|
}
|
|
return t.getZltxOrderLog(channel, c, req.OrderNumber, req.SerialNumber)
|
|
}
|
|
|
|
func (t *ZltxOrderLogTool) getZltxOrderLog(channel chan entitys.Response, c *websocket.Conn, orderNumber, serialNumber string) (err error) {
|
|
//查询订单详情
|
|
var auth string
|
|
if c != nil {
|
|
auth = c.Query("x-authorization", "")
|
|
}
|
|
if len(auth) == 0 {
|
|
auth = t.config.APIKey
|
|
}
|
|
url := fmt.Sprintf("%s%s/%s", t.config.BaseURL, orderNumber, serialNumber)
|
|
req := l_request.Request{
|
|
Url: url,
|
|
Headers: map[string]string{
|
|
"Authorization": fmt.Sprintf("Bearer %s", auth),
|
|
},
|
|
Method: "GET",
|
|
}
|
|
res, err := req.Send()
|
|
var resData ZltxOrderDirectLogResponse
|
|
if err != nil {
|
|
return
|
|
}
|
|
if resData.Code != 200 {
|
|
return fmt.Errorf("订单查询失败：%s", resData.Error)
|
|
}
|
|
if err = json.Unmarshal(res.Content, &resData); err != nil {
|
|
return
|
|
}
|
|
channel <- entitys.Response{
|
|
Index: t.Name(),
|
|
Content: res.Text,
|
|
Type: entitys.ResponseJson,
|
|
}
|
|
return
|
|
}
|
|
|
|
func NewZltxOrderLogTool(config config.ToolConfig) *ZltxOrderLogTool {
|
|
return &ZltxOrderLogTool{
|
|
config: config,
|
|
}
|
|
}
|