VehicleClassification/AIT/api/record.go

package api

import (
	"AIT/common"
	"AIT/exception/errmsg"
	"AIT/global"
	"AIT/models"
	"AIT/response"
	"AIT/service"
	"encoding/json"
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
	"path/filepath"
)

type RecordService struct {
	service.RecordService
}

func GetRecord() *RecordService {
	return &RecordService{}
}

// SaveRecord
// @Summary 新增测试
// @Description 新增测试
// @Tags 测试记录
// @Param recordFile  formData  file true "文件"
// @Param userName  formData  string true "用户名"
// @Param fileType  formData  string true "文件类型"
// @Accept  json
// @Produce  json
// @Success 200 {string} string	"ok"
// @Router /v1/record/create [post]
// @Security ApiKeyAuth
func (f *RecordService) SaveRecord(c *gin.Context) {

	file, err := c.FormFile("file")
	if err != nil {
		fmt.Printf(errmsg.FileUploadError+"%s", err.Error())
		response.Failed(errmsg.FileUploadError, c)
	}
	message := make(map[string]string)
	userName := c.PostForm("userName")
	fileType := c.PostForm("fileType")
	message["msg_type"] = fileType
	message["Sender"] = "server"
	message["Recipient"] = "client"
	//文件格式限制
	fileName := file.Filename
	fileExtendName := common.CheckFile(fileName)
	if len(fileExtendName) <= 0 {
		response.Failed(errmsg.FileExtendNameError, c)
		return
	}
	fmt.Printf("fileName====%s\n", fileName)

	fileName = common.GenerateUUID() + "." + fileExtendName
	fileLocation := global.Config.Upload
	//保存文件
	err = c.SaveUploadedFile(file, fileLocation.SavePath+fileName)
	if err != nil {
		return
	}

	//获取文件绝对路径
	fileAbsolutePath, err := filepath.Abs(fileLocation.SavePath + fileName)
	if err != nil {
		return
	}

	message["content"] = fileAbsolutePath
	fileURL := fileLocation.AccessUrl + fileName
	//添加记录
	var record models.Record
	record.FilePath = fileAbsolutePath
	record.CreateTime = common.NowTime()
	record.UserName = userName
	record.FileType = fileType
	record.Id = common.GenerateUUID()
	record.FileUrl = fileURL
	if count := f.CreateRecord(record); count <= 0 {
		response.Failed(errmsg.RecordSaveError, c)
	} else {
		response.Success(errmsg.SaveSuccess, record.Id, c)
	}
	message["id"] = record.Id
	//invoke AI

	ws, _, err := websocket.DefaultDialer.Dial(global.Config.Websocket.WSUrl, nil)
	if err != nil {
		fmt.Println("Websocket client init error" + err.Error())
		return
	}
	defer func(ws *websocket.Conn) {
		err := ws.Close()
		if err != nil {
			fmt.Printf("socket关闭异常：%s", err.Error())
			response.Failed(errmsg.SocketCloseError, c)
		}
	}(ws)
	//发送消息
	bytes, err := json.Marshal(message)
	if err != nil {
		fmt.Printf("json异常：%s", err.Error())
		response.Failed(errmsg.JSONParseError, c)
		return
	}
	err = ws.WriteMessage(websocket.TextMessage, bytes)
	if err != nil {
		fmt.Printf("json异常：%s", err.Error())
		response.Failed(errmsg.SendMessageError, c)
		return
	}

	return
}

// RecordList
// @Summary 记录列表查询
// @Description 记录列表查询
// @Tags 测试记录
// @Param r body string true "role：角色；userName：用户名;pageNum:页数；pageSize：每页记录数"
// @Accept json
// @Produce json
// @Success 200 {string} string "ok"
// @Router /v1/record/find [get]
func (f *RecordService) RecordList(c *gin.Context) {
	var param models.RecordListParam
	if err := c.ShouldBindJSON(&param); err != nil {
		fmt.Printf(errmsg.ParamInvalid+"%s", err.Error())
		response.Failed(errmsg.ParamInvalid, c)
		return
	}
	recordList, rows := f.GetRecordList(param)
	response.SuccessPage(errmsg.FindSuccess, recordList, rows, c)

}