eta_api/controllers/message.go

package controllers

import (
	"eta_gn/eta_api/models"
	biapprove "eta_gn/eta_api/models/bi_approve"
	"eta_gn/eta_api/models/data_manage/data_manage_permission"
	"eta_gn/eta_api/models/report_approve"
	"fmt"
)

type MessageController struct {
	BaseAuthController
}

// @router /unread/num [get]
func (c *MessageController) UnReadMessageNum() {
	br := new(models.BaseResponse).Init()
	defer func() {
		if br.ErrMsg == "" {
			br.IsSendEmail = false
		}
		c.Data["json"] = br
		c.ServeJSON()
	}()
	sysUser := c.SysUser
	if sysUser == nil {
		br.Msg = "请登录"
		br.ErrMsg = "请登录，SysUser Is Empty"
		br.Ret = 408
		return
	}

	var unReadReportNum, unReadDataPermissionNum, unReadBiNum int

	{
		cond := fmt.Sprintf(` AND %s = ?`, report_approve.ReportApproveMessageCols.ReceiveUserId)
		pars := make([]interface{}, 0)
		pars = append(pars, sysUser.AdminId)

		messageOb := new(report_approve.ReportApproveMessage)

		cond += fmt.Sprintf(` AND %s = ?`, report_approve.ReportApproveMessageCols.IsRead)
		pars = append(pars, 0)
		unreadTotal, e := messageOb.GetCountByCondition(cond, pars)
		if e != nil {
			br.Msg = "获取失败"
			br.ErrMsg = "获取审批消息列表总数失败, Err: " + e.Error()
			return
		}
		unReadReportNum = unreadTotal
	}

	{
		cond := fmt.Sprintf(` AND %s = ?`, data_manage_permission.DataPermissionMessageCols.ReceiveUserId)
		pars := make([]interface{}, 0)
		pars = append(pars, sysUser.AdminId)

		messageOb := new(data_manage_permission.DataPermissionMessage)
		cond += fmt.Sprintf(` AND %s = ?`, data_manage_permission.DataPermissionMessageCols.IsRead)
		pars = append(pars, 0)
		unreadTotal, e := messageOb.GetCountByCondition(cond, pars)
		if e != nil {
			br.Msg = "获取失败"
			br.ErrMsg = "获取资产消息列表总数失败, Err: " + e.Error()
			return
		}
		unReadDataPermissionNum = unreadTotal
	}

	{
		messageOb := new(biapprove.BiApproveMessage)
		pars := make([]interface{}, 0)
		cond := fmt.Sprintf(` AND %s = ? AND %s = ?`, biapprove.BiApproveMessageCols.ReceiveUserId, biapprove.BiApproveMessageCols.IsRead)
		pars = append(pars, sysUser.AdminId, 0)
		unreadTotal, e := messageOb.GetCountByCondition(cond, pars)
		if e != nil {
			br.Msg = "获取失败"
			br.ErrMsg = "获取看板消息列表总数失败, Err: " + e.Error()
			return
		}
		unReadBiNum = unreadTotal
	}

	num := unReadReportNum + unReadDataPermissionNum + unReadBiNum

	br.Data = num
	br.Ret = 200
	br.Success = true
	br.Msg = "获取成功"
}