org.mariadb.jdbc.internal.com.send.ComStmtExecute Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mariadb-java-client Show documentation
Show all versions of mariadb-java-client Show documentation
JDBC driver for MariaDB and MySQL
/*
*
* MariaDB Client for Java
*
* Copyright (c) 2012-2014 Monty Program Ab.
* Copyright (c) 2015-2020 MariaDB Corporation Ab.
*
* This library is free software; you can redistribute it and/or modify it under
* the terms of the GNU Lesser General Public License as published by the Free
* Software Foundation; either version 2.1 of the License, or (at your option)
* any later version.
*
* This library is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
* for more details.
*
* You should have received a copy of the GNU Lesser General Public License along
* with this library; if not, write to Monty Program Ab [email protected].
*
* This particular MariaDB Client for Java file is work
* derived from a Drizzle-JDBC. Drizzle-JDBC file which is covered by subject to
* the following copyright and notice provisions:
*
* Copyright (c) 2009-2011, Marcus Eriksson
*
* Redistribution and use in source and binary forms, with or without modification,
* are permitted provided that the following conditions are met:
* Redistributions of source code must retain the above copyright notice, this list
* of conditions and the following disclaimer.
*
* Redistributions in binary form must reproduce the above copyright notice, this
* list of conditions and the following disclaimer in the documentation and/or
* other materials provided with the distribution.
*
* Neither the name of the driver nor the names of its contributors may not be
* used to endorse or promote products derived from this software without specific
* prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
* IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
* WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
* OF SUCH DAMAGE.
*
*/
package org.mariadb.jdbc.internal.com.send;
import java.io.IOException;
import org.mariadb.jdbc.internal.ColumnType;
import org.mariadb.jdbc.internal.com.Packet;
import org.mariadb.jdbc.internal.com.send.parameters.ParameterHolder;
import org.mariadb.jdbc.internal.io.output.PacketOutputStream;
public class ComStmtExecute {
/**
* Write COM_STMT_EXECUTE sub-command to output buffer.
*
* @param statementId prepareResult object received after preparation.
* @param parameters parameters
* @param parameterCount parameters number
* @param parameterTypeHeader parameters header1
* @param pos outputStream
* @param cursorFlag cursor flag. Possible values :
*
* - CURSOR_TYPE_NO_CURSOR = fetch all
*
- CURSOR_TYPE_READ_ONLY = fetch by bunch
*
- CURSOR_TYPE_FOR_UPDATE = fetch by bunch with lock ?
*
- CURSOR_TYPE_SCROLLABLE = //reserved, but not working
*
*
* @throws IOException if a connection error occur
*/
public static void writeCmd(
final int statementId,
final ParameterHolder[] parameters,
final int parameterCount,
ColumnType[] parameterTypeHeader,
final PacketOutputStream pos,
final byte cursorFlag)
throws IOException {
pos.write(Packet.COM_STMT_EXECUTE);
pos.writeInt(statementId);
pos.write(cursorFlag);
pos.writeInt(1); // Iteration pos
// create null bitmap
if (parameterCount > 0) {
int nullCount = (parameterCount + 7) / 8;
byte[] nullBitsBuffer = new byte[nullCount];
for (int i = 0; i < parameterCount; i++) {
if (parameters[i].isNullData()) {
nullBitsBuffer[i / 8] |= (1 << (i % 8));
}
}
pos.write(nullBitsBuffer, 0, nullCount);
// check if parameters type (using setXXX) have change since previous request, and resend new
// header type if so
boolean mustSendHeaderType = false;
if (parameterTypeHeader[0] == null) {
mustSendHeaderType = true;
} else {
for (int i = 0; i < parameterCount; i++) {
if (!parameterTypeHeader[i].equals(parameters[i].getColumnType())) {
mustSendHeaderType = true;
break;
}
}
}
if (mustSendHeaderType) {
pos.write((byte) 0x01);
// Store types of parameters in first in first package that is sent to the server.
for (int i = 0; i < parameterCount; i++) {
parameterTypeHeader[i] = parameters[i].getColumnType();
pos.writeShort(parameterTypeHeader[i].getType());
}
} else {
pos.write((byte) 0x00);
}
}
for (int i = 0; i < parameterCount; i++) {
ParameterHolder holder = parameters[i];
if (!holder.isNullData() && !holder.canBeLongData()) {
holder.writeBinary(pos);
}
}
}
/**
* Send a prepare statement binary stream.
*
* @param pos database socket
* @param statementId prepareResult object received after preparation.
* @param parameters parameters
* @param parameterCount parameters number
* @param parameterTypeHeader parameters header
* @param cursorFlag cursor flag. Possible values :
*
* - CURSOR_TYPE_NO_CURSOR = fetch all
*
- CURSOR_TYPE_READ_ONLY = fetch by bunch
*
- CURSOR_TYPE_FOR_UPDATE = fetch by bunch with lock ?
*
- CURSOR_TYPE_SCROLLABLE = //reserved, but not working
*
*
* @throws IOException if a connection error occur
*/
public static void send(
final PacketOutputStream pos,
final int statementId,
final ParameterHolder[] parameters,
final int parameterCount,
ColumnType[] parameterTypeHeader,
byte cursorFlag)
throws IOException {
pos.startPacket(0);
writeCmd(statementId, parameters, parameterCount, parameterTypeHeader, pos, cursorFlag);
pos.flush();
}
}