com.datastax.dse.protocol.internal.request.DseExecuteCodec Maven / Gradle / Ivy
/*
* Copyright DataStax, 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,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.datastax.dse.protocol.internal.request;
import static com.datastax.dse.protocol.internal.DseProtocolConstants.Version.DSE_V2;
import com.datastax.dse.protocol.internal.DseProtocolConstants;
import com.datastax.dse.protocol.internal.request.query.DseQueryOptionsCodec;
import com.datastax.oss.protocol.internal.Message;
import com.datastax.oss.protocol.internal.PrimitiveCodec;
import com.datastax.oss.protocol.internal.PrimitiveSizes;
import com.datastax.oss.protocol.internal.ProtocolConstants;
import com.datastax.oss.protocol.internal.request.Execute;
import com.datastax.oss.protocol.internal.request.query.QueryOptions;
public class DseExecuteCodec extends Message.Codec {
private final QueryOptions.Codec optionsCodec;
public DseExecuteCodec(int protocolVersion) {
super(ProtocolConstants.Opcode.EXECUTE, protocolVersion);
assert protocolVersion >= DseProtocolConstants.Version.DSE_V1;
this.optionsCodec = new DseQueryOptionsCodec(protocolVersion);
}
@Override
public void encode(B dest, Message message, PrimitiveCodec encoder) {
Execute execute = (Execute) message;
encoder.writeShortBytes(execute.queryId, dest);
if (protocolVersion >= DSE_V2) {
encoder.writeShortBytes(execute.resultMetadataId, dest);
}
optionsCodec.encode(dest, execute.options, encoder);
}
@Override
public int encodedSize(Message message) {
Execute execute = (Execute) message;
int size = PrimitiveSizes.sizeOfShortBytes(execute.queryId);
if (protocolVersion >= DSE_V2) {
assert execute.resultMetadataId != null;
size += PrimitiveSizes.sizeOfShortBytes(execute.resultMetadataId);
}
size += optionsCodec.encodedSize(execute.options);
return size;
}
@Override
public Message decode(B source, PrimitiveCodec decoder) {
byte[] queryId = decoder.readShortBytes(source);
byte[] resultMetadataId = (protocolVersion >= DSE_V2) ? decoder.readShortBytes(source) : null;
QueryOptions options = optionsCodec.decode(source, decoder);
return new Execute(queryId, resultMetadataId, options);
}
}