func (zkNode *ZkNode) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
key := bson.ReadCString(buf)
switch key {
case "Path":
zkNode.Path = bson.DecodeString(buf, kind)
case "Data":
zkNode.Data = bson.DecodeString(buf, kind)
case "Stat":
if kind != bson.Object {
panic(bson.NewBsonError("Unexpected data type %v for Stat", kind))
}
zkNode.Stat.UnmarshalBson(buf, kind)
case "Children":
zkNode.Children = bson.DecodeStringArray(buf, kind)
case "Cached":
zkNode.Cached = bson.DecodeBool(buf, kind)
case "Stale":
zkNode.Stale = bson.DecodeBool(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
// UnmarshalBson unmarshals BatchQueryShard from buf.
func (bqs *BatchQueryShard) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
keyName := bson.ReadCString(buf)
switch keyName {
case "Queries":
bqs.Queries = tproto.DecodeQueriesBson(buf, kind)
case "Keyspace":
bqs.Keyspace = bson.DecodeString(buf, kind)
case "Shards":
bqs.Shards = bson.DecodeStringArray(buf, kind)
case "TabletType":
bqs.TabletType = topo.TabletType(bson.DecodeString(buf, kind))
case "Session":
if kind != bson.Null {
bqs.Session = new(Session)
bqs.Session.UnmarshalBson(buf, kind)
}
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
// UnmarshalBson unmarshals QueryResult from buf.
func (qr *QueryResult) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
keyName := bson.ReadCString(buf)
switch keyName {
case "Fields":
qr.Fields = mproto.DecodeFieldsBson(buf, kind)
case "RowsAffected":
qr.RowsAffected = bson.DecodeUint64(buf, kind)
case "InsertId":
qr.InsertId = bson.DecodeUint64(buf, kind)
case "Rows":
qr.Rows = mproto.DecodeRowsBson(buf, kind)
case "Session":
if kind != bson.Null {
qr.Session = new(Session)
qr.Session.UnmarshalBson(buf, kind)
}
case "Error":
qr.Error = bson.DecodeString(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
func (sk *SrvKeyspace) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
keyName := bson.ReadCString(buf)
switch keyName {
case "Partitions":
sk.Partitions = DecodeKeyspacePartitionMap(buf, kind)
case "Shards":
sk.Shards = DecodeSrvShardArray(buf, kind)
case "TabletTypes":
sk.TabletTypes = DecodeTabletTypeArray(buf, kind)
case "ShardingColumnName":
sk.ShardingColumnName = bson.DecodeString(buf, kind)
case "ShardingColumnType":
sk.ShardingColumnType = key.KeyspaceIdType(bson.DecodeString(buf, kind))
case "ServedFrom":
sk.ServedFrom = DecodeServedFrom(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
func (sqs *StreamQueryKeyRange) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
keyName := bson.ReadCString(buf)
switch keyName {
case "Sql":
sqs.Sql = bson.DecodeString(buf, kind)
case "BindVariables":
sqs.BindVariables = tproto.DecodeBindVariablesBson(buf, kind)
case "Keyspace":
sqs.Keyspace = bson.DecodeString(buf, kind)
case "KeyRange":
sqs.KeyRange = bson.DecodeString(buf, kind)
case "TabletType":
sqs.TabletType = topo.TabletType(bson.DecodeString(buf, kind))
case "Session":
if kind != bson.Null {
sqs.Session = new(Session)
sqs.Session.UnmarshalBson(buf, kind)
}
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
func (ste *StreamEvent) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
key := bson.ReadCString(buf)
switch key {
case "Category":
ste.Category = bson.DecodeString(buf, kind)
case "TableName":
ste.TableName = bson.DecodeString(buf, kind)
case "PKColNames":
ste.PKColNames = bson.DecodeStringArray(buf, kind)
case "PKValues":
ste.PKValues = UnmarshalPKValuesBson(buf, kind)
case "Sql":
ste.Sql = bson.DecodeString(buf, kind)
case "Timestamp":
ste.Timestamp = bson.DecodeInt64(buf, kind)
case "GroupId":
ste.GroupId = bson.DecodeInt64(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
func (kp *KeyspacePartition) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
keyName := bson.ReadCString(buf)
switch keyName {
case "Shards":
kp.Shards = DecodeSrvShardArray(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
func (qrl *QueryResultList) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
key := bson.ReadCString(buf)
switch key {
case "List":
qrl.List = DecodeResultsBson(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
func (zkPathV *ZkPathV) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
key := bson.ReadCString(buf)
switch key {
case "Paths":
zkPathV.Paths = bson.DecodeStringArray(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
func (zkNodeV *ZkNodeV) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
key := bson.ReadCString(buf)
switch key {
case "Nodes":
zkNodeV.Nodes = unmarshalZkNodeArray(buf, "ZkNodeV.Nodes", kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
func (blt *BinlogTransaction) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
key := bson.ReadCString(buf)
switch key {
case "Statements":
blt.Statements = UnmarshalStatementsBson(buf, kind)
case "GroupId":
blt.GroupId = bson.DecodeInt64(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
func (stmt *Statement) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
key := bson.ReadCString(buf)
switch key {
case "Category":
stmt.Category = int(bson.DecodeInt64(buf, kind))
case "Sql":
stmt.Sql = bson.DecodeBinary(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
func (kr *KeyRange) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
key := bson.ReadCString(buf)
switch key {
case "Start":
kr.Start = KeyspaceId(bson.DecodeString(buf, kind))
case "End":
kr.End = KeyspaceId(bson.DecodeString(buf, kind))
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
// UnmarshalBson unmarshals request to the given byte buffer as verifying the
// kind
func (req *RequestBson) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
key := bson.ReadCString(buf)
switch key {
case "ServiceMethod":
req.ServiceMethod = bson.DecodeString(buf, kind)
case "Seq":
req.Seq = bson.DecodeUint64(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
func (session *Session) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
key := bson.ReadCString(buf)
switch key {
case "TransactionId":
session.TransactionId = bson.DecodeInt64(buf, kind)
case "SessionId":
session.SessionId = bson.DecodeInt64(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
func (bdq *BoundQuery) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
key := bson.ReadCString(buf)
switch key {
case "Sql":
bdq.Sql = bson.DecodeString(buf, kind)
case "BindVariables":
bdq.BindVariables = DecodeBindVariablesBson(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
// UnmarshalBson unmarshals Session from buf.
func (session *Session) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
keyName := bson.ReadCString(buf)
switch keyName {
case "InTransaction":
session.InTransaction = bson.DecodeBool(buf, kind)
case "ShardSessions":
session.ShardSessions = decodeShardSessionsBson(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
func (ss *SrvShard) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
keyName := bson.ReadCString(buf)
switch keyName {
case "KeyRange":
ss.KeyRange.UnmarshalBson(buf, kind)
case "ServedTypes":
ss.ServedTypes = DecodeTabletTypeArray(buf, kind)
case "TabletTypes":
ss.TabletTypes = DecodeTabletTypeArray(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
func (query *Query) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
key := bson.ReadCString(buf)
switch key {
case "Sql":
query.Sql = bson.DecodeString(buf, kind)
case "BindVariables":
query.BindVariables = DecodeBindVariablesBson(buf, kind)
case "TransactionId":
query.TransactionId = bson.DecodeInt64(buf, kind)
case "SessionId":
query.SessionId = bson.DecodeInt64(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
func (qr *QueryResult) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
key := bson.ReadCString(buf)
switch key {
case "Fields":
qr.Fields = DecodeFieldsBson(buf, kind)
case "RowsAffected":
qr.RowsAffected = bson.DecodeUint64(buf, kind)
case "InsertId":
qr.InsertId = bson.DecodeUint64(buf, kind)
case "Rows":
qr.Rows = DecodeRowsBson(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
// UnmarshalBson unmarshals ShardSession from buf.
func (shardSession *ShardSession) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
keyName := bson.ReadCString(buf)
switch keyName {
case "Keyspace":
shardSession.Keyspace = bson.DecodeString(buf, kind)
case "Shard":
shardSession.Shard = bson.DecodeString(buf, kind)
case "TabletType":
shardSession.TabletType = topo.TabletType(bson.DecodeString(buf, kind))
case "TransactionId":
shardSession.TransactionId = bson.DecodeInt64(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
// UnmarshalBson unmarshals QueryResultList from buf.
func (qrl *QueryResultList) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
keyName := bson.ReadCString(buf)
switch keyName {
case "List":
qrl.List = tproto.DecodeResultsBson(buf, kind)
case "Session":
if kind != bson.Null {
qrl.Session = new(Session)
qrl.Session.UnmarshalBson(buf, kind)
}
case "Error":
qrl.Error = bson.DecodeString(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
func (zkStat *ZkStat) UnmarshalBson(buf *bytes.Buffer, kind byte) {
bson.VerifyObject(kind)
bson.Next(buf, 4)
kind = bson.NextByte(buf)
for kind != bson.EOO {
key := bson.ReadCString(buf)
switch key {
case "Czxid":
zkStat.czxid = bson.DecodeInt64(buf, kind)
case "Mzxid":
zkStat.mzxid = bson.DecodeInt64(buf, kind)
case "CTime":
zkStat.cTime = bson.DecodeTime(buf, kind)
case "MTime":
zkStat.mTime = bson.DecodeTime(buf, kind)
case "Version":
zkStat.version = bson.DecodeInt(buf, kind)
case "CVersion":
zkStat.cVersion = bson.DecodeInt(buf, kind)
case "AVersion":
zkStat.aVersion = bson.DecodeInt(buf, kind)
case "EphemeralOwner":
zkStat.ephemeralOwner = bson.DecodeInt64(buf, kind)
case "DataLength":
zkStat.dataLength = bson.DecodeInt(buf, kind)
case "NumChildren":
zkStat.numChildren = bson.DecodeInt(buf, kind)
case "Pzxid":
zkStat.pzxid = bson.DecodeInt64(buf, kind)
default:
bson.Skip(buf, kind)
}
kind = bson.NextByte(buf)
}
}
