Merge pull request #10 from vlkong/master

Use post process listener to write results

Author: vlkong

build.xml

@@ -29,6 +29,7 @@
 
 	<property name="oplall.jar" value="${example.home}/lib/oplall.jar"/>
 
+	<property name="debug" value="off"/>
 
     <path id="project.class.path">
         <pathelement location="build/"/>
@@ -43,7 +44,7 @@
     </target>
 
     <target name="compile" depends="init">
-        <javac srcdir="${src}" destdir="${build}">
+        <javac srcdir="${src}" destdir="${build}" debug="${debug}">
             <classpath refid="project.class.path"/>
         </javac>
     </target>

lib/jdbc-custom-data-source.jar

@@ -1,6 +1,7 @@
 package com.ibm.opl.customdatasource;
 
 import ilog.opl.IloCustomOplDataSource;
+import ilog.opl.IloCustomOplPostProcessListener;
 import ilog.opl.IloOplDataHandler;
 import ilog.opl.IloOplElement;
 import ilog.opl.IloOplElementDefinition;
@@ -20,6 +21,8 @@
 import java.util.Enumeration;
 import java.util.Properties;
 
+
+
 /**
  * An custom data source reading data using JDBC.
  *
@@ -28,6 +31,30 @@ public class JdbcCustomDataSource extends IloCustomOplDataSource {
     private JdbcConfiguration _configuration;
     private IloOplModelDefinition _def;
 
+    /**
+     * A post process listener to write output
+     */
+    private static class JdbcCustomDataSourcePublisher extends IloCustomOplPostProcessListener {
+      IloOplModel _model;
+      JdbcConfiguration _config;
+      
+      
+      JdbcCustomDataSourcePublisher(IloOplFactory factory, IloOplModel model, JdbcConfiguration config) {
+        super(factory);
+        _model = model;
+        _config = config;
+      }
+
+      @Override
+      public void customStartPostProcess() {
+      }
+      
+      @Override
+      public void customEndPostProcess() {
+        JdbcWriter.writeOutput(_config, _model);
+      }
+    }
+    
     /**
      * Adds a custom data source to a model.
      *
@@ -41,6 +68,7 @@ public static void addDataSourceXMLConfig(String xmlFile, IloOplModel model) thr
         IloOplModelDefinition definition = model.getModelDefinition();
         JdbcCustomDataSource source = new JdbcCustomDataSource(config, factory, definition);
         model.addDataSource(source);
+        model.registerPostProcessListener(new JdbcCustomDataSourcePublisher(factory, model, config));
     }
 
     /**
@@ -54,6 +82,7 @@ public static void addDataSource(JdbcConfiguration config, IloOplModel model) {
         IloOplModelDefinition definition = model.getModelDefinition();
         JdbcCustomDataSource source = new JdbcCustomDataSource(config, factory, definition);
         model.addDataSource(source);
+        model.registerPostProcessListener(new JdbcCustomDataSourcePublisher(factory, model, config));
     }
 
     /**

src/main/java/com/ibm/opl/customdatasource/JdbcCustomDataSource.java

@@ -41,6 +41,7 @@ public JdbcWriter(JdbcConfiguration configuration, IloOplModelDefinition def, Il
     }
 
     public void customWrite() {
+        long startTime = System.currentTimeMillis();
         System.out.println("Writing elements to database");
         Properties prop = _configuration.getWriteMapping();
         Enumeration<?> propertyNames = prop.propertyNames();
@@ -50,7 +51,8 @@ public void customWrite() {
             System.out.println("Writing " + name + " using table " + table + "");
             customWrite(name, table);
         }
-        System.out.println("Done");
+        long endTime = System.currentTimeMillis();
+        System.out.println("Done (" + (endTime - startTime)/1000.0 + " s)");
     }
 
     static final String CREATE_QUERY = "CREATE TABLE %(";
@@ -74,7 +76,7 @@ else if (type == Type.STRING)
                 query += ", ";
         }
         query += ")";
-        System.out.println("Create query = " + query);
+        // System.out.println("Create query = " + query);
         return query;
     }