func (tbl *Table) Sync(req *PkgArgs) ([]byte, bool) {
var in proto.PkgMultiOp
_, err := in.Decode(req.Pkg)
if err == nil {
// Sync full DB, including DB 0
var wb = tbl.db.NewWriteBatch()
for i := 0; i < len(in.Kvs); i++ {
tbl.setSyncKV(wb, in.DbId, &in.Kvs[i])
}
tbl.rwMtx.RLock()
err = tbl.db.Commit(wb)
tbl.rwMtx.RUnlock()
wb.Destroy()
if err == nil {
return nil, true
} else {
in.SetErrCode(table.EcWriteFail)
log.Printf("setRawKV failed: %s\n", err)
}
} else {
in.SetErrCode(table.EcDecodeFail)
}
return replyHandle(&in), table.EcOk == in.ErrCode
}
func myMGet(in proto.PkgMultiOp, au Authorize, wa *WriteAccess,
t *testing.T) proto.PkgMultiOp {
var pkg = make([]byte, in.Length())
_, err := in.Encode(pkg)
if err != nil {
t.Fatalf("Encode failed: ", err)
}
pkg = testTbl.MGet(&PkgArgs{in.Cmd, in.DbId, in.Seq, pkg}, au, wa)
var out proto.PkgMultiOp
_, err = out.Decode(pkg)
if err != nil {
t.Fatalf("Decode failed: ", err)
}
if out.ErrCode != 0 {
t.Fatalf("Failed with ErrCode %d", out.ErrCode)
}
if out.DbId != in.DbId || out.Seq != in.Seq {
t.Fatalf("DbId/Seq mismatch")
}
return out
}
func myMIncr(in proto.PkgMultiOp, au Authorize, wa *WriteAccess, expected bool,
t *testing.T) proto.PkgMultiOp {
var pkg = make([]byte, in.Length())
_, err := in.Encode(pkg)
if err != nil {
t.Fatalf("Encode failed: ", err)
}
pkg, ok := testTbl.MIncr(&PkgArgs{in.Cmd, in.DbId, in.Seq, pkg}, au, wa)
if ok != expected {
if expected {
t.Fatalf("MIncr failed")
} else {
t.Fatalf("MIncr should fail")
}
}
var out proto.PkgMultiOp
_, err = out.Decode(pkg)
if err != nil {
t.Fatalf("Decode failed: ", err)
}
if expected {
if out.ErrCode != 0 {
t.Fatalf("Failed with ErrCode %d", out.ErrCode)
}
}
if out.DbId != in.DbId || out.Seq != in.Seq {
t.Fatalf("DbId/Seq mismatch")
}
return out
}
func checkMultiOp(in *proto.PkgMultiOp, req *PkgArgs, au Authorize) bool {
n, err := in.Decode(req.Pkg)
if err != nil || n != len(req.Pkg) {
in.ErrCode = table.EcDecodeFail
}
if in.ErrCode == 0 && in.DbId == proto.AdminDbId {
in.ErrCode = table.EcInvDbId
}
if in.ErrCode == 0 && !au.IsAuth(in.DbId) {
in.ErrCode = table.EcNoPrivilege
}
if in.ErrCode != 0 {
in.Kvs = nil
}
return in.ErrCode == 0
}
// Get call reply. The real reply types are:
// Auth/Ping/(Z)Set/(Z)Del: nil;
// (Z)Get: GetReply;
// (Z)Incr: IncrReply;
// (Z)MGet: []GetReply;
// (Z)MSet: []SetReply;
// (Z)MDel: []DelReply;
// (Z)MIncr: []IncrReply;
// (Z)Scan: ScanReply;
// Dump: DumpReply;
func (call *Call) Reply() (interface{}, error) {
if call.err != nil {
return nil, call.err
}
if !call.ready {
return nil, ErrCallNotReady
}
if proto.CmdAuth == call.cmd ||
proto.CmdPing == call.cmd ||
proto.CmdIncr == call.cmd ||
proto.CmdDel == call.cmd ||
proto.CmdSet == call.cmd ||
proto.CmdGet == call.cmd {
var p proto.PkgOneOp
_, err := p.Decode(call.pkg)
if err != nil {
call.err = err
return nil, call.err
}
if p.ErrCode < 0 {
return nil, getErr(p.ErrCode)
}
switch call.cmd {
case proto.CmdAuth:
return nil, nil
case proto.CmdPing:
return nil, nil
case proto.CmdIncr:
return IncrReply{p.ErrCode, p.TableId, copyBytes(p.RowKey),
copyBytes(p.ColKey), copyBytes(p.Value), p.Score}, nil
case proto.CmdDel:
return nil, nil
case proto.CmdSet:
return nil, nil
case proto.CmdGet:
return GetReply{p.ErrCode, p.TableId, copyBytes(p.RowKey),
copyBytes(p.ColKey), copyBytes(p.Value), p.Score, p.Cas}, nil
}
}
if proto.CmdMIncr == call.cmd ||
proto.CmdMDel == call.cmd ||
proto.CmdMSet == call.cmd ||
proto.CmdMGet == call.cmd {
var p proto.PkgMultiOp
_, err := p.Decode(call.pkg)
if err != nil {
call.err = err
return nil, call.err
}
if p.ErrCode < 0 {
return nil, getErr(p.ErrCode)
}
switch call.cmd {
case proto.CmdMIncr:
var r = make([]IncrReply, len(p.Kvs))
for i := 0; i < len(r); i++ {
r[i] = IncrReply{p.Kvs[i].ErrCode, p.Kvs[i].TableId,
copyBytes(p.Kvs[i].RowKey), copyBytes(p.Kvs[i].ColKey),
copyBytes(p.Kvs[i].Value), p.Kvs[i].Score}
}
return r, nil
case proto.CmdMDel:
var r = make([]DelReply, len(p.Kvs))
for i := 0; i < len(r); i++ {
r[i] = DelReply{p.Kvs[i].ErrCode, p.Kvs[i].TableId,
copyBytes(p.Kvs[i].RowKey), copyBytes(p.Kvs[i].ColKey)}
}
return r, nil
case proto.CmdMSet:
var r = make([]SetReply, len(p.Kvs))
for i := 0; i < len(r); i++ {
r[i] = SetReply{p.Kvs[i].ErrCode, p.Kvs[i].TableId,
copyBytes(p.Kvs[i].RowKey), copyBytes(p.Kvs[i].ColKey)}
}
return r, nil
case proto.CmdMGet:
var r = make([]GetReply, len(p.Kvs))
for i := 0; i < len(r); i++ {
r[i] = GetReply{p.Kvs[i].ErrCode, p.Kvs[i].TableId,
copyBytes(p.Kvs[i].RowKey), copyBytes(p.Kvs[i].ColKey),
copyBytes(p.Kvs[i].Value), p.Kvs[i].Score, p.Kvs[i].Cas}
}
return r, nil
}
}
switch call.cmd {
case proto.CmdScan:
var p proto.PkgScanResp
_, err := p.Decode(call.pkg)
if err != nil {
call.err = err
return nil, call.err
}
if p.ErrCode < 0 {
return nil, getErr(p.ErrCode)
}
var r ScanReply
r.ctx = call.ctx.(scanContext)
r.TableId = r.ctx.tableId
r.RowKey = r.ctx.rowKey
r.End = (p.PkgFlag&proto.FlagScanEnd != 0)
r.Kvs = make([]ScanKV, len(p.Kvs))
for i := 0; i < len(p.Kvs); i++ {
r.Kvs[i] = ScanKV{copyBytes(p.Kvs[i].ColKey),
copyBytes(p.Kvs[i].Value), p.Kvs[i].Score}
}
return r, nil
case proto.CmdDump:
var p proto.PkgDumpResp
_, err := p.Decode(call.pkg)
if err != nil {
call.err = err
return nil, call.err
}
if p.ErrCode < 0 {
return nil, getErr(p.ErrCode)
}
var r DumpReply
r.ctx = call.ctx.(dumpContext)
r.ctx.lastSlotId = p.LastSlotId
r.ctx.slotStart = (p.PkgFlag&proto.FlagDumpSlotStart != 0)
r.End = (p.PkgFlag&proto.FlagDumpEnd != 0)
r.Kvs = make([]DumpKV, len(p.Kvs))
for i := 0; i < len(p.Kvs); i++ {
r.Kvs[i] = DumpKV{p.Kvs[i].TableId, p.Kvs[i].ColSpace,
copyBytes(p.Kvs[i].RowKey), copyBytes(p.Kvs[i].ColKey),
copyBytes(p.Kvs[i].Value), p.Kvs[i].Score}
}
return r, nil
}
switch call.cmd {
case proto.CmdSlaveOf:
return call.replyInnerCtrl(&ctrl.PkgSlaveOf{})
case proto.CmdMigrate:
return call.replyInnerCtrl(&ctrl.PkgMigrate{})
case proto.CmdSlaveSt:
return call.replyInnerCtrl(&ctrl.PkgSlaveStatus{})
case proto.CmdDelSlot:
return call.replyInnerCtrl(&ctrl.PkgDelSlot{})
}
return nil, ErrUnknownCmd
}