gitea/vendor/github.com/pingcap/go-themis/themis_primary_lock.go

113 lines
2.6 KiB
Go

package themis
import (
"bytes"
"encoding/binary"
"github.com/juju/errors"
"github.com/ngaut/log"
"github.com/pingcap/go-hbase"
"github.com/pingcap/go-hbase/iohelper"
)
type themisPrimaryLock struct {
*themisLock
// {coordinate => type}
secondaries map[string]hbase.Type
}
func newThemisPrimaryLock() *themisPrimaryLock {
return &themisPrimaryLock{
themisLock: &themisLock{
clientAddr: "null",
},
secondaries: map[string]hbase.Type{},
}
}
func (l *themisPrimaryLock) Primary() Lock {
return l
}
func (l *themisPrimaryLock) Secondaries() []Lock {
var slocks []Lock
for k, v := range l.secondaries {
c := &hbase.ColumnCoordinate{}
// TODO: handle error, now just ignore
if err := c.ParseFromString(k); err != nil {
log.Warnf("parse from string error, column coordinate: %s, secondary: %s, error: %v", c, k, err)
continue
}
slock := newThemisSecondaryLock()
slock.primaryCoordinate = l.coordinate
slock.coordinate = c
slock.ts = l.ts
slock.typ = v
slocks = append(slocks, slock)
}
return slocks
}
func (l *themisPrimaryLock) Encode() []byte {
buf := bytes.NewBuffer(nil)
// set is primary
binary.Write(buf, binary.BigEndian, uint8(1))
l.themisLock.write(buf)
// write secondaries
binary.Write(buf, binary.BigEndian, int32(len(l.secondaries)))
for k, v := range l.secondaries {
c := &hbase.ColumnCoordinate{}
// TODO: handle error, now just log
if err := c.ParseFromString(k); err != nil {
log.Warnf("parse from string error, column coordinate: %s, secondary: %s, error: %v", c, k, err)
}
// TODO: handle error, now just log
if err := c.Write(buf); err != nil {
log.Warnf("write error, column coordinate: %s, buf: %s, error: %v", c, buf, err)
}
buf.WriteByte(uint8(v))
}
return buf.Bytes()
}
func (l *themisPrimaryLock) IsExpired() bool {
return l.themisLock.expired
}
func (l *themisPrimaryLock) getSecondaryColumnType(c *hbase.ColumnCoordinate) hbase.Type {
v, ok := l.secondaries[c.String()]
if !ok {
return hbase.TypeMinimum
}
return v
}
func (l *themisPrimaryLock) Role() LockRole {
return RolePrimary
}
func (l *themisPrimaryLock) addSecondary(col *hbase.ColumnCoordinate, t hbase.Type) {
l.secondaries[col.String()] = t
}
func (l *themisPrimaryLock) parse(buf iohelper.ByteMultiReader) error {
l.themisLock.parse(buf)
var sz int32
err := binary.Read(buf, binary.BigEndian, &sz)
if err != nil {
return errors.Trace(err)
}
for i := 0; i < int(sz); i++ {
c := &hbase.ColumnCoordinate{}
c.ParseField(buf)
b, err := buf.ReadByte()
if err != nil {
return errors.Trace(err)
}
t := hbase.Type(b)
l.addSecondary(c, t)
}
return nil
}