You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
89 lines
4.1 KiB
89 lines
4.1 KiB
package memcache
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
)
|
|
|
|
func legalKey(key string) bool {
|
|
if len(key) > 250 || len(key) == 0 {
|
|
return false
|
|
}
|
|
for i := 0; i < len(key); i++ {
|
|
if key[i] <= ' ' || key[i] == 0x7f {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
// MockWith error
|
|
func MockWith(err error) Conn {
|
|
return errConn{err}
|
|
}
|
|
|
|
type errConn struct{ err error }
|
|
|
|
func (c errConn) Err() error { return c.err }
|
|
func (c errConn) Close() error { return c.err }
|
|
func (c errConn) Add(*Item) error { return c.err }
|
|
func (c errConn) Set(*Item) error { return c.err }
|
|
func (c errConn) Replace(*Item) error { return c.err }
|
|
func (c errConn) CompareAndSwap(*Item) error { return c.err }
|
|
func (c errConn) Get(string) (*Item, error) { return nil, c.err }
|
|
func (c errConn) GetMulti([]string) (map[string]*Item, error) { return nil, c.err }
|
|
func (c errConn) Touch(string, int32) error { return c.err }
|
|
func (c errConn) Delete(string) error { return c.err }
|
|
func (c errConn) Increment(string, uint64) (uint64, error) { return 0, c.err }
|
|
func (c errConn) Decrement(string, uint64) (uint64, error) { return 0, c.err }
|
|
func (c errConn) Scan(*Item, interface{}) error { return c.err }
|
|
func (c errConn) AddContext(context.Context, *Item) error { return c.err }
|
|
func (c errConn) SetContext(context.Context, *Item) error { return c.err }
|
|
func (c errConn) ReplaceContext(context.Context, *Item) error { return c.err }
|
|
func (c errConn) GetContext(context.Context, string) (*Item, error) { return nil, c.err }
|
|
func (c errConn) DecrementContext(context.Context, string, uint64) (uint64, error) { return 0, c.err }
|
|
func (c errConn) CompareAndSwapContext(context.Context, *Item) error { return c.err }
|
|
func (c errConn) TouchContext(context.Context, string, int32) error { return c.err }
|
|
func (c errConn) DeleteContext(context.Context, string) error { return c.err }
|
|
func (c errConn) IncrementContext(context.Context, string, uint64) (uint64, error) { return 0, c.err }
|
|
func (c errConn) GetMultiContext(context.Context, []string) (map[string]*Item, error) {
|
|
return nil, c.err
|
|
}
|
|
|
|
// RawItem item with FlagRAW flag.
|
|
//
|
|
// Expiration is the cache expiration time, in seconds: either a relative
|
|
// time from now (up to 1 month), or an absolute Unix epoch time.
|
|
// Zero means the Item has no expiration time.
|
|
func RawItem(key string, data []byte, flags uint32, expiration int32) *Item {
|
|
return &Item{Key: key, Flags: flags | FlagRAW, Value: data, Expiration: expiration}
|
|
}
|
|
|
|
// JSONItem item with FlagJSON flag.
|
|
//
|
|
// Expiration is the cache expiration time, in seconds: either a relative
|
|
// time from now (up to 1 month), or an absolute Unix epoch time.
|
|
// Zero means the Item has no expiration time.
|
|
func JSONItem(key string, v interface{}, flags uint32, expiration int32) *Item {
|
|
return &Item{Key: key, Flags: flags | FlagJSON, Object: v, Expiration: expiration}
|
|
}
|
|
|
|
// ProtobufItem item with FlagProtobuf flag.
|
|
//
|
|
// Expiration is the cache expiration time, in seconds: either a relative
|
|
// time from now (up to 1 month), or an absolute Unix epoch time.
|
|
// Zero means the Item has no expiration time.
|
|
func ProtobufItem(key string, message proto.Message, flags uint32, expiration int32) *Item {
|
|
return &Item{Key: key, Flags: flags | FlagProtobuf, Object: message, Expiration: expiration}
|
|
}
|
|
|
|
func shrinkDeadline(ctx context.Context, timeout time.Duration) time.Time {
|
|
// TODO: ignored context deadline to compatible old behaviour.
|
|
//deadline, ok := ctx.Deadline()
|
|
//if ok {
|
|
// return deadline
|
|
//}
|
|
return time.Now().Add(timeout)
|
|
}
|
|
|