import java.io.ObjectInputStream.GetField;
import java.math.BigInteger;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.SQLWarning;
import java.sql.Statement;
import java.util.HashMap;
import java.util.Map;
 
import org.neo4j.graphdb.DynamicRelationshipType;
import org.neo4j.kernel.impl.batchinsert.BatchInserter;
import org.neo4j.kernel.impl.batchinsert.BatchInserterImpl;
 
public class Importer {
 
        public static void main(String[] argv) {
                
                 Map<String,String> props = new HashMap<String, String>();
                 props.put("neostore.nodestore.db.mapped_memory", "22000000"); // <expected number of nodes * 9 bytes>
                 props.put("neostore.relationshipstore.db.mapped_memory", "22000000"); // <expected number of relationships * 33 bytes>
                  // create the batch inserter 
                BatchInserter inserter = new
                  BatchInserterImpl("neo4j-db/",props
                 );
                  
 
 
                 
                
                System.out.println("-------- PostgreSQL "
                                + "JDBC Connection Testing ------------");
 
                try {
 
                        Class.forName("org.postgresql.Driver");
 
                } catch (ClassNotFoundException e) {
 
                        System.out.println("Where is your PostgreSQL JDBC Driver? "
                                        + "Include in your library path!");
                        e.printStackTrace();
 
                }
 
                System.out.println("PostgreSQL JDBC Driver Registered!");
 
                Connection conn = null;
 
                try {
 
                        conn = DriverManager.getConnection(
                                        "jdbc:postgresql://127.0.0.1:5432/contractDashboard",
                                        "postgres", "snmc");
 
                } catch (SQLException e) {
 
                        System.out.println("Connection Failed! Check output console");
                        e.printStackTrace();
 
                }
 
                if (conn != null) {
                        System.out.println("You made it, take control your database now!");
                } else {
                        System.out.println("Failed to make connection!");
                }
                try {
                        // Print all warnings
                        for (SQLWarning warn = conn.getWarnings(); warn != null; warn = warn
                                        .getNextWarning()) {
                                System.out.println("SQL Warning:");
                                System.out.println("State  : " + warn.getSQLState());
                                System.out.println("Message: " + warn.getMessage());
                                System.out.println("Error  : " + warn.getErrorCode());
                        }
 
                        // Get a statement from the connection
                        Statement stmt = conn.createStatement();
 
                        // Execute the query
                        ResultSet rs = stmt.executeQuery("SELECT distinct contractnotice.\"agencyName\",   contractnotice.\"supplierABN\",   contractnotice.\"supplierName\" FROM  public.contractnotice limit 300;");
String previousAgency = "";
 
                        // Loop through the result set
                        while (rs.next()) {
                                long supplier,agency;
                                agency = doHash(rs.getString("agencyName"));
                                if (rs.getString("agencyName") != previousAgency) {
                                        if (!inserter.nodeExists(agency)) {
                                        Map<String, Object> properties = new HashMap<String, Object>();
                                          properties.put("Label", rs.getString("agencyName"));
                                          inserter.createNode(agency, properties);
                                        }
                                }
                                if (rs.getString("supplierABN") != "0") {
                                        supplier = doHash(rs.getString("supplierABN"));
                                } else {
                                supplier = doHash(rs.getString("supplierName"));
                                }
                                  // inject some data 
                        if (!inserter.nodeExists(supplier)) {
                                Map<String, Object> properties = new HashMap<String, Object>();
                        
                                  properties.put("Label", rs.getString("supplierName"));
                                  inserter.createNode(supplier, properties);
                        }
                                  
                                  inserter.createRelationship(agency, supplier,
                                  DynamicRelationshipType.withName("KNOWS"), null);
                                  
                        }
                        // Close the result set, statement and the connection
                        rs.close();
                        stmt.close();
                        conn.close();
                } catch (SQLException se) {
                        System.out.println("SQL Exception:");
 
                        // Loop through the SQL Exceptions
                        while (se != null) {
                                System.out.println("State  : " + se.getSQLState());
                                System.out.println("Message: " + se.getMessage());
                                System.out.println("Error  : " + se.getErrorCode());
 
                                se = se.getNextException();
                        }
                } catch (Exception e) {
                        System.out.println(e);
                }
                  // shutdown, makes sure all changes are written to disk
                  inserter.shutdown();
        }
 
        static long doHash(String input) {
                MessageDigest m;
                try {
                        m = MessageDigest.getInstance("MD5");
                        m.reset();
                        m.update(input.getBytes());
                        byte[] digest = m.digest();
                        return new BigInteger(1, digest).longValue();
                } catch (NoSuchAlgorithmException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                }
                return 0;
 
        }
 
 
}