/*
    * Copyright 2001-2004 The Apache Software Foundation.
    * 
    * Licensed under the Apache 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.apache.org/licenses/LICENSE-2.0
    * 
   * 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.apache.juddi.datastore.jdbc;
  
  import java.sql.Connection;
  import java.sql.PreparedStatement;
  import java.sql.ResultSet;
  import java.util.Vector;
  
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import org.apache.juddi.datatype.TModelBag;
  import org.apache.juddi.datatype.request.FindQualifiers;
  import org.apache.juddi.registry.RegistryEngine;
  import org.apache.juddi.util.Config;
  import org.apache.juddi.util.jdbc.DynamicQuery;
  
  /***
   * @author Steve Viens (sviens@apache.org)
   */
  class FindBusinessByTModelKeyQuery
  {
    // private reference to the jUDDI logger
    private static Log log = LogFactory.getLog(FindBusinessByTModelKeyQuery.class);
  
    static String selectSQL;
    static String tablePrefix;
    static
    {
     tablePrefix = Config.getStringProperty(
          RegistryEngine.PROPNAME_TABLE_PREFIX,RegistryEngine.DEFAULT_TABLE_PREFIX);
      // build selectSQL
      StringBuffer sql = new StringBuffer(200);
      sql.append("SELECT B.BUSINESS_KEY,B.LAST_UPDATE ");
      sql.append("FROM ").append(tablePrefix).append("BUSINESS_ENTITY B,")
                         .append(tablePrefix).append("BUSINESS_SERVICE S,")
                         .append(tablePrefix).append("BINDING_TEMPLATE T,")
                         .append(tablePrefix).append("TMODEL_INSTANCE_INFO I ");
      selectSQL = sql.toString();
    }
  
    /***
     * Select ...
     *
     * @param tModelBag
     * @param keysIn
     * @param qualifiers
     * @param connection JDBC connection
     * @throws java.sql.SQLException
     */
    public static Vector select(TModelBag tModelBag,Vector keysIn,FindQualifiers qualifiers,Connection connection)
      throws java.sql.SQLException
    {
      // if there is a keysIn vector but it doesn't contain
      // any keys then the previous query has exhausted
      // all possibilities of a match so skip this call.
      if ((keysIn != null) && (keysIn.size() == 0))
        return keysIn;
  
      Vector keysOut = new Vector();
      PreparedStatement statement = null;
      ResultSet resultSet = null;
  
      // construct the SQL statement
      DynamicQuery sql = new DynamicQuery(selectSQL);
      appendWhere(sql,tModelBag,qualifiers);
      appendIn(sql,keysIn);
      appendOrderBy(sql,qualifiers);
  
      try
      {
        log.debug(sql.toString());
  
        statement = sql.buildPreparedStatement(connection);
        resultSet = statement.executeQuery();
  
        while (resultSet.next())
          keysOut.addElement(resultSet.getString(1));//("BUSINESS_KEY"));
  
        return keysOut;
      }
      finally
      {
        try {
          resultSet.close();
        }
       catch (Exception e)
       {
         log.warn("An Exception was encountered while attempting to close " +
           "the Find BusinessEntity ResultSet: "+e.getMessage(),e);
       }
 
       try {
         statement.close();
       }
       catch (Exception e)
       {
         log.warn("An Exception was encountered while attempting to close " +
           "the Find BusinessEntity Statement: "+e.getMessage(),e);
       }
     }
   }
 
   /***
    *
    */
   private static void appendWhere(DynamicQuery sql,TModelBag tModelBag,FindQualifiers qualifiers)
   {
     sql.append("WHERE I.BINDING_KEY = T.BINDING_KEY ");
     sql.append("AND T.SERVICE_KEY = S.SERVICE_KEY ");
     sql.append("AND S.BUSINESS_KEY = B.BUSINESS_KEY ");
 
     Vector keyVector = tModelBag.getTModelKeyVector();
 
     int vectorSize = keyVector.size();
     if (vectorSize > 0)
     {
       sql.append("AND (");
 
       for (int i=0; i<vectorSize; i++)
       {
         String key = (String) keyVector.elementAt(i);
 
         sql.append("I.TMODEL_KEY = ? ");
         sql.addValue(key);
 
         if (i+1 < vectorSize)
           sql.append(" OR ");
       }
 
       sql.append(") ");
     }
   }
 
   /***
    * Select ...
    *
    * @param tModelKey
    * @param keysIn
    * @param qualifiers
    * @param connection JDBC connection
    * @throws java.sql.SQLException
    */
   public static Vector select(String tModelKey,Vector keysIn,FindQualifiers qualifiers,Connection connection)
     throws java.sql.SQLException
   {
     // if there is a keysIn vector but it doesn't contain
     // any keys then the previous query has exhausted
     // all possibilities of a match so skip this call.
     if ((keysIn != null) && (keysIn.size() == 0))
       return keysIn;
 
     Vector keysOut = new Vector();
     PreparedStatement statement = null;
     ResultSet resultSet = null;
 
     // construct the SQL statement
     DynamicQuery sql = new DynamicQuery(selectSQL);
     appendWhere(sql,tModelKey,qualifiers);
     appendIn(sql,keysIn);
     appendOrderBy(sql,qualifiers);
 
     try
     {
       log.debug(sql.toString());
 
       statement = sql.buildPreparedStatement(connection);
       resultSet = statement.executeQuery();
 
       while (resultSet.next())
         keysOut.addElement(resultSet.getString(1));//("BUSINESS_KEY"));
 
       return keysOut;
     }
     finally
     {
       try {
         resultSet.close();
       }
       catch (Exception e)
       {
         log.warn("An Exception was encountered while attempting to close " +
           "the Find BusinessEntity ResultSet: "+e.getMessage(),e);
       }
 
       try {
         statement.close();
       }
       catch (Exception e)
       {
         log.warn("An Exception was encountered while attempting to close " +
           "the Find BusinessEntity Statement: "+e.getMessage(),e);
       }
     }
   }
 
   /***
    *
    */
   private static void appendWhere(DynamicQuery sql,String tModelKey,FindQualifiers qualifiers)
   {
     sql.append("WHERE I.BINDING_KEY = T.BINDING_KEY ");
     sql.append("AND T.SERVICE_KEY = S.SERVICE_KEY ");
     sql.append("AND S.BUSINESS_KEY = B.BUSINESS_KEY ");
 
     if ((tModelKey != null) && (tModelKey.trim().length() > 0))
     {
       sql.append("AND I.TMODEL_KEY = ? ");
       sql.addValue(tModelKey);
     }
   }
 
   /***
    * Utility method used to construct SQL "IN" statements such as
    * the following SQL example:
    *
    *   SELECT * FROM TABLE WHERE MONTH IN ('jan','feb','mar')
    *
    * @param sql StringBuffer to append the final results to
    * @param keysIn Vector of Strings used to construct the "IN" clause
    */
   private static void appendIn(DynamicQuery sql,Vector keysIn)
   {
     if (keysIn == null)
       return;
 
     sql.append("AND B.BUSINESS_KEY IN (");
 
     int keyCount = keysIn.size();
     for (int i=0; i<keyCount; i++)
     {
       String key = (String)keysIn.elementAt(i);
       sql.append("?");
       sql.addValue(key);
 
       if ((i+1) < keyCount)
         sql.append(",");
     }
 
     sql.append(") ");
   }
 
   /***
    *
    */
   private static void appendOrderBy(DynamicQuery sql,FindQualifiers qualifiers)
   {
     sql.append("ORDER BY ");
 
     if (qualifiers == null)
       sql.append("B.LAST_UPDATE DESC");
     else if (qualifiers.sortByDateAsc)
       sql.append("B.LAST_UPDATE ASC");
     else
       sql.append("B.LAST_UPDATE DESC");
   }
 }