59 lines
1.9 KiB
Go
59 lines
1.9 KiB
Go
package document
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/redis/go-redis/v9"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type customDocumentLogicModel interface {
|
|
QueryDocumentDetail(ctx context.Context, id int64) (*Document, error)
|
|
QueryDocumentList(ctx context.Context, page, size int, tag string, search string) (int64, []*Document, error)
|
|
GetDocumentListByAll(ctx context.Context) (int64, []*Document, error)
|
|
}
|
|
|
|
// NewModel returns a model for the database table.
|
|
func NewModel(conn *gorm.DB, c *redis.Client) Model {
|
|
return &customDocumentModel{
|
|
defaultDocumentModel: newDocumentModel(conn, c),
|
|
}
|
|
}
|
|
|
|
// QueryDocumentDetail queries the details of a document.
|
|
func (m *customDocumentModel) QueryDocumentDetail(ctx context.Context, id int64) (*Document, error) {
|
|
var data Document
|
|
err := m.QueryNoCacheCtx(ctx, &data, func(conn *gorm.DB, v interface{}) error {
|
|
return conn.Model(&Document{}).Preload("Group").Where("id = ?", id).Find(v).Error
|
|
})
|
|
return &data, err
|
|
}
|
|
|
|
// QueryDocumentList queries a list of documents.
|
|
func (m *customDocumentModel) QueryDocumentList(ctx context.Context, page, size int, tag string, search string) (int64, []*Document, error) {
|
|
var data []*Document
|
|
var total int64
|
|
err := m.QueryNoCacheCtx(ctx, &data, func(conn *gorm.DB, v interface{}) error {
|
|
db := conn.Model(&Document{})
|
|
if tag != "" {
|
|
db = db.Where("FIND_IN_SET(?, tags)", tag)
|
|
}
|
|
if search != "" {
|
|
db = db.Where("title LIKE ? OR content LIKE ?", "%"+search+"%", "%"+search+"%")
|
|
}
|
|
return db.Count(&total).Offset((page - 1) * size).Limit(size).Find(v).Error
|
|
})
|
|
return total, data, err
|
|
}
|
|
|
|
// GetDocumentListByAll queries a list of documents.
|
|
func (m *customDocumentModel) GetDocumentListByAll(ctx context.Context) (int64, []*Document, error) {
|
|
var data []*Document
|
|
var total int64
|
|
show := true
|
|
err := m.QueryNoCacheCtx(ctx, &data, func(conn *gorm.DB, v interface{}) error {
|
|
return conn.Model(&Document{}).Where("`show` = ?", &show).Count(&total).Find(v).Error
|
|
})
|
|
return total, data, err
|
|
}
|