142 lines
3.4 KiB
Go
142 lines
3.4 KiB
Go
// Copyright (C) 2017 ScyllaDB
|
|
// Use of this source code is governed by a ALv2-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package qb
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
// TokenBuilder helps implement pagination using token function.
|
|
type TokenBuilder []string
|
|
|
|
// Token creates a new TokenBuilder.
|
|
func Token(columns ...string) TokenBuilder {
|
|
return TokenBuilder(columns)
|
|
}
|
|
|
|
// Eq produces token(column)=token(?).
|
|
func (t TokenBuilder) Eq() Cmp {
|
|
return t.cmp(eq, nil)
|
|
}
|
|
|
|
// EqValue produces token(column)=?
|
|
func (t TokenBuilder) EqValue() Cmp {
|
|
return t.valueCmp(eq, "")
|
|
}
|
|
|
|
// EqNamed produces token(column)=token(?) with a custom parameter name.
|
|
func (t TokenBuilder) EqNamed(names ...string) Cmp {
|
|
return t.cmp(eq, names)
|
|
}
|
|
|
|
// EqValueNamed produces token(column)=? with a custom parameter name.
|
|
func (t TokenBuilder) EqValueNamed(name string) Cmp {
|
|
return t.valueCmp(eq, name)
|
|
}
|
|
|
|
// Lt produces token(column)<token(?).
|
|
func (t TokenBuilder) Lt() Cmp {
|
|
return t.cmp(lt, nil)
|
|
}
|
|
|
|
// LtValue produces token(column)<?.
|
|
func (t TokenBuilder) LtValue() Cmp {
|
|
return t.valueCmp(lt, "")
|
|
}
|
|
|
|
// LtNamed produces token(column)<token(?) with a custom parameter name.
|
|
func (t TokenBuilder) LtNamed(names ...string) Cmp {
|
|
return t.cmp(lt, names)
|
|
}
|
|
|
|
// LtValueNamed produces token(column)<? with a custom parameter name.
|
|
func (t TokenBuilder) LtValueNamed(name string) Cmp {
|
|
return t.valueCmp(lt, name)
|
|
}
|
|
|
|
// LtOrEq produces token(column)<=token(?).
|
|
func (t TokenBuilder) LtOrEq() Cmp {
|
|
return t.cmp(leq, nil)
|
|
}
|
|
|
|
// LtOrEqValue produces token(column)<=?.
|
|
func (t TokenBuilder) LtOrEqValue() Cmp {
|
|
return t.valueCmp(leq, "")
|
|
}
|
|
|
|
// LtOrEqNamed produces token(column)<=token(?) with a custom parameter name.
|
|
func (t TokenBuilder) LtOrEqNamed(names ...string) Cmp {
|
|
return t.cmp(leq, names)
|
|
}
|
|
|
|
// LtOrEqValueNamed produces token(column)<=? with a custom parameter name.
|
|
func (t TokenBuilder) LtOrEqValueNamed(name string) Cmp {
|
|
return t.valueCmp(leq, name)
|
|
}
|
|
|
|
// Gt produces token(column)>token(?).
|
|
func (t TokenBuilder) Gt() Cmp {
|
|
return t.cmp(gt, nil)
|
|
}
|
|
|
|
// GtValue produces token(column)>?.
|
|
func (t TokenBuilder) GtValue() Cmp {
|
|
return t.valueCmp(gt, "")
|
|
}
|
|
|
|
// GtNamed produces token(column)>token(?) with a custom parameter name.
|
|
func (t TokenBuilder) GtNamed(names ...string) Cmp {
|
|
return t.cmp(gt, names)
|
|
}
|
|
|
|
// GtValueNamed produces token(column)>? with a custom parameter name.
|
|
func (t TokenBuilder) GtValueNamed(name string) Cmp {
|
|
return t.valueCmp(gt, name)
|
|
}
|
|
|
|
// GtOrEq produces token(column)>=token(?).
|
|
func (t TokenBuilder) GtOrEq() Cmp {
|
|
return t.cmp(geq, nil)
|
|
}
|
|
|
|
// GtOrEqValue produces token(column)>=?.
|
|
func (t TokenBuilder) GtOrEqValue() Cmp {
|
|
return t.valueCmp(geq, "")
|
|
}
|
|
|
|
// GtOrEqNamed produces token(column)>=token(?) with a custom parameter name.
|
|
func (t TokenBuilder) GtOrEqNamed(names ...string) Cmp {
|
|
return t.cmp(geq, names)
|
|
}
|
|
|
|
// GtOrEqValueNamed produces token(column)>=? with a custom parameter name.
|
|
func (t TokenBuilder) GtOrEqValueNamed(name string) Cmp {
|
|
return t.valueCmp(geq, name)
|
|
}
|
|
|
|
func (t TokenBuilder) cmp(op op, names []string) Cmp {
|
|
s := names
|
|
if s == nil {
|
|
s = t
|
|
}
|
|
return Cmp{
|
|
op: op,
|
|
column: fmt.Sprint("token(", strings.Join(t, ","), ")"),
|
|
value: Fn("token", s...),
|
|
}
|
|
}
|
|
|
|
func (t TokenBuilder) valueCmp(op op, name string) Cmp {
|
|
if name == "" {
|
|
name = "token"
|
|
}
|
|
return Cmp{
|
|
op: op,
|
|
column: fmt.Sprint("token(", strings.Join(t, ","), ")"),
|
|
value: param(name),
|
|
}
|
|
}
|