99 lines
3.6 KiB
Go
99 lines
3.6 KiB
Go
package ticket
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/redis/go-redis/v9"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var cacheTicketDetailPrefix = "cache:ticket:detail:"
|
|
|
|
type Details struct {
|
|
Id int64 `gorm:"primaryKey"`
|
|
Title string `gorm:"type:varchar(255);not null;default:'';comment:Title"`
|
|
Description string `gorm:"type:text;comment:Description"`
|
|
UserId int64 `gorm:"type:bigint;not null;default:0;comment:UserId"`
|
|
Status uint8 `gorm:"type:tinyint(1);not null;default:1;comment:Status"`
|
|
Follows []Follow `gorm:"foreignKey:TicketId;references:Id"`
|
|
CreatedAt time.Time `gorm:"<-:create;comment:Create Time"`
|
|
UpdatedAt time.Time `gorm:"comment:Update Time"`
|
|
}
|
|
type customTicketLogicModel interface {
|
|
QueryTicketDetail(ctx context.Context, id int64) (*Details, error)
|
|
InsertTicketFollow(ctx context.Context, data *Follow) error
|
|
QueryTicketList(ctx context.Context, page, size int, userId int64, status *uint8, search string) (int64, []*Ticket, error)
|
|
UpdateTicketStatus(ctx context.Context, id, userId int64, status uint8) error
|
|
QueryWaitReplyTotal(ctx context.Context) (int64, error)
|
|
}
|
|
|
|
// NewModel returns a model for the database table.
|
|
func NewModel(conn *gorm.DB, c *redis.Client) Model {
|
|
return &customTicketModel{
|
|
defaultTicketModel: newTicketModel(conn, c),
|
|
}
|
|
}
|
|
|
|
// QueryTicketDetail returns the ticket details.
|
|
func (m *customTicketModel) QueryTicketDetail(ctx context.Context, id int64) (*Details, error) {
|
|
key := fmt.Sprintf("%s%v", cacheTicketDetailPrefix, id)
|
|
var data *Details
|
|
err := m.QueryCtx(ctx, &data, key, func(conn *gorm.DB, v interface{}) error {
|
|
return conn.Model(&Ticket{}).Where("id = ?", id).Preload("Follows").First(v).Error
|
|
})
|
|
return data, err
|
|
}
|
|
|
|
// InsertTicketFollow inserts a follow record.
|
|
func (m *customTicketModel) InsertTicketFollow(ctx context.Context, data *Follow) error {
|
|
key := fmt.Sprintf("%s%v", cacheTicketDetailPrefix, data.TicketId)
|
|
return m.ExecCtx(ctx, func(conn *gorm.DB) error {
|
|
return conn.Model(&Follow{}).Create(data).Error
|
|
}, key)
|
|
}
|
|
|
|
// QueryTicketList returns the ticket list.
|
|
func (m *customTicketModel) QueryTicketList(ctx context.Context, page, size int, userId int64, status *uint8, search string) (int64, []*Ticket, error) {
|
|
var data []*Ticket
|
|
var total int64
|
|
err := m.QueryNoCacheCtx(ctx, &data, func(conn *gorm.DB, v interface{}) error {
|
|
query := conn.Model(&Ticket{})
|
|
if userId > 0 {
|
|
query = query.Where("user_id = ?", userId)
|
|
}
|
|
if status != nil {
|
|
query = query.Where("status = ?", status)
|
|
} else {
|
|
query = query.Where("status != ?", 4)
|
|
}
|
|
if search != "" {
|
|
query = query.Where("title like ? or description like ?", "%"+search+"%", "%"+search+"%")
|
|
}
|
|
return query.Count(&total).Order("id desc").Limit(size).Offset((page - 1) * size).Find(v).Error
|
|
})
|
|
return total, data, err
|
|
}
|
|
|
|
// UpdateTicketStatus updates the ticket status.
|
|
func (m *customTicketModel) UpdateTicketStatus(ctx context.Context, id, userId int64, status uint8) error {
|
|
key := fmt.Sprintf("%s%v", cacheTicketDetailPrefix, id)
|
|
return m.ExecCtx(ctx, func(conn *gorm.DB) error {
|
|
conn = conn.Model(&Ticket{})
|
|
if userId > 0 {
|
|
conn = conn.Where("user_id = ?", userId)
|
|
}
|
|
return conn.Where("id = ?", id).Update("status", status).Error
|
|
}, key)
|
|
}
|
|
|
|
// QueryWaitReplyTotal returns the total number of tickets that are waiting for a reply.
|
|
func (m *customTicketModel) QueryWaitReplyTotal(ctx context.Context) (int64, error) {
|
|
var total int64
|
|
err := m.QueryNoCacheCtx(ctx, &total, func(conn *gorm.DB, v interface{}) error {
|
|
return conn.Model(&Ticket{}).Where("status = ?", Pending).Count(&total).Error
|
|
})
|
|
return total, err
|
|
}
|