SCADATrainSys/Tech/controllers/评价.go

package controllers

import (
	// 根据实际项目结构调整导入路径

	"Tech/models"
	"encoding/json"

	"github.com/beego/beego/v2/client/orm"
	beego "github.com/beego/beego/v2/server/web"
	// 新增导入
)

type EvaController struct {
	beego.Controller
}

// /eva/list 获取课程评价列表
func (c *EvaController) List() {
	type Request struct {
		CourseID int `json:"courseID"`
	}

	var req Request
	requestBody := c.Ctx.Input.RequestBody
	println("Request Body:", string(requestBody))

	if err := json.Unmarshal(requestBody, &req); err != nil {
		c.Ctx.Output.SetStatus(400)
		c.Data["json"] = map[string]interface{}{
			"code": 400,
			"msg":  "请求参数格式错误: " + err.Error(),
		}
		c.ServeJSON()
		return
	}

	if req.CourseID == 0 {
		c.Ctx.Output.SetStatus(400)
		c.Data["json"] = map[string]interface{}{
			"code": 400,
			"msg":  "缺少课程ID参数",
		}
		c.ServeJSON()
		return
	}

	o := orm.NewOrm()
	var evaList []*models.CourseEvaList

	// 根据课程ID查询评价列表
	count, err := o.QueryTable(new(models.CourseEvaList)).
		Filter("CourseID", req.CourseID).
		All(&evaList)

	if err != nil {
		c.Data["json"] = map[string]interface{}{
			"code": 500,
			"msg":  "查询评价列表失败: " + err.Error(),
		}
		c.ServeJSON()
		return
	}

	// 获取所有角色信息用于关联角色名称
	var roleInfos []*models.RoleInfo
	_, err = o.QueryTable(new(models.RoleInfo)).All(&roleInfos)
	if err != nil {
		c.Data["json"] = map[string]interface{}{
			"code": 500,
			"msg":  "查询角色信息失败: " + err.Error(),
		}
		c.ServeJSON()
		return
	}

	// 创建角色ID到角色名称的映射
	roleMap := make(map[int]string)
	for _, role := range roleInfos {
		roleMap[role.RoleID] = role.RoleName
	}

	// 为每个评价项添加角色名称
	type EvaWithRoleName struct {
		*models.CourseEvaList
		RoleName string `json:"roleName"`
	}

	evaWithRoles := make([]EvaWithRoleName, len(evaList))
	for i, eva := range evaList {
		evaWithRoles[i] = EvaWithRoleName{
			CourseEvaList: eva,
			RoleName:      roleMap[eva.EvaTargetRole],
		}
	}

	c.Data["json"] = map[string]interface{}{
		"code":  200,
		"count": count,
		"msg":   "成功",
		"data":  evaWithRoles,
	}
	c.ServeJSON()
}

// /eva/add 添加课程评价
func (c *EvaController) Add() {
	var eva models.CourseEvaList
	requestBody := c.Ctx.Input.RequestBody
	println("Request Body:", string(requestBody))

	if err := json.Unmarshal(requestBody, &eva); err != nil {
		c.Ctx.Output.SetStatus(400)
		c.Data["json"] = map[string]interface{}{
			"code": 400,
			"msg":  "请求参数格式错误: " + err.Error(),
		}
		c.ServeJSON()
		return
	}

	if eva.CourseID == 0 {
		c.Ctx.Output.SetStatus(400)
		c.Data["json"] = map[string]interface{}{
			"code": 400,
			"msg":  "缺少课程ID参数",
		}
		c.ServeJSON()
		return
	}

	o := orm.NewOrm()

	// 如果SeqNo为0，则查询该课程下最大的SeqNo并加1
	if eva.SeqNo == 0 {
		var maxSeqNo int
		err := o.Raw("SELECT COALESCE(MAX(序号), 0) FROM CourseEvaList WHERE 课程ID = ?", eva.CourseID).QueryRow(&maxSeqNo)
		if err != nil {
			c.Data["json"] = map[string]interface{}{
				"code": 500,
				"msg":  "查询序号失败: " + err.Error(),
			}
			c.ServeJSON()
			return
		}
		eva.SeqNo = maxSeqNo + 1
	} else {
		// 如果指定了SeqNo，检查该SeqNo是否已存在
		var count int64
		count, err := o.QueryTable(new(models.CourseEvaList)).
			Filter("CourseID", eva.CourseID).
			Filter("SeqNo", eva.SeqNo).
			Count()

		if err != nil {
			c.Data["json"] = map[string]interface{}{
				"code": 500,
				"msg":  "查询序号失败: " + err.Error(),
			}
			c.ServeJSON()
			return
		}

		// 如果SeqNo已存在，则将该SeqNo及之后的所有序号都加1
		if count > 0 {
			// 使用倒序更新，避免主键冲突
			_, err := o.Raw("UPDATE CourseEvaList SET 序号 = 序号 + 1 WHERE 课程ID = ? AND 序号 >= ? ORDER BY 序号 DESC",
				eva.CourseID, eva.SeqNo).Exec()
			if err != nil {
				c.Data["json"] = map[string]interface{}{
					"code": 500,
					"msg":  "调整序号失败: " + err.Error(),
				}
				c.ServeJSON()
				return
			}
		}
	}

	// 设置评价来源为教员自定义
	eva.EvaSource = 1

	_, err := o.Insert(&eva)
	if err != nil {
		c.Data["json"] = map[string]interface{}{
			"code": 500,
			"msg":  "添加评价失败: " + err.Error(),
		}
		c.ServeJSON()
		return
	}

	c.Data["json"] = map[string]interface{}{
		"code":  200,
		"msg":   "添加评价成功",
		"seqNo": eva.SeqNo,
	}
	c.ServeJSON()
}

// /eva/delete 删除课程评价
func (c *EvaController) Delete() {
	type Request struct {
		CourseID int `json:"courseID"`
		SeqNo    int `json:"seqNo"`
	}

	var req Request
	requestBody := c.Ctx.Input.RequestBody
	println("Request Body:", string(requestBody))

	if err := json.Unmarshal(requestBody, &req); err != nil {
		c.Ctx.Output.SetStatus(400)
		c.Data["json"] = map[string]interface{}{
			"code": 400,
			"msg":  "请求参数格式错误: " + err.Error(),
		}
		c.ServeJSON()
		return
	}

	if req.CourseID == 0 || req.SeqNo == 0 {
		c.Ctx.Output.SetStatus(400)
		c.Data["json"] = map[string]interface{}{
			"code": 400,
			"msg":  "缺少必要参数",
		}
		c.ServeJSON()
		return
	}

	o := orm.NewOrm()

	// 先查询评价记录，判断是否来自流程图
	eva := models.CourseEvaList{CourseID: req.CourseID, SeqNo: req.SeqNo}
	err := o.Read(&eva, "CourseID", "SeqNo")
	if err != nil {
		c.Data["json"] = map[string]interface{}{
			"code": 500,
			"msg":  "查询评价记录失败: " + err.Error(),
		}
		c.ServeJSON()
		return
	}

	// 如果评价来自流程图(EvaSource=0)，则不允许删除
	if eva.EvaSource == 0 {
		c.Data["json"] = map[string]interface{}{
			"code": 403,
			"msg":  "来自流程图的评价不能删除",
		}
		c.ServeJSON()
		return
	}

	// 执行删除操作（仅对教员自定义的评价）
	n, err := o.Delete(&models.CourseEvaList{CourseID: req.CourseID, SeqNo: req.SeqNo})
	if err != nil {
		c.Data["json"] = map[string]interface{}{
			"code": 500,
			"msg":  "删除评价失败: " + err.Error(),
		}
		c.ServeJSON()
		return
	}

	if n == 0 {
		c.Data["json"] = map[string]interface{}{
			"code": 404,
			"msg":  "未找到对应的评价记录",
		}
		c.ServeJSON()
		return
	}

	c.Data["json"] = map[string]interface{}{
		"code": 200,
		"msg":  "删除评价成功",
	}
	c.ServeJSON()
}

// /eva/update 更新课程评价
func (c *EvaController) Update() {
	var eva models.CourseEvaList
	requestBody := c.Ctx.Input.RequestBody
	println("Request Body:", string(requestBody))

	if err := json.Unmarshal(requestBody, &eva); err != nil {
		c.Ctx.Output.SetStatus(400)
		c.Data["json"] = map[string]interface{}{
			"code": 400,
			"msg":  "请求参数格式错误: " + err.Error(),
		}
		c.ServeJSON()
		return
	}

	if eva.CourseID == 0 || eva.SeqNo == 0 {
		c.Ctx.Output.SetStatus(400)
		c.Data["json"] = map[string]interface{}{
			"code": 400,
			"msg":  "缺少必要参数",
		}
		c.ServeJSON()
		return
	}

	o := orm.NewOrm()

	// 先尝试读取记录，判断是否存在
	existingEva := models.CourseEvaList{CourseID: eva.CourseID, SeqNo: eva.SeqNo}
	err := o.Read(&existingEva, "CourseID", "SeqNo")
	if err != nil {
		if err == orm.ErrNoRows {
			c.Data["json"] = map[string]interface{}{
				"code": 404,
				"msg":  "未找到对应的评价记录",
			}
			c.ServeJSON()
			return
		} else {
			c.Data["json"] = map[string]interface{}{
				"code": 500,
				"msg":  "查询评价记录失败: " + err.Error(),
			}
			c.ServeJSON()
			return
		}
	}

	// 使用原生SQL更新记录
	res, err := o.Raw("UPDATE CourseEvaList SET 评价目标角色=?, 评价名称=?, 评价映射ID=?, 评价类型=?, 评价赋分=?, 评价失分=?, 评价来源=? WHERE 课程ID=? AND 序号=?",
		eva.EvaTargetRole, eva.EvaName, eva.EvaMapID, eva.EvaType, eva.EvaScore, eva.EvaDeduction, eva.EvaSource,
		eva.CourseID, eva.SeqNo).Exec()

	if err != nil {
		c.Data["json"] = map[string]interface{}{
			"code": 500,
			"msg":  "更新评价失败: " + err.Error(),
		}
		c.ServeJSON()
		return
	}

	// 获取影响的行数
	num, _ := res.RowsAffected()
	if num == 0 {
		c.Data["json"] = map[string]interface{}{
			"code": 500,
			"msg":  "更新评价失败: 无记录被更新",
		}
		c.ServeJSON()
		return
	}

	c.Data["json"] = map[string]interface{}{
		"code": 200,
		"msg":  "更新评价成功",
		"data": eva,
	}
	c.ServeJSON()
}

// /eva/setscore 设置评价分数
func (c *EvaController) SetScore() {
	type Request struct {
		RunLogID  int     `json:"runLogID"`
		RunOperID int     `json:"runOperID"`
		OperScore float64 `json:"operScore"`
	}

	var req Request
	requestBody := c.Ctx.Input.RequestBody
	println("Request Body:", string(requestBody))

	if err := json.Unmarshal(requestBody, &req); err != nil {
		c.Ctx.Output.SetStatus(400)
		c.Data["json"] = map[string]interface{}{
			"code": 400,
			"msg":  "请求参数格式错误: " + err.Error(),
		}
		c.ServeJSON()
		return
	}

	if req.RunLogID == 0 || req.RunOperID == 0 {
		c.Ctx.Output.SetStatus(400)
		c.Data["json"] = map[string]interface{}{
			"code": 400,
			"msg":  "缺少必要参数",
		}
		c.ServeJSON()
		return
	}

	o := orm.NewOrm()

	// 先检查记录是否存在
	var count int64
	err := o.Raw("SELECT COUNT(*) FROM CourseEvaLog WHERE 运行记录编号 = ? AND 运行操作ID = ?",
		req.RunLogID, req.RunOperID).QueryRow(&count)
	if err != nil {
		c.Data["json"] = map[string]interface{}{
			"code": 500,
			"msg":  "查询评价记录失败: " + err.Error(),
		}
		c.ServeJSON()
		return
	}

	if count == 0 {
		c.Data["json"] = map[string]interface{}{
			"code": 404,
			"msg":  "未找到对应的评价记录",
		}
		c.ServeJSON()
		return
	}

	// 使用原生SQL更新操作得分字段
	res, err := o.Raw("UPDATE CourseEvaLog SET 操作得分 = ? WHERE 运行记录编号 = ? AND 运行操作ID = ?",
		req.OperScore, req.RunLogID, req.RunOperID).Exec()
	if err != nil {
		c.Data["json"] = map[string]interface{}{
			"code": 500,
			"msg":  "更新操作得分失败: " + err.Error(),
		}
		c.ServeJSON()
		return
	}

	// 获取影响的行数
	num, _ := res.RowsAffected()
	if num == 0 {
		c.Data["json"] = map[string]interface{}{
			"code": 500,
			"msg":  "更新操作得分失败: 无记录被更新",
		}
		c.ServeJSON()
		return
	}

	c.Data["json"] = map[string]interface{}{
		"code": 200,
		"msg":  "设置操作得分成功",
	}
	c.ServeJSON()
}

// /eva/record/list 获取课程评价记录列表
func (c *EvaController) EvaRecordList() {
	type Request struct {
		RunLogID int `json:"runLogID"`
	}

	var req Request
	requestBody := c.Ctx.Input.RequestBody
	println("Request Body:", string(requestBody))

	if err := json.Unmarshal(requestBody, &req); err != nil {
		c.Ctx.Output.SetStatus(400)
		c.Data["json"] = map[string]interface{}{
			"code": 400,
			"msg":  "请求参数格式错误: " + err.Error(),
		}
		c.ServeJSON()
		return
	}

	if req.RunLogID == 0 {
		c.Ctx.Output.SetStatus(400)
		c.Data["json"] = map[string]interface{}{
			"code": 400,
			"msg":  "缺少运行记录编号参数",
		}
		c.ServeJSON()
		return
	}

	o := orm.NewOrm()
	var evaLogList []*models.CourseEvaLog

	// 根据运行记录编号查询评价记录列表
	count, err := o.QueryTable(new(models.CourseEvaLog)).
		Filter("RunLogID", req.RunLogID).
		All(&evaLogList)

	if err != nil {
		c.Data["json"] = map[string]interface{}{
			"code": 500,
			"msg":  "查询评价记录列表失败: " + err.Error(),
		}
		c.ServeJSON()
		return
	}

	c.Data["json"] = map[string]interface{}{
		"code":  200,
		"count": count,
		"msg":   "成功",
		"data":  evaLogList,
	}
	c.ServeJSON()
}