/*
 * Jython Database Specification API 2.0
 *
 * $Id: MySQLDataHandler.java,v 1.2 2001/12/04 03:09:45 bzimmer Exp $
 *
 * Copyright (c) 2001 brian zimmer <bzimmer@ziclix.com>
 *
 */
package com.ziclix.python.sql.handler;

import java.io.*;
import java.sql.*;
import org.python.core.*;
import com.ziclix.python.sql.*;

/**
 * MySQL specific data handling.
 *
 * @author brian zimmer
 * @author last revised by $Author: bzimmer $
 * @version $Revision: 1.2 $
 */
public class MySQLDataHandler extends FilterDataHandler {

        /**
         * Decorator for handling MySql specific issues.
         *
         * @param datahandler the delegate DataHandler
         */
        public MySQLDataHandler(DataHandler datahandler) {
                super(datahandler);
        }

        /**
         * Returns the last insert id for the statement.
         *
         * @param Statement stmt
         *
         * @return PyObject
         *
         * @throws SQLException
         *
         */
        public PyObject getRowId(Statement stmt) throws SQLException {

                if (stmt instanceof com.mysql.jdbc.Statement) {
                        return Py.newInteger(((com.mysql.jdbc.Statement)stmt).getLastInsertID());
                }

                return super.getRowId(stmt);
        }

        /**
         * Handle LONGVARCHAR.
         */
        public void setJDBCObject(PreparedStatement stmt, int index, PyObject object, int type) throws SQLException {

                if (DataHandler.checkNull(stmt, index, object, type)) {
                        return;
                }

                switch (type) {

                        case Types.LONGVARCHAR :
                                if (object instanceof PyFile) {
                                        object = ((PyFile)object).read();
                                }

                                String varchar = (String)object.__tojava__(String.class);
                                InputStream stream = new ByteArrayInputStream(varchar.getBytes());

                                stream = new BufferedInputStream(stream);

                                stmt.setAsciiStream(index, stream, varchar.length());
                                break;

                        default :
                                super.setJDBCObject(stmt, index, object, type);
                                break;
                }
        }
}