import java.sql.*;
import java.util.Iterator;
import java.util.Map;
import java.util.Collection;
import javax.security.auth.Subject;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
 
import com.adeptia.indigo.security.AuthUtil;
import com.adeptia.indigo.services.transport.connector.DatabaseConnectionInfo;
import com.adeptia.indigo.storage.Entity;
import com.adeptia.indigo.storage.EntityManager;
import com.adeptia.indigo.storage.EntityManagerFactory;
import com.adeptia.indigo.storage.StorageException;
import com.adeptia.indigo.utils.IndigoException;
import com.adeptia.indigo.utils.JdbcUtils;
import com.adeptia.indigo.utils.HandleJTADBConnection;


//Logic to clear connection pool
Map conMap = HandleJTADBConnection.getConnectionMap();
conMap.clear();
 
ArrayList tableName = context.get("Context_Variable");//Write the context variable here in which you are storing the table name's.

for(int i = 0;i<tableName.size();i++){
String Query="TRUNCATE TABLE "+((String)tableName.get(i)).trim() +";"; 
 int TotalInvoice= 0;
 //String str = "";
 Connection  _con = null;
Statement _statement =null;
Subject _subject = AuthUtil.getAdminSubject();
DatabaseConnectionInfo dbInfo = null ;
try{
 
  Entity beanObject1 = (Entity) DatabaseConnectionInfo.class.newInstance();
 
  EntityManager entityManager1 = EntityManagerFactory.getEntityManager( beanObject1.getClass() , _subject );
 
  Iterator it = entityManager1.retrieve();
 

            while( it.hasNext() )
            {
                DatabaseConnectionInfo temp = (DatabaseConnectionInfo) it.next() ;
                String entityId = temp.getId() ;
                if( entityId.equals( "192168001031132127506207800002") ) // Enter the EntityId of database info.
                {
                    dbInfo = temp ;
                    break ;
                }
            }
            
_con = JdbcUtils.getConnection(dbInfo , _subject );
 
 _statement = _con.createStatement() ;
 _statement.executeUpdate(Query) ;
  }
  catch(SQLException e)
  {
   e.printStackTrace();
   throw e;
  } 
  catch (ClassNotFoundException e) {
   e.printStackTrace();
   throw e;
  }
  finally{  
    if(_statement != null){
   _statement.close();
   }
   if(_con != null){
   _con.close();
   }
   }
}