src-main.org.awakefw.sql.jdbc.http.JdbcHttpStatementTransfer Maven / Gradle / Ivy

Go to download
/*
 * Awake File: Easy file upload & download through HTTP with Java
 * Awake SQL: Remote JDBC access through HTTP.                                    
 * Copyright (C) 2012, Kawan Softwares S.A.S.
 * (http://www.awakeframework.org). All rights reserved.                                
 *                                                                               
 * Awake File/SQL 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.            
 *                                                                               
 * Awake File/SQL 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 the Free Software           
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  
 * 02110-1301  USA
 *
 * Any modifications to this file must keep this entire header
 * intact.
 */

// 29/10/11 12:05 NDP : JdbcHttpStatementTransfer: remove unnecessary fields 
// 14/11/11 17:45 NDP : JdbcHttpStatementTransfer: add deleteFiles() at end of work
// 23/11/11 15:00 NDP : JdbcHttpStatementTransfer: encrypt statement holder list
// 30/11/11 19:35 NDP : JdbcHttpStatementTransfer: delete receiveFile if there is an Exception
// 30/11/11 19:35 NDP : JdbcHttpStatementTransfer: delete receiveFile if there is an Exception
// 13/03/12 14:05 NDP : JdbcHttpStatementTransfer: uses now StatementHolderFileList to store statements holders for batchs

package org.awakefw.sql.jdbc.http;

import java.io.File;
import java.io.IOException;
import java.sql.SQLException;
import java.util.List;
import java.util.Vector;

import org.apache.http.message.BasicNameValuePair;
import org.awakefw.commons.api.client.InvalidLoginException;
import org.awakefw.file.api.util.AwakeDebug;
import org.awakefw.file.http.HttpTransfer;
import org.awakefw.file.http.HttpTransferOne;
import org.awakefw.file.util.AwakeFileUtil;
import org.awakefw.file.util.KeepTempFilePolicyParms;
import org.awakefw.file.util.Tag;
import org.awakefw.file.util.parms.Parameter;
import org.awakefw.file.util.parms.ReturnCode;
import org.awakefw.sql.jdbc.ConnectionHttp;
import org.awakefw.sql.jdbc.util.StatementHolderFileList;
import org.awakefw.sql.jdbc.util.crypto.StatementHolderCrypto;
import org.awakefw.sql.json.ConnectionHolder;
import org.awakefw.sql.json.ConnectionHolderTransport;
import org.awakefw.sql.json.StatementHolder;
import org.awakefw.sql.json.StatementHolderTransport;
import org.awakefw.sql.util.SqlAction;
import org.awakefw.sql.util.TransportConverter;

/**
 * 
 * Class that execute all http transfer to the server, and gets back the results
 * 
 */
public class JdbcHttpStatementTransfer {
    /** Set to true to display/log debug info */
    private static boolean DEBUG = AwakeDebug
	    .isSet(JdbcHttpStatementTransfer.class);

    /** The Http Connection */
    private ConnectionHttp connectionHttp = null;

    /** The Authentication Token */
    private String authenticationToken = null;

    /**
     * Protected constructor, not callable
     */
    protected JdbcHttpStatementTransfer() {
    }

    /**
     * Constructor
     * 
     * @param connectionHttp
     *            the http Connection
     * @param authenticationToken
     *            the Http Connection Authentication Token
     */

    public JdbcHttpStatementTransfer(ConnectionHttp connectionHttp,
	    String authenticationToken) {
	this.connectionHttp = connectionHttp;
	this.authenticationToken = authenticationToken;
	
	debug(new java.util.Date() + " Uploading statement file...");
    }

    /**
     * Execute a remote sql execute query statement and get the the file of the
     * result
     * 
     * @param statementHolder
     *            the statement to execute on server
     * 
     * @return the received file
     */
    public File getFileFromExecuteQueryOnServer(StatementHolder statementHolder)
	    throws SQLException {
	HttpTransfer httpTransfer = null;

	httpTransfer = new HttpTransferOne(connectionHttp.getUrl(),
		connectionHttp.getHttpProxy(),
		connectionHttp.getHttpProtocolParameters(),
		connectionHttp.getAwakeProgressManager());

	// Prepare the request parameters
	List requestParams = new Vector();
	requestParams.add(new BasicNameValuePair(Parameter.ACTION,
		SqlAction.ACTION_SQL_STATEMENT));
	requestParams.add(new BasicNameValuePair(Parameter.LOGIN,
		connectionHttp.getUsername()));
	requestParams.add(new BasicNameValuePair(Parameter.TOKEN,
		authenticationToken));

	// Store the Connections commit mode / read only mode / transaction
	// isolation
	ConnectionHolder connectionHolder = new ConnectionHolder(connectionHttp);
	String jsonString = ConnectionHolderTransport.toJson(connectionHolder);
	requestParams.add(new BasicNameValuePair(SqlAction.CONNECTION_HOLDER,
		jsonString));

	// Execute the statement waiting
	List statementHolderList = new Vector();
	statementHolderList.add(statementHolder); // Only one StatementHolder

	StatementHolderCrypto.encrypt(statementHolderList, connectionHttp);	
	
	jsonString = StatementHolderTransport.toJson(statementHolderList);
	requestParams.add(new BasicNameValuePair(SqlAction.STATEMENT_HOLDER,
		jsonString));

	httpTransfer.setReceiveInFile(true); // To say we get the result into a
					     // file
	File receiveFile = null;

	try {
	    httpTransfer.send(requestParams);

	    receiveFile = httpTransfer.getReceiveFile();
	    String receive = AwakeFileUtil.getFirstLineOfFile(receiveFile);
	    debug("httpTransfer.getReceiveFile(): " + receiveFile + ":");

	    if (receive.startsWith(ReturnCode.INVALID_LOGIN_OR_PASSWORD)) {

		if (!DEBUG && !KeepTempFilePolicyParms.KEEP_TEMP_FILE) {
		    receiveFile.delete();
		}

		throw new InvalidLoginException();
	    }

	} catch (Exception e) {
	    JdbcHttpTransferUtil.wrapExceptionAsSQLException(e);
	}

	// We are done: return the received string
	return receiveFile;

    }

    /**
     * Execute a remote sql execute update statement and get the string
     * 
     * @param statementHolderFileList
     *            the list of statement + parameters to execute
     * @return the result of the execute update
     */
    public String getStringFromExecuteUpdateListOnServer(
	    StatementHolderFileList statementHolderFileList ) throws SQLException {
	
	// First, we close the file!
	statementHolderFileList.close();
	
	if (statementHolderFileList.isEmpty()) {
	    String message = Tag.AWAKE_PRODUCT_FAIL
		    + "Nothing to execute: statementHolderList is empty!";
	    throw new SQLException(message, new IOException(message));
	}
	
	HttpTransfer httpTransfer = null;
	//debug("this.connectionHttp.getHost()    : " + connectionHttp.getUrl());

	// Launch the Servlet
	httpTransfer = new HttpTransferOne(connectionHttp.getUrl(),
		connectionHttp.getHttpProxy(),
		connectionHttp.getHttpProtocolParameters(),
		connectionHttp.getAwakeProgressManager());

	List requestParams = new Vector();
	requestParams.add(new BasicNameValuePair(Parameter.ACTION,
		SqlAction.ACTION_SQL_STATEMENT));
	requestParams.add(new BasicNameValuePair(Parameter.LOGIN,
		connectionHttp.getUsername()));
	requestParams.add(new BasicNameValuePair(Parameter.TOKEN,
		this.authenticationToken));

	// Store the Connections commit mode / read only mode / transaction
	// isolation
	ConnectionHolder connectionHolder = new ConnectionHolder(
		this.connectionHttp);
	String jsonString = ConnectionHolderTransport.toJson(connectionHolder);
	requestParams.add(new BasicNameValuePair(SqlAction.CONNECTION_HOLDER,
		jsonString));

	// Execute all the statements waiting
	if (statementHolderFileList.size() < connectionHttp
		.getMaxStatementsForMemoryTransport()) {	        
	    
	    jsonString = StatementHolderTransport.toJson(statementHolderFileList.getStatementHolderList());
	    requestParams.add(new BasicNameValuePair(
		    SqlAction.STATEMENT_HOLDER, jsonString));
	    
	    // Delete the associated file
	    statementHolderFileList.delete();
	    
	    try {
		httpTransfer.send(requestParams);
	    } catch (Exception e) {
		JdbcHttpTransferUtil.wrapExceptionAsSQLException(e);
	    }
		
	} else {
	    //debug("StatementHolderFileList.size() >= maxStatementsForMemoryTransport)");

	    JdbcHttpFileTransfer jdbcHttpFileTransfer = new JdbcHttpFileTransfer(
		    this.connectionHttp);	    
	    
	    jdbcHttpFileTransfer.uploadFile(
		    statementHolderFileList.getLocalFile(),
		    statementHolderFileList.getRemoteFileName());

	    String statementHolderParam = TransportConverter.AWAKE_STREAM_FILE
		    + statementHolderFileList.getRemoteFileName();
	    requestParams.add(new BasicNameValuePair(
		    SqlAction.STATEMENT_HOLDER, statementHolderParam));
	    
	    try {
		httpTransfer.send(requestParams);
	    } catch (Exception e) {
		JdbcHttpTransferUtil.wrapExceptionAsSQLException(e);
	    }
	}

	debug(new java.util.Date() + " Upload and processing done...");
	
	String receive = httpTransfer.recv();

	debug("httpTransfer.recv(): " + receive + ":");

	if (receive.startsWith(ReturnCode.INVALID_LOGIN_OR_PASSWORD)) {
	    throw new SQLException(new InvalidLoginException());
	}

	// We are done: return the received string
	return receive;

    }

    
    /**
     * Debug tool
     * 
     * @param s
     */

    // @SuppressWarnings("unused")
    private void debug(String s) {
	if (DEBUG) {
	    //AwakeLogger.log(s);
	    System.out.println(s);
	}
    }

}