/*

  * Copyright 2005-2009 The Kuali Foundation

  * 

  * 

  * Licensed under the Educational Community 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.opensource.org/licenses/ecl2.php

  * 

  * 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 org.kuali.rice.core.database.platform;

 import java.math.BigInteger;

 import java.sql.Connection;

 import java.sql.PreparedStatement;

 import java.sql.ResultSet;

 import java.sql.SQLException;

 import java.util.regex.Matcher;

 import java.util.regex.Pattern;

 import javax.persistence.EntityManager;

 import javax.persistence.EntityTransaction;

 import org.apache.ojb.broker.PersistenceBroker;

 import org.apache.ojb.broker.accesslayer.LookupException;

 import org.apache.ojb.broker.query.Criteria;

 public class MySQLDatabasePlatform extends ANSISqlDatabasePlatform {

         private static final Pattern APOS_PAT = Pattern.compile("'");

         private static final Pattern BSLASH_PAT = Pattern.compile(Matcher.quoteReplacement("\\"));

     public String getLockRouteHeaderQuerySQL(Long routeHeaderId, boolean wait) {

         return "SELECT DOC_HDR_ID FROM KREW_DOC_HDR_T WHERE DOC_HDR_ID=? FOR UPDATE";

     }

     public String getStrToDateFunction() {

         return "STR_TO_DATE";

     }

     public String getCurTimeFunction() {

         return "NOW()";

     }

     public void applyLimit(Integer limit, Criteria criteria) {

         if (limit != null) {

             criteria.addSql(" 1 LIMIT 0," + limit.intValue()); // 1 has to be there because the criteria is ANDed

         }

     }

     public Long getNextValSQL(String sequenceName,        PersistenceBroker persistenceBroker) {

                   PreparedStatement statement = null;

                   ResultSet resultSet = null;

                   try {

                           Connection connection = persistenceBroker.serviceConnectionManager().getConnection();

                           statement = connection.prepareStatement("INSERT INTO " + sequenceName + " VALUES (NULL);");

                           statement.executeUpdate();

                           statement = connection.prepareStatement("SELECT LAST_INSERT_ID()");

                           resultSet = statement.executeQuery();

                           if (!resultSet.next()) {

                                   throw new RuntimeException("Error retrieving next option id for action list from sequence.");

                           }

                           return new Long(resultSet.getLong(1));

                   } catch (SQLException e) {

                           throw new RuntimeException("Error retrieving next option id for action list from sequence.", e);

                   } catch (LookupException e) {

                           throw new RuntimeException("Error retrieving next option id for action list from sequence.", e);

                   } finally {

                           if (statement != null) {

                                   try {

                                           statement.close();

                                   } catch (SQLException e) {

                                   }

                           }

                           if (resultSet != null) {

                                   try {

                                           resultSet.close();

                                   } catch (SQLException e) {

                                   }

                           }

                   }

           }

     public Long getNextValSQL(String sequenceName, EntityManager entityManager) {

             EntityTransaction tx = entityManager.getTransaction();

             try {

                 tx.begin();                

             entityManager.createNativeQuery("INSERT INTO " + sequenceName + " VALUES (NULL);").executeUpdate();

             Long result = new Long(((BigInteger) entityManager.createNativeQuery("SELECT LAST_INSERT_ID()").getSingleResult()).longValue());                           

                 tx.commit();                

                 return result;

             }

             finally {

                 if ( tx.isActive() ) {

                     tx.rollback();

                 }

             }

     }

     public boolean isSITCacheSupported() {

             return false;

     }

     public String toString() {

         return "[MySQLDatabasePlatform]";

     }

     public String getSelectForUpdateSuffix(long waitMillis) {

         return "for update";

     }

     public String getDateFormatString(String dateFormatString) {

         String newString = "";

         if ("yyyy-mm-dd".equalsIgnoreCase(dateFormatString)) {

             newString = "'%Y-%m-%d'";

         }

         else if ("DD/MM/YYYY HH12:MI:SS PM".equalsIgnoreCase(dateFormatString)) {

             newString = "'%d/%m/%Y %r'";

         }

         return newString;

     }

     /**

      * Performs MySQL-specific escaping of String parameters.

      * 

      * @see org.kuali.rice.core.database.platform.DatabasePlatform#escapeString(java.lang.String)

      */

     public String escapeString(String sqlString) {

             return (sqlString != null) ? BSLASH_PAT.matcher(APOS_PAT.matcher(sqlString).replaceAll("''")).replaceAll(Matcher.quoteReplacement("\\\\")) : null;

     } 

 }