public void Should_Build_Req()
{
Store cmd = BuildStoreReq();
Assert.AreEqual(MessageCode.TsPutResp, cmd.ExpectedCode);
TsPutReq pb = (TsPutReq)cmd.ConstructPbRequest();
Assert.AreEqual(Table, RiakString.FromBytes(pb.table));
CollectionAssert.AreEqual(TsCols, pb.columns);
CollectionAssert.AreEqual(TsRows, pb.rows);
}
public override RpbReq ConstructPbRequest()
{
var req = new TsPutReq();
req.table = CommandOptions.Table;
if (EnumerableUtil.NotNullOrEmpty(CommandOptions.Columns))
{
req.columns.AddRange(CommandOptions.Columns.Select(c => c.ToTsColumn()));
}
req.rows.AddRange(CommandOptions.Rows.Select(r => r.ToTsRow()));
return(req);
}
// TODO FUTURE this all really should be in a codec
public override RiakReq ConstructRequest(bool useTtb)
{
RiakReq rv;
if (useTtb)
{
usingTtb = true;
expectedCode = MessageCode.TsTtbMsg;
byte[] buffer;
string tableName = CommandOptions.Table;
ICollection <Row> rows = CommandOptions.Rows;
using (var os = new OtpOutputStream())
{
os.WriteByte(OtpExternal.VersionTag);
// {tsputreq, tableName, emptyList, rows}
os.WriteTupleHead(4);
os.WriteAtom(TsPutReqAtom);
os.WriteStringAsBinary(tableName);
os.WriteNil();
if (rows.Count > 0)
{
os.WriteListHead(rows.Count);
foreach (Row r in CommandOptions.Rows)
{
os.WriteTupleHead(r.Cells.Count);
foreach (Cell c in r.Cells)
{
c.ToTtbCell(os);
}
}
os.WriteNil();
}
else
{
os.WriteNil();
}
buffer = os.ToArray();
}
rv = new TsTtbMsg(buffer);
}
else
{
var req = new TsPutReq();
req.table = CommandOptions.Table;
if (EnumerableUtil.NotNullOrEmpty(CommandOptions.Columns))
{
req.columns.AddRange(CommandOptions.Columns.Select(c => c.ToTsColumn()));
}
req.rows.AddRange(CommandOptions.Rows.Select(r => r.ToTsRow()));
rv = req;
}
return(rv);
}
