src/java/org/apache/juddi/datastore/jdbc/TModelCategoryTable.java - juddi - Git at Google

 /*
  * 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.KeyedReference;
 import org.apache.juddi.datatype.OverviewDoc;
 import org.apache.juddi.datatype.tmodel.TModel;
 import org.apache.juddi.util.Config;
 import org.apache.juddi.util.jdbc.ConnectionManager;
 import org.apache.juddi.util.jdbc.Transaction;
 import org.apache.juddi.uuidgen.UUIDGen;
 import org.apache.juddi.uuidgen.UUIDGenFactory;

 /**
  * @author Steve Viens (sviens@apache.org)
  */
 class TModelCategoryTable
 {
   // private reference to the jUDDI logger
   private static Log log = LogFactory.getLog(TModelCategoryTable.class);

   static String insertSQL = null;
   static String selectSQL = null;
   static String deleteSQL = null;

   static {
     // buffer used to build SQL statements
     StringBuffer sql = null;

     // build insertSQL
     sql = new StringBuffer(150);
     sql.append("INSERT INTO TMODEL_CATEGORY (");
     sql.append("TMODEL_KEY,");
     sql.append("CATEGORY_ID,");
     sql.append("TMODEL_KEY_REF,");
     sql.append("KEY_NAME,");
     sql.append("KEY_VALUE) ");
     sql.append("VALUES (?,?,?,?,?)");
     insertSQL = sql.toString();

     // build selectSQL
     sql = new StringBuffer(200);
     sql.append("SELECT ");
     sql.append("TMODEL_KEY_REF,");
     sql.append("KEY_NAME,");
     sql.append("KEY_VALUE, ");
     sql.append("CATEGORY_ID ");
     sql.append("FROM TMODEL_CATEGORY ");
     sql.append("WHERE TMODEL_KEY=? ");
     sql.append("ORDER BY CATEGORY_ID");
     selectSQL = sql.toString();

     // build deleteSQL
     sql = new StringBuffer(100);
     sql.append("DELETE FROM TMODEL_CATEGORY ");
     sql.append("WHERE TMODEL_KEY=?");
     deleteSQL = sql.toString();
   }

   /**
    * Insert new row into the TMODEL_CATEGORY table.
    *
    * @param tModelKey String to the parent TModelEntity object.
    * @param keyRefs A Vector of KeyedReference instances to insert.
    * @param connection JDBC connection
    * @throws java.sql.SQLException
    */
   public static void insert(
     String tModelKey,
     Vector keyRefs,
     Connection connection)
     throws java.sql.SQLException
   {
     PreparedStatement statement = null;

     try
     {
       statement = connection.prepareStatement(insertSQL);
       statement.setString(1, tModelKey.toString());

       int listSize = keyRefs.size();
       for (int categoryID = 0; categoryID < listSize; categoryID++)
       {
         KeyedReference keyRef = (KeyedReference) keyRefs.elementAt(categoryID);

         // extract values to insert
         String tModelKeyValue = null;
         if (keyRef.getTModelKey() != null)
           tModelKeyValue = keyRef.getTModelKey().toString();

         // set the values
         statement.setInt(2, categoryID);
         statement.setString(3, tModelKeyValue);
         statement.setString(4, keyRef.getKeyName());
         statement.setString(5, keyRef.getKeyValue());

         log.debug(
           "insert into TMODEL_CATEGORY table:\n\n\t"
             + insertSQL
             + "\n\t BUSINESS_KEY="
             + tModelKey.toString()
             + "\n\t CATEGORY_ID="
             + categoryID
             + "\n\t TMODEL_KEY_REF="
             + tModelKeyValue
             + "\n\t KEY_NAME="
             + keyRef.getKeyName()
             + "\n\t KEY_VALUE="
             + keyRef.getKeyValue()
             + "\n");

         // insert!
         statement.executeUpdate();
       }
     }
     finally
     {
       try
       {
         statement.close();
       }
       catch (Exception e)
       { /* ignored */
       }
     }
   }

   /**
    * Select all rows from the TMODEL_CATEGORY table for a given TModelKey.
    *
    * @param  tModelKey String
    * @param  connection JDBC connection
    * @throws java.sql.SQLException
    */
   public static Vector select(String tModelKey, Connection connection)
     throws java.sql.SQLException
   {
     Vector keyRefList = new Vector();
     PreparedStatement statement = null;
     ResultSet resultSet = null;

     try
     {
       // create a statement to query with
       statement = connection.prepareStatement(selectSQL);
       statement.setString(1, tModelKey.toString());

       log.debug(
         "select from TMODEL_CATEGORY table:\n\n\t"
           + selectSQL
           + "\n\t TMODEL_KEY="
           + tModelKey.toString()
           + "\n");

       // execute the statement
       resultSet = statement.executeQuery();
       while (resultSet.next())
       {
         KeyedReference keyRef = new KeyedReference();
         keyRef.setTModelKey(resultSet.getString(1));//("TMODEL_KEY_REF"));
         keyRef.setKeyName(resultSet.getString(2));//("KEY_NAME"));
         keyRef.setKeyValue(resultSet.getString(3));//("KEY_VALUE"));
         keyRefList.add(keyRef);
       }

       return keyRefList;
     }
     finally
     {
       try
       {
         resultSet.close();
         statement.close();
       }
       catch (Exception e)
       { /* ignored */
       }
     }
   }

   /**
    * Delete multiple rows from the TMODEL_CATEGORY table that are assigned to the
    * TModelKey specified.
    *
    * @param  tModelKey String
    * @param  connection JDBC connection
    * @throws java.sql.SQLException
    */
   public static void delete(String tModelKey, Connection connection)
     throws java.sql.SQLException
   {
     PreparedStatement statement = null;

     try
     {
       // prepare the delete
       statement = connection.prepareStatement(deleteSQL);
       statement.setString(1, tModelKey.toString());

       log.debug(
         "delete from TMODEL_CATEGORY table:\n\n\t"
           + deleteSQL
           + "\n\t TMODEL_KEY="
           + tModelKey.toString()
           + "\n");

       // execute
       statement.executeUpdate();
     }
     finally
     {
       try
       {
         statement.close();
       }
       catch (Exception e)
       { /* ignored */
       }
     }
   }

   /***************************************************************************/
   /***************************** TEST DRIVER *********************************/
   /***************************************************************************/


   public static void main(String[] args)
     throws Exception
   {
     // make sure we're using a DBCP DataSource and
     // not trying to use JNDI to aquire one.
     Config.setStringProperty("juddi.useConnectionPool","true");

     Connection conn = null;
     try {
       conn = ConnectionManager.aquireConnection();
       test(conn);
     }
     finally {
       if (conn != null)
         conn.close();
     }
   }

   public static void test(Connection connection) throws Exception
   {
     Transaction txn = new Transaction();
     UUIDGen uuidgen = UUIDGenFactory.getUUIDGen();

     if (connection != null)
     {
       try
       {
         OverviewDoc overviewDoc = new OverviewDoc();
         overviewDoc.setOverviewURL(
           "http://www.steveviens.com/overviewdoc.html");

         String tModelKey = uuidgen.uuidgen();
         TModel tModel = new TModel();
         tModel.setTModelKey(tModelKey);
         tModel.setAuthorizedName("sviens");
         tModel.setOperator("WebServiceRegistry.com");
         tModel.setName("Tuscany Web Service Company");
         tModel.setOverviewDoc(overviewDoc);

         Vector keyRefs = new Vector();
         keyRefs.add(new KeyedReference(uuidgen.uuidgen(), "blah, blah, blah"));
         keyRefs.add(
           new KeyedReference(uuidgen.uuidgen(), "Yadda, Yadda, Yadda"));
         keyRefs.add(
           new KeyedReference(uuidgen.uuidgen(), "WhoobWhoobWhoobWhoob"));
         keyRefs.add(new KeyedReference(uuidgen.uuidgen(), "Haachachachacha"));

         String authorizedUserID = "sviens";

         // begin a new transaction
         txn.begin(connection);

         // insert a new TModel
         TModelTable.insert(tModel, authorizedUserID, connection);

         // insert a Collection of new Category KeyedReference objects
         TModelCategoryTable.insert(tModelKey, keyRefs, connection);

         // insert another new TModel
         tModel.setTModelKey(uuidgen.uuidgen());
         TModelTable.insert(tModel, authorizedUserID, connection);

         // insert another Collection of new Category KeyedReference objects
         TModelCategoryTable.insert(tModel.getTModelKey(), keyRefs, connection);

         // select a Collection of Category KeyedReference objects
         keyRefs = TModelCategoryTable.select(tModelKey, connection);

         // delete a Collection of Category KeyedReference objects
         TModelCategoryTable.delete(tModelKey, connection);

         // re-select a Collection of Category KeyedReference objects
         keyRefs = TModelCategoryTable.select(tModelKey, connection);

         // commit the transaction
         txn.commit();
       }
       catch (Exception ex)
       {
         try
         {
           txn.rollback();
         }
         catch (java.sql.SQLException sqlex)
         {
           sqlex.printStackTrace();
         }
         throw ex;
       }
     }
   }
 }
	/*
	* 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.KeyedReference;
	import org.apache.juddi.datatype.OverviewDoc;
	import org.apache.juddi.datatype.tmodel.TModel;
	import org.apache.juddi.util.Config;
	import org.apache.juddi.util.jdbc.ConnectionManager;
	import org.apache.juddi.util.jdbc.Transaction;
	import org.apache.juddi.uuidgen.UUIDGen;
	import org.apache.juddi.uuidgen.UUIDGenFactory;

	/**
	* @author Steve Viens (sviens@apache.org)
	*/
	class TModelCategoryTable
	{
	// private reference to the jUDDI logger
	private static Log log = LogFactory.getLog(TModelCategoryTable.class);

	static String insertSQL = null;
	static String selectSQL = null;
	static String deleteSQL = null;

	static {
	// buffer used to build SQL statements
	StringBuffer sql = null;

	// build insertSQL
	sql = new StringBuffer(150);
	sql.append("INSERT INTO TMODEL_CATEGORY (");
	sql.append("TMODEL_KEY,");
	sql.append("CATEGORY_ID,");
	sql.append("TMODEL_KEY_REF,");
	sql.append("KEY_NAME,");
	sql.append("KEY_VALUE) ");
	sql.append("VALUES (?,?,?,?,?)");
	insertSQL = sql.toString();

	// build selectSQL
	sql = new StringBuffer(200);
	sql.append("SELECT ");
	sql.append("TMODEL_KEY_REF,");
	sql.append("KEY_NAME,");
	sql.append("KEY_VALUE, ");
	sql.append("CATEGORY_ID ");
	sql.append("FROM TMODEL_CATEGORY ");
	sql.append("WHERE TMODEL_KEY=? ");
	sql.append("ORDER BY CATEGORY_ID");
	selectSQL = sql.toString();

	// build deleteSQL
	sql = new StringBuffer(100);
	sql.append("DELETE FROM TMODEL_CATEGORY ");
	sql.append("WHERE TMODEL_KEY=?");
	deleteSQL = sql.toString();
	}

	/**
	* Insert new row into the TMODEL_CATEGORY table.
	*
	* @param tModelKey String to the parent TModelEntity object.
	* @param keyRefs A Vector of KeyedReference instances to insert.
	* @param connection JDBC connection
	* @throws java.sql.SQLException
	*/
	public static void insert(
	String tModelKey,
	Vector keyRefs,
	Connection connection)
	throws java.sql.SQLException
	{
	PreparedStatement statement = null;

	try
	{
	statement = connection.prepareStatement(insertSQL);
	statement.setString(1, tModelKey.toString());

	int listSize = keyRefs.size();
	for (int categoryID = 0; categoryID < listSize; categoryID++)
	{
	KeyedReference keyRef = (KeyedReference) keyRefs.elementAt(categoryID);

	// extract values to insert
	String tModelKeyValue = null;
	if (keyRef.getTModelKey() != null)
	tModelKeyValue = keyRef.getTModelKey().toString();

	// set the values
	statement.setInt(2, categoryID);
	statement.setString(3, tModelKeyValue);
	statement.setString(4, keyRef.getKeyName());
	statement.setString(5, keyRef.getKeyValue());

	log.debug(
	"insert into TMODEL_CATEGORY table:\n\n\t"
	+ insertSQL
	+ "\n\t BUSINESS_KEY="
	+ tModelKey.toString()
	+ "\n\t CATEGORY_ID="
	+ categoryID
	+ "\n\t TMODEL_KEY_REF="
	+ tModelKeyValue
	+ "\n\t KEY_NAME="
	+ keyRef.getKeyName()
	+ "\n\t KEY_VALUE="
	+ keyRef.getKeyValue()
	+ "\n");

	// insert!
	statement.executeUpdate();
	}
	}
	finally
	{
	try
	{
	statement.close();
	}
	catch (Exception e)
	{ /* ignored */
	}
	}
	}

	/**
	* Select all rows from the TMODEL_CATEGORY table for a given TModelKey.
	*
	* @param tModelKey String
	* @param connection JDBC connection
	* @throws java.sql.SQLException
	*/
	public static Vector select(String tModelKey, Connection connection)
	throws java.sql.SQLException
	{
	Vector keyRefList = new Vector();
	PreparedStatement statement = null;
	ResultSet resultSet = null;

	try
	{
	// create a statement to query with
	statement = connection.prepareStatement(selectSQL);
	statement.setString(1, tModelKey.toString());

	log.debug(
	"select from TMODEL_CATEGORY table:\n\n\t"
	+ selectSQL
	+ "\n\t TMODEL_KEY="
	+ tModelKey.toString()
	+ "\n");

	// execute the statement
	resultSet = statement.executeQuery();
	while (resultSet.next())
	{
	KeyedReference keyRef = new KeyedReference();
	keyRef.setTModelKey(resultSet.getString(1));//("TMODEL_KEY_REF"));
	keyRef.setKeyName(resultSet.getString(2));//("KEY_NAME"));
	keyRef.setKeyValue(resultSet.getString(3));//("KEY_VALUE"));
	keyRefList.add(keyRef);
	}

	return keyRefList;
	}
	finally
	{
	try
	{
	resultSet.close();
	statement.close();
	}
	catch (Exception e)
	{ /* ignored */
	}
	}
	}

	/**
	* Delete multiple rows from the TMODEL_CATEGORY table that are assigned to the
	* TModelKey specified.
	*
	* @param tModelKey String
	* @param connection JDBC connection
	* @throws java.sql.SQLException
	*/
	public static void delete(String tModelKey, Connection connection)
	throws java.sql.SQLException
	{
	PreparedStatement statement = null;

	try
	{
	// prepare the delete
	statement = connection.prepareStatement(deleteSQL);
	statement.setString(1, tModelKey.toString());

	log.debug(
	"delete from TMODEL_CATEGORY table:\n\n\t"
	+ deleteSQL
	+ "\n\t TMODEL_KEY="
	+ tModelKey.toString()
	+ "\n");

	// execute
	statement.executeUpdate();
	}
	finally
	{
	try
	{
	statement.close();
	}
	catch (Exception e)
	{ /* ignored */
	}
	}
	}

	/***************************************************************************/
	/*************************** TEST DRIVER *******************************/
	/***************************************************************************/


	public static void main(String[] args)
	throws Exception
	{
	// make sure we're using a DBCP DataSource and
	// not trying to use JNDI to aquire one.
	Config.setStringProperty("juddi.useConnectionPool","true");

	Connection conn = null;
	try {
	conn = ConnectionManager.aquireConnection();
	test(conn);
	}
	finally {
	if (conn != null)
	conn.close();
	}
	}

	public static void test(Connection connection) throws Exception
	{
	Transaction txn = new Transaction();
	UUIDGen uuidgen = UUIDGenFactory.getUUIDGen();

	if (connection != null)
	{
	try
	{
	OverviewDoc overviewDoc = new OverviewDoc();
	overviewDoc.setOverviewURL(
	"http://www.steveviens.com/overviewdoc.html");

	String tModelKey = uuidgen.uuidgen();
	TModel tModel = new TModel();
	tModel.setTModelKey(tModelKey);
	tModel.setAuthorizedName("sviens");
	tModel.setOperator("WebServiceRegistry.com");
	tModel.setName("Tuscany Web Service Company");
	tModel.setOverviewDoc(overviewDoc);

	Vector keyRefs = new Vector();
	keyRefs.add(new KeyedReference(uuidgen.uuidgen(), "blah, blah, blah"));
	keyRefs.add(
	new KeyedReference(uuidgen.uuidgen(), "Yadda, Yadda, Yadda"));
	keyRefs.add(
	new KeyedReference(uuidgen.uuidgen(), "WhoobWhoobWhoobWhoob"));
	keyRefs.add(new KeyedReference(uuidgen.uuidgen(), "Haachachachacha"));

	String authorizedUserID = "sviens";

	// begin a new transaction
	txn.begin(connection);

	// insert a new TModel
	TModelTable.insert(tModel, authorizedUserID, connection);

	// insert a Collection of new Category KeyedReference objects
	TModelCategoryTable.insert(tModelKey, keyRefs, connection);

	// insert another new TModel
	tModel.setTModelKey(uuidgen.uuidgen());
	TModelTable.insert(tModel, authorizedUserID, connection);

	// insert another Collection of new Category KeyedReference objects
	TModelCategoryTable.insert(tModel.getTModelKey(), keyRefs, connection);

	// select a Collection of Category KeyedReference objects
	keyRefs = TModelCategoryTable.select(tModelKey, connection);

	// delete a Collection of Category KeyedReference objects
	TModelCategoryTable.delete(tModelKey, connection);

	// re-select a Collection of Category KeyedReference objects
	keyRefs = TModelCategoryTable.select(tModelKey, connection);

	// commit the transaction
	txn.commit();
	}
	catch (Exception ex)
	{
	try
	{
	txn.rollback();
	}
	catch (java.sql.SQLException sqlex)
	{
	sqlex.printStackTrace();
	}
	throw ex;
	}
	}
	}
	}