gitroast/vendor/github.com/pingcap/tidb/structure/list.go

213 lines
5.0 KiB
Go

// Copyright 2015 PingCAP, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// See the License for the specific language governing permissions and
// limitations under the License.
package structure
import (
"encoding/binary"
"github.com/juju/errors"
"github.com/pingcap/tidb/kv"
"github.com/pingcap/tidb/terror"
)
type listMeta struct {
LIndex int64
RIndex int64
}
func (meta listMeta) Value() []byte {
buf := make([]byte, 16)
binary.BigEndian.PutUint64(buf[0:8], uint64(meta.LIndex))
binary.BigEndian.PutUint64(buf[8:16], uint64(meta.RIndex))
return buf
}
func (meta listMeta) IsEmpty() bool {
return meta.LIndex >= meta.RIndex
}
// LPush prepends one or multiple values to a list.
func (t *TxStructure) LPush(key []byte, values ...[]byte) error {
return t.listPush(key, true, values...)
}
// RPush appends one or multiple values to a list.
func (t *TxStructure) RPush(key []byte, values ...[]byte) error {
return t.listPush(key, false, values...)
}
func (t *TxStructure) listPush(key []byte, left bool, values ...[]byte) error {
if len(values) == 0 {
return nil
}
metaKey := t.encodeListMetaKey(key)
meta, err := t.loadListMeta(metaKey)
if err != nil {
return errors.Trace(err)
}
index := int64(0)
for _, v := range values {
if left {
meta.LIndex--
index = meta.LIndex
} else {
index = meta.RIndex
meta.RIndex++
}
dataKey := t.encodeListDataKey(key, index)
if err = t.txn.Set(dataKey, v); err != nil {
return errors.Trace(err)
}
}
return t.txn.Set(metaKey, meta.Value())
}
// LPop removes and gets the first element in a list.
func (t *TxStructure) LPop(key []byte) ([]byte, error) {
return t.listPop(key, true)
}
// RPop removes and gets the last element in a list.
func (t *TxStructure) RPop(key []byte) ([]byte, error) {
return t.listPop(key, false)
}
func (t *TxStructure) listPop(key []byte, left bool) ([]byte, error) {
metaKey := t.encodeListMetaKey(key)
meta, err := t.loadListMeta(metaKey)
if err != nil || meta.IsEmpty() {
return nil, errors.Trace(err)
}
index := int64(0)
if left {
index = meta.LIndex
meta.LIndex++
} else {
meta.RIndex--
index = meta.RIndex
}
dataKey := t.encodeListDataKey(key, index)
var data []byte
data, err = t.txn.Get(dataKey)
if err != nil {
return nil, errors.Trace(err)
}
if err = t.txn.Delete(dataKey); err != nil {
return nil, errors.Trace(err)
}
if !meta.IsEmpty() {
err = t.txn.Set(metaKey, meta.Value())
} else {
err = t.txn.Delete(metaKey)
}
return data, errors.Trace(err)
}
// LLen gets the length of a list.
func (t *TxStructure) LLen(key []byte) (int64, error) {
metaKey := t.encodeListMetaKey(key)
meta, err := t.loadListMeta(metaKey)
return meta.RIndex - meta.LIndex, errors.Trace(err)
}
// LIndex gets an element from a list by its index.
func (t *TxStructure) LIndex(key []byte, index int64) ([]byte, error) {
metaKey := t.encodeListMetaKey(key)
meta, err := t.loadListMeta(metaKey)
if err != nil || meta.IsEmpty() {
return nil, errors.Trace(err)
}
index = adjustIndex(index, meta.LIndex, meta.RIndex)
if index >= meta.LIndex && index < meta.RIndex {
return t.txn.Get(t.encodeListDataKey(key, index))
}
return nil, nil
}
// LSet updates an element in the list by its index.
func (t *TxStructure) LSet(key []byte, index int64, value []byte) error {
metaKey := t.encodeListMetaKey(key)
meta, err := t.loadListMeta(metaKey)
if err != nil || meta.IsEmpty() {
return errors.Trace(err)
}
index = adjustIndex(index, meta.LIndex, meta.RIndex)
if index >= meta.LIndex && index < meta.RIndex {
return t.txn.Set(t.encodeListDataKey(key, index), value)
}
return errors.Errorf("invalid index %d", index)
}
// LClear removes the list of the key.
func (t *TxStructure) LClear(key []byte) error {
metaKey := t.encodeListMetaKey(key)
meta, err := t.loadListMeta(metaKey)
if err != nil || meta.IsEmpty() {
return errors.Trace(err)
}
for index := meta.LIndex; index < meta.RIndex; index++ {
dataKey := t.encodeListDataKey(key, index)
if err = t.txn.Delete(dataKey); err != nil {
return errors.Trace(err)
}
}
return t.txn.Delete(metaKey)
}
func (t *TxStructure) loadListMeta(metaKey []byte) (listMeta, error) {
v, err := t.txn.Get(metaKey)
if terror.ErrorEqual(err, kv.ErrNotExist) {
err = nil
} else if err != nil {
return listMeta{}, errors.Trace(err)
}
meta := listMeta{0, 0}
if v == nil {
return meta, nil
}
if len(v) != 16 {
return meta, errors.Errorf("invalid list meta data")
}
meta.LIndex = int64(binary.BigEndian.Uint64(v[0:8]))
meta.RIndex = int64(binary.BigEndian.Uint64(v[8:16]))
return meta, nil
}
func adjustIndex(index int64, min, max int64) int64 {
if index >= 0 {
return index + min
}
return index + max
}