diff --git a/internal/model/nh_game_pit_model.go b/internal/model/nh_game_pit_model.go index efae8e1..850f7a6 100755 --- a/internal/model/nh_game_pit_model.go +++ b/internal/model/nh_game_pit_model.go @@ -4,6 +4,7 @@ import ( "context" "fmt" "github.com/zeromicro/go-zero/core/stores/sqlx" + "time" ) var _ NhGamePitModel = (*customNhGamePitModel)(nil) @@ -23,9 +24,11 @@ type ( ) func (m *customNhGamePitModel) UserExist(ctx context.Context, uid uint) bool { - query := fmt.Sprintf("SELECT `id` FROM %s WHERE `uid` = ? LIMIT 1", m.table) + + query := fmt.Sprintf("SELECT `id` FROM %s WHERE `uid` = ? AND `created_at` > ? LIMIT 1", m.table) var id int - err := m.conn.QueryRowCtx(ctx, &id, query, uid) + t := time.Now().Add(-time.Hour * 24) + err := m.conn.QueryRowCtx(ctx, &id, query, uid, t) if err != nil { return false }