+
−/*
+
− * This Source Code Form is subject to the terms of the Mozilla Public
+
− * License, v. 2.0.  If a copy of the MPL was not distributed with this
+
− * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
− *
+
− * Copyright 1997 - July 2008 CWI, August 2008 - 2017 MonetDB B.V.
+
− */
+
−
+
−package nl.cwi.monetdb.mcl.protocol.oldmapi;
+
−
+
−import nl.cwi.monetdb.jdbc.MonetConnection;
+
−import nl.cwi.monetdb.mcl.connection.mapi.OldMapiSocket;
+
−import nl.cwi.monetdb.mcl.protocol.ProtocolException;
+
−import nl.cwi.monetdb.mcl.protocol.AbstractProtocol;
+
−import nl.cwi.monetdb.mcl.protocol.ServerResponses;
+
−import nl.cwi.monetdb.mcl.protocol.StarterHeaders;
+
−import nl.cwi.monetdb.mcl.responses.AutoCommitResponse;
+
−import nl.cwi.monetdb.mcl.responses.UpdateResponse;
+
−import nl.cwi.monetdb.mcl.responses.DataBlockResponse;
+
−import nl.cwi.monetdb.mcl.responses.ResultSetResponse;
+
−
+
−import java.io.IOException;
+
−import java.nio.CharBuffer;
+
−import java.text.SimpleDateFormat;
+
−import java.util.Map;
+
−
+
−/**
+
− * The JDBC abstract protocol implementation on a MAPI connection using the protocol version 9. The connection holds a
+
− * lineBuffer which will be reused during the whole connection for memory saving purposes. An additional tupleLineBuffer
+
− * is used to help parsing tuple lines from a BLOCK response.
+
− *
+
− * @author Pedro Ferreira
+
− */
+
−public class OldMapiProtocol extends AbstractProtocol {
+
−
+
−    /**
+
−     * The default size for the tuple lines' CharBuffer (it should be less than the OldMapiSocket BLOCK size).
+
−     */
+
−    private static final int TUPLE_LINE_BUFFER_DEFAULT_SIZE = 1024;
+
−
+
−    /** Format of a time string from the old MAPI connection */
+
−    final SimpleDateFormat timeParser = new SimpleDateFormat("HH:mm:ss");
+
−
+
−    /** Format of a timestamp string from the old MAPI connection */
+
−    final SimpleDateFormat timestampParser = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+
−
+
−    /**The current server response */
+
−    private int currentServerResponseHeader = ServerResponses.UNKNOWN;
+
−
+
−    /** The underlying MAPI socket connection */
+
−    private final OldMapiSocket socket;
+
−
+
−    /** The buffer used to parse server's responses */
+
−    CharBuffer lineBuffer;
+
−
+
−    /** A helper buffer used to parse tuple line responses */
+
−    CharBuffer tupleLineBuffer;
+
−
+
−    public OldMapiProtocol(OldMapiSocket socket) {
+
−        this.socket = socket;
+
−        this.lineBuffer = CharBuffer.wrap(new char[OldMapiSocket.FULL_BLOCK]);
+
−        this.tupleLineBuffer = CharBuffer.wrap(new char[TUPLE_LINE_BUFFER_DEFAULT_SIZE]);
+
−    }
+
−
+
−    /**
+
−     * Gets the underlying socket.
+
−     *
+
−     * @return The underlying socket
+
−     */
+
−    public OldMapiSocket getSocket() {
+
−        return socket;
+
−    }
+
−
+
−    /**
+
−     * Gets the current server response, obtained through the fetchNextResponseData method.
+
−     *
+
−     * @return The integer representation of {@link ServerResponses}
+
−     */
+
−    @Override
+
−    public int getCurrentServerResponse() {
+
−        return currentServerResponseHeader;
+
−    }
+
−
+
−    /**
+
−     * Reads up till the MonetDB prompt, indicating the server is ready for a new command.
+
−     *
+
−     * If there are errors present in the lines that are read, then they are put in one string and returned <b>after</b>
+
−     * the prompt has been found.
+
−     *
+
−     * @throws IOException if an IO exception occurs while talking to the server
+
−     */
+
−    @Override
+
−    public void waitUntilPrompt() throws IOException {
+
−        while(this.currentServerResponseHeader != ServerResponses.PROMPT) {
+
−            this.lineBuffer = this.socket.readLine(this.lineBuffer);
+
−            if(this.lineBuffer.limit() == 0) {
+
−                throw new IOException("Connection to server lost!");
+
−            }
+
−            this.currentServerResponseHeader = OldMapiServerResponseParser.ParseOldMapiServerResponse(this);
+
−            this.lineBuffer.position(0);
+
−            if (this.currentServerResponseHeader == ServerResponses.ERROR) {
+
−                this.lineBuffer.position(1);
+
−            }
+
−        }
+
−    }
+
−
+
−    /**
+
−     * Reads a line of text from the socket. A line is considered to be terminated by any one of a line feed ('\n').
+
−     *
+
−     * Warning: until the server properly prefixes all of its error messages with SQLSTATE codes, this method prefixes
+
−     * all errors it sees without sqlstate with the generic data exception code (22000).
+
−     *
+
−     * @throws IOException If an I/O error occurs
+
−     */
+
−    @Override
+
−    public void fetchNextResponseData() throws IOException {
+
−        this.lineBuffer = this.socket.readLine(this.lineBuffer);
+
−        if(this.lineBuffer.limit() == 0) {
+
−            throw new IOException("Connection to server lost!");
+
−        }
+
−        this.currentServerResponseHeader = OldMapiServerResponseParser.ParseOldMapiServerResponse(this);
+
−        if (this.currentServerResponseHeader == ServerResponses.ERROR && !this.lineBuffer.toString()
+
−                .matches("^[0-9A-Z]{5}!.+")) {
+
−            CharBuffer newbuffer = CharBuffer.wrap(new char[this.lineBuffer.capacity() + 7]);
+
−            newbuffer.put("!22000!");
+
−            newbuffer.put(this.lineBuffer.array());
+
−            newbuffer.flip();
+
−            this.lineBuffer = newbuffer;
+
−        }
+
−        this.lineBuffer.position(1);
+
−    }
+
−
+
−    /**
+
−     * Gets the next starter header of a server response.
+
−     *
+
−     * @return The integer representation of {@link StarterHeaders}
+
−     */
+
−    @Override
+
−    public int getNextStarterHeader() {
+
−        return OldMapiStartOfHeaderParser.GetNextStartHeaderOnOldMapi(this);
+
−    }
+
−
+
−    /**
+
−     * Gets the next ResultSet response from the server, belonging to a ResponseList.
+
−     *
+
−     * @param con The current MonetDB's JDBC connection
+
−     * @param list The Response List this result set will belong to
+
−     * @param seqnr The sequence number of this result set on the Response List
+
−     * @return The ResultSet instance
+
−     * @throws ProtocolException If an error in the underlying connection happened.
+
−     */
+
−    @Override
+
−    public ResultSetResponse getNextResultSetResponse(MonetConnection con, MonetConnection.ResponseList list, int seqnr)
+
−            throws ProtocolException {
+
−        int id = OldMapiStartOfHeaderParser.GetNextResponseDataAsInt(this); //The order cannot be switched!!
+
−        int tuplecount = OldMapiStartOfHeaderParser.GetNextResponseDataAsInt(this);
+
−        int columncount = OldMapiStartOfHeaderParser.GetNextResponseDataAsInt(this);
+
−        int rowcount = OldMapiStartOfHeaderParser.GetNextResponseDataAsInt(this);
+
−        return new ResultSetResponse(con, list, id, seqnr, rowcount, tuplecount, columncount);
+
−    }
+
−
+
−    /**
+
−     * Gets the next UpdateResponse response from the server.
+
−     *
+
−     * @return The UpdateResponse instance
+
−     * @throws ProtocolException If an error in the underlying connection happened.
+
−     */
+
−    @Override
+
−    public UpdateResponse getNextUpdateResponse() throws ProtocolException {
+
−        int count = OldMapiStartOfHeaderParser.GetNextResponseDataAsInt(this); //The order cannot be switched!!
+
−        int lastId = OldMapiStartOfHeaderParser.GetNextResponseDataAsInt(this);
+
−        return new UpdateResponse(lastId, count);
+
−    }
+
−
+
−    /**
+
−     * Gets the next AutoCommitResponse response from the server.
+
−     *
+
−     * @return The AutoCommitResponse instance
+
−     * @throws ProtocolException If an error in the underlying connection happened.
+
−     */
+
−    @Override
+
−    public AutoCommitResponse getNextAutoCommitResponse() throws ProtocolException {
+
−        boolean ac = this.lineBuffer.get() == 't';
+
−        return new AutoCommitResponse(ac);
+
−    }
+
−
+
−    /**
+
−     * Gets the next DataBlockResponse response from the server, belonging to a ResultSetResponse
+
−     *
+
−     * @param rsresponses A map of ResultSetResponse, in which this Block will belong to one of them, by checking its id
+
−     *                    against the keys of the Map.
+
−     * @return The DataBlockResponse instance
+
−     * @throws ProtocolException If an error in the underlying connection happened.
+
−     */
+
−    @Override
+
−    public DataBlockResponse getNextDatablockResponse(Map<Integer, ResultSetResponse> rsresponses)
+
−            throws ProtocolException {
+
−        int id = OldMapiStartOfHeaderParser.GetNextResponseDataAsInt(this); //The order cannot be switched!!
+
−        OldMapiStartOfHeaderParser.GetNextResponseDataAsInt(this); //column count
+
−        int rowcount = OldMapiStartOfHeaderParser.GetNextResponseDataAsInt(this);
+
−        int offset = OldMapiStartOfHeaderParser.GetNextResponseDataAsInt(this);
+
−
+
−        ResultSetResponse rs = rsresponses.get(id);
+
−        if (rs == null) {
+
−            return null;
+
−        }
+
−        return rs.addDataBlockResponse(offset, rowcount);
+
−    }
+
−
+
−    /**
+
−     * Gets the next Table Header for a ResultSetResponse. More than one of the parameter arrays can be filled at once.
+
−     *
+
−     * @param columnNames The column names array
+
−     * @param columnLengths The column lengths array
+
−     * @param types The columns SQL names array
+
−     * @param tableNames The columns schemas and names in format schema.table
+
−     * @return A TableResultHeaders integer representation, representing which of the fields was filled
+
−     * @throws ProtocolException If an error in the underlying connection happened.
+
−     */
+
−    @Override
+
−    public int getNextTableHeader(String[] columnNames, int[] columnLengths, String[] types, String[] tableNames)
+
−            throws ProtocolException {
+
−        return OldMapiTableHeaderParser.GetNextTableHeader(this.lineBuffer, columnNames, columnLengths, types,
+
−                tableNames);
+
−    }
+
−
+
−    /**
+
−     * Retrieves the next values in a DataBlockResponse from the underlying connection, starting at a specific line
+
−     * number.
+
−     *
+
−     * @param firstLineNumber The first line number in the response to retrieve
+
−     * @param typesMap The JDBC types mapping array for every column in the ResultSetResponse of the DataBlock
+
−     * @param values An array of columns to fill the values
+
−     * @return The number of lines parsed from the underlying connection
+
−     * @throws ProtocolException If an error in the underlying connection happened.
+
−     */
+
−    @Override
+
−    public int parseTupleLines(int firstLineNumber, int[] typesMap, Object[] values) throws ProtocolException {
+
−        OldMapiTupleLineParser.OldMapiParseTupleLine(this, firstLineNumber, typesMap, values);
+
−        return firstLineNumber;
+
−    }
+
−
+
−    /**
+
−     * Gets the remaining response line from the underlying connection as a Java String. This method is mostly used to
+
−     * retrieve error Strings, when they are detected while parsing a response line.
+
−     *
+
−     * @param startIndex The first index in the response line to retrieve the String
+
−     * @return The String representation of the line starting at the provided index
+
−     */
+
−    @Override
+
−    public String getRemainingStringLine(int startIndex) {
+
−        if(this.lineBuffer.limit() > startIndex) {
+
−            if(this.lineBuffer.array()[startIndex] == '!') {
+
−                startIndex++;
+
−            }
+
−            return new String(this.lineBuffer.array(), startIndex, this.lineBuffer.limit() - startIndex);
+
−        } else {
+
−            return null;
+
−        }
+
−    }
+
−
+
−    /**
+
−     * Writes a user query to the server, while providing the respective prefixes and suffixes depending on the current
+
−     * language and connection used.
+
−     *
+
−     * @param prefix The prefix to append at the beginning of the query string
+
−     * @param query The user query to submit to the server
+
−     * @param suffix The suffix to append at the end of the query string
+
−     * @throws IOException If an error in the underlying connection happened.
+
−     */
+
−    @Override
+
−    public void writeNextQuery(String prefix, String query, String suffix) throws IOException {
+
−        this.socket.writeNextLine(prefix, query, suffix);
+
−        // reset reader state, last line isn't valid any more now
+
−        this.currentServerResponseHeader = ServerResponses.UNKNOWN;
+
−    }
+
−}