jager/novatask
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: 翻页

Browse Source
This commit is contained in:
jager
2025-05-07 14:36:19 +08:00
parent a063c2eae5
commit 2cc980266d
4 changed files with 55 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
internal/model/nh_castile_token_log_model.go
View File

@@ -16,7 +16,7 @@ type (
nhCastileTokenLogModel
WithSession(session sqlx.Session) NhCastileTokenLogModel
UpdateStatus(ctx context.Context, status uint, id uint) error
List(ctx context.Context, uid uint, roleId uint64, page, pageSize int) ([]*NhCastileTokenLog, error)
List(ctx context.Context, uid uint, roleId uint64, page, pageSize int) ([]*NhCastileTokenLog, int64, error)
}
customNhCastileTokenLogModel struct {
@@ -41,20 +41,35 @@ func (m *customNhCastileTokenLogModel) UpdateStatus(ctx context.Context, status
return err
}
func (m *customNhCastileTokenLogModel) List(ctx context.Context, uid uint, roleId uint64, page, pageSize int) ([]*NhCastileTokenLog, error) {
func (m *customNhCastileTokenLogModel) List(ctx context.Context, uid uint, roleId uint64, page, pageSize int) ([]*NhCastileTokenLog, int64, error) {
var query string
var result []*NhCastileTokenLog
var err error
var args = []any{uid}
if roleId == 0 {
query = fmt.Sprintf("select %s from %s where uid = ? order by id desc limit ?, ?", nhCastileTokenLogRows, m.table)
err = m.conn.QueryRowsCtx(ctx, &result, query, uid, (page-1)*pageSize, pageSize)
query = "select %s from %s where uid = ?"
} else {
query = fmt.Sprintf("select %s from %s where uid = ? and role_id = ? order by id desc limit ?, ?", nhCastileTokenLogRows, m.table)
err = m.conn.QueryRowsCtx(ctx, &result, query, uid, roleId, (page-1)*pageSize, pageSize)
query = "select %s from %s where uid = ? and role_id = ?"
args = append(args, roleId)
}
var err error
var count int64
var result []*NhCastileTokenLog
err = m.conn.QueryRowCtx(ctx, &count, fmt.Sprintf(query, "count(*) as count", m.table), args...)
if err != nil {
return nil, 0, err
}
offset := (page - 1) * pageSize
if offset >= int(count) {
return result, count, nil
}
query += " order by id desc limit ?, ?"
args = append(args, offset, pageSize)
err = m.conn.QueryRowsCtx(ctx, &result, fmt.Sprintf(query, nhCastileTokenLogRows, m.table), args...)
if err != nil && !errors.Is(err, sqlx.ErrNotFound) {
return nil, err
return nil, 0, err
}
return result, nil
return result, count, nil
}

37
internal/model/nh_stake_points_log_model.go
View File

@@ -15,7 +15,7 @@ type (
NhStakePointsLogModel interface {
nhStakePointsLogModel
WithSession(session sqlx.Session) NhStakePointsLogModel
List(ctx context.Context, uid uint, roleId uint64, page, pageSize int) ([]*NhStakePointsLog, error)
List(ctx context.Context, uid uint, roleId uint64, page, pageSize int) ([]*NhStakePointsLog, int64, error)
}
customNhStakePointsLogModel struct {
@@ -23,22 +23,37 @@ type (
}
)
func (m *customNhStakePointsLogModel) List(ctx context.Context, uid uint, roleId uint64, page, pageSize int) ([]*NhStakePointsLog, error) {
func (m *customNhStakePointsLogModel) List(ctx context.Context, uid uint, roleId uint64, page, pageSize int) ([]*NhStakePointsLog, int64, error) {
var query string
var result []*NhStakePointsLog
var err error
var args = []any{uid}
if roleId == 0 {
query = fmt.Sprintf("select %s from %s where uid = ? order by id desc limit ?, ?", nhStakePointsLogRows, m.table)
err = m.conn.QueryRowsCtx(ctx, &result, query, uid, (page-1)*pageSize, pageSize)
query = "select %s from %s where uid = ?"
} else {
query = fmt.Sprintf("select %s from %s where uid = ? and role_id = ? order by id desc limit ?, ?", nhStakePointsLogRows, m.table)
err = m.conn.QueryRowsCtx(ctx, &result, query, uid, roleId, (page-1)*pageSize, pageSize)
query = "select %s from %s where uid = ? and role_id = ?"
args = append(args, roleId)
}
var err error
var count int64
var result []*NhStakePointsLog
err = m.conn.QueryRowCtx(ctx, &count, fmt.Sprintf(query, "count(*) as count", m.table), args...)
if err != nil {
return nil, 0, err
}
if err != nil && !errors.Is(err, sqlx.ErrNotFound) {
return nil, err
offset := (page - 1) * pageSize
if offset >= int(count) {
return result, count, nil
}
return result, nil
query += " order by id desc limit ?, ?"
args = append(args, offset, pageSize)
err = m.conn.QueryRowsCtx(ctx, &result, query, args...)
if err != nil && !errors.Is(err, sqlx.ErrNotFound) {
return nil, count, err
}
return result, count, nil
}
// NewNhStakePointsLogModel returns a model for the database table.