eta_task/models/data_manage/edb_inspection/edb_inspection_message.go

package edb_inspection

import (
	"eta/eta_task/global"
	"eta/eta_task/utils"
	"time"
)

// EdbInspectionMessage
// @Description: 巡检消息表
type EdbInspectionMessage struct {
	MessageId          int64     `gorm:"column:message_id;primaryKey;autoIncrement"`
	InspectionRecordId int64     `gorm:"column:inspection_record_id"`
	AdminId            int64     `gorm:"column:admin_id"`
	Message            string    `gorm:"column:message"`
	IsRead             int8      `gorm:"column:is_read"`
	Source             int8      `gorm:"column:source"`
	TerminalCode       string    `gorm:"column:terminal_code"`
	InspectionTime     time.Time `gorm:"column:inspection_time"`
	CreateTime         time.Time `gorm:"column:create_time"`
	ModifyTime         time.Time `gorm:"column:modify_time"`
}

// Add
// @Description: 添加巡检消息
// @receiver m
// @return err error
func (m *EdbInspectionMessage) Add() (err error) {
	m.CreateTime = time.Now()
	m.ModifyTime = time.Now()
	err = global.DbMap[utils.DbNameIndex].Create(m).Error
	return
}

// Update
// @Description: 更新巡检消息
// @receiver m
// @param cols []string
// @return err error
func (m *EdbInspectionMessage) Update(cols []string) (err error) {
	m.ModifyTime = time.Now()
	err = global.DbMap[utils.DbNameIndex].Select(cols).Updates(m).Error
	return
}

// GetById
// @Description: 根据ID获取巡检消息
// @param messageId int64
// @return item *EdbInspectionMessage
// @return err error
func GetMessageById(messageId int64) (item *EdbInspectionMessage, err error) {
	sql := `SELECT * FROM edb_inspection_message WHERE message_id = ?`
	err = global.DbMap[utils.DbNameIndex].Raw(sql, messageId).First(&item).Error
	return
}

// GetListByInspectionRecordId
// @Description: 根据巡检记录ID获取巡检消息列表
// @param inspectionRecordId int64
// @return list []*EdbInspectionMessage
// @return err error
func GetMessageListByInspectionRecordId(inspectionRecordId int64) (list []*EdbInspectionMessage, err error) {
	sql := `SELECT * FROM edb_inspection_message WHERE inspection_record_id = ? ORDER BY create_time DESC`
	err = global.DbMap[utils.DbNameIndex].Raw(sql, inspectionRecordId).Find(&list).Error
	return
}

// GetListBySendStatus
// @Description: 根据发送状态获取巡检消息列表
// @param sendStatus int8
// @return list []*EdbInspectionMessage
// @return err error
func GetMessageListBySendStatus(sendStatus int8) (list []*EdbInspectionMessage, err error) {
	sql := `SELECT * FROM edb_inspection_message WHERE send_status = ? ORDER BY create_time DESC`
	err = global.DbMap[utils.DbNameIndex].Raw(sql, sendStatus).Find(&list).Error
	return
}

// UpdateSendStatus
// @Description: 更新消息发送状态
// @receiver m
// @param sendStatus int8
// @return err error
func (m *EdbInspectionMessage) UpdateSendStatus(sendStatus int8) (err error) {
	sql := `UPDATE edb_inspection_message SET send_status = ?, modify_time = ? WHERE message_id = ?`
	err = global.DbMap[utils.DbNameIndex].Exec(sql, sendStatus, time.Now(), m.MessageId).Error
	return
}