/*
* Copyright (C) 2017 Julien Viet
*
* 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 io.vertx.pgclient.impl.codec;
import io.vertx.sqlclient.impl.command.PrepareStatementCommand;
import io.vertx.sqlclient.impl.PreparedStatement;
import java.util.List;
class PrepareStatementCommandCodec extends PgCommandCodec<PreparedStatement, PrepareStatementCommand> {
private PgParamDesc parameterDesc;
private PgRowDesc rowDesc;
private long statement;
PrepareStatementCommandCodec(PrepareStatementCommand cmd) {
super(cmd);
}
@Override
void encode(PgEncoder encoder) {
if (cmd.isManaged()) {
statement = encoder.nextStatementName();
} else {
// Use unnamed prepared statements that don't need to be closed
statement = 0L;
}
List<Class<?>> parameterTypes = cmd.parameterTypes();
DataType[] parameterTypes2 = parameterTypes != null ? build(parameterTypes) : null;
encoder.writeParse(cmd.sql(), statement, parameterTypes2);
encoder.writeDescribe(new Describe(statement, null));
encoder.writeSync();
}
private DataType[] build(List<Class<?>> parameterTypes) {
int len = parameterTypes.size();
DataType[] dataType = new DataType[len];
for (int i = 0;i < len;i++) {
DataType type = DataType.lookup(parameterTypes.get(i));
if (type == null) {
return null;
}
dataType[i] = type;
}
return dataType;
}
@Override
public void handleParseComplete() {
// Response to parse
}
@Override
public void handleParameterDescription(PgParamDesc paramDesc) {
// Response to Describe
this.parameterDesc = paramDesc;
}
@Override
public void handleRowDescription(PgColumnDesc[] rowDesc) {
// Response to Describe
this.rowDesc = PgRowDesc.createBinary(rowDesc);
}
@Override
public void handleNoData() {
// Response to Describe
}
@Override
public void handleErrorResponse(ErrorResponse errorResponse) {
failure = errorResponse.toException();
}
@Override
public void handleReadyForQuery() {
result = new PgPreparedStatement(cmd.sql(), statement, this.parameterDesc, this.rowDesc, cmd.isManaged());
super.handleReadyForQuery();
}
}