webservice: Add inventory database insert query

Signed-off-by: Eduardo L. Buratti <elb09@c3sl.ufpr.br>

webservice/Main.jws

@@ -20,28 +20,32 @@
  * USA.
  */
 
-/* General libs */
 import java.io.*;
 import java.util.*;
-import java.text.*;
+import java.util.regex.*;
 
-/* File libs */
-import java.io.BufferedReader;
-import java.io.File;
-import java.io.FileReader;
-import java.io.StringWriter;
+import java.sql.Timestamp;
+import java.sql.SQLException;
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.Types;
 
-/* DOM */
-import javax.xml.parsers.DocumentBuilder;
-import javax.xml.parsers.DocumentBuilderFactory;
-import org.w3c.dom.*;
+import javax.naming.InitialContext;
+import javax.sql.DataSource;
+import javax.xml.bind.*;
+import javax.xml.bind.util.ValidationEventCollector;
+import javax.xml.bind.helpers.DefaultValidationEventHandler;
+import javax.xml.validation.Schema;
+import javax.xml.validation.SchemaFactory;
 
-/* Regular Expressions */
-import java.util.regex.Pattern;
-import java.util.regex.Matcher;
+import br.ufpr.c3sl.datasid.*;
 
 public class Main {
 
+    private static final String SA_INVENTORY = "sidtb00_sa_inventory";
+
+    private static final File XML_SCHEMA = new File("/home/datasid/collected-data.xsd");
+
     private static final String AGENT_VERSION = "1.0.0";
     private static final String AGENT_UPDATE_LINK = "http://localhost:8280/webservice/datasid-1.0.0-update.run";
 
@@ -104,7 +108,7 @@ public class Main {
      */
     private static String getTimestamp() {
         // gets the current time
-        Date date = new Date();
+        java.util.Date date = new java.util.Date();
         long milisecs = date.getTime();
 
         // generates timestamp string
@@ -112,27 +116,6 @@ public class Main {
         return timestamp.toString();
     }
 
-
-    /**
-     * Parse a XML and returns the root DOM Element
-     *
-     * @author                 Eduardo Luis Buratti
-     * @param     xml          The xml string to be parsed
-     * @return                 Element
-     */
-    private static Element parseXML(String xml) throws Exception {
-        // transform the xml string into a InputStream
-        InputStream is = new ByteArrayInputStream(xml.getBytes());
-
-        // creates a builder
-        DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
-        DocumentBuilder builder = dbf.newDocumentBuilder();
-
-        Document dom = builder.parse(is);
-
-        return dom.getDocumentElement();
-    }
-
     /**
      * Receive an XML string which has the inventory data to be parsed and
      * inserted into database. Return "Success" string if insertion operation
@@ -144,11 +127,41 @@ public class Main {
      */
     public static String setInventory(String xmlData) {
         try {
-            Element root = parseXML(xmlData);
-            System.out.println("RECEIVED: " + root.getTagName());
+            InitialContext cxt = new InitialContext();
+            DataSource ds = (DataSource) cxt.lookup("java:/comp/env/jdbc/datasid");
+            if (ds == null)
+                throw new Exception("Data source not found!");
+
+            Connection con = ds.getConnection();
+            if (con == null)
+                throw new Exception("Failed to get a database connection!");
+
+            SchemaFactory factory = SchemaFactory.newInstance("http://www.w3.org/2001/XMLSchema");
+            Schema schema = factory.newSchema(XML_SCHEMA);
+
+            JAXBContext context = JAXBContext.newInstance("br.ufpr.c3sl.datasid");
+            Unmarshaller unmarshaller = context.createUnmarshaller();
+            unmarshaller.setSchema(schema);
+
+            // Strip spaces in the beginning of the xml
+            xmlData = xmlData.replaceAll("^\\s+", "");
+
+            // transform the xml string into a InputStream
+            InputStream is = new ByteArrayInputStream(xmlData.getBytes());
+
+            // Decode the XML into a Java Object
+            JAXBElement<CollectedData> element = (JAXBElement<CollectedData>) unmarshaller.unmarshal(is);
+            CollectedData collected = element.getValue();
+
+            PreparedStatement st = createInventoryStatement(con, collected);
+            st.executeUpdate();
+
+            con.close();
+
             return "Success";
         } catch (Exception e) {
             log(ERROR, e.getMessage() + " " + xmlData);
+            e.printStackTrace();
             return "ERROR: " + e.getMessage();
         }
     }
@@ -175,4 +188,92 @@ public class Main {
         return AGENT_UPDATE_LINK;
     }
 
+
+    private static PreparedStatement createInventoryStatement(Connection con, CollectedData collectedData) throws SQLException {
+        final String query = "INSERT INTO " + SA_INVENTORY + " " +
+            "(sa_contact_date, sa_gesacid, sa_machine, sa_versao, " +
+            "sa_hd_model, sa_hd_size, sa_hd_used, " +
+            "sa_hd2_model, sa_hd2_size, sa_hd2_used, " +
+            "sa_extra_hds, sa_memory_size, sa_processor, " +
+            "sa_os_type, sa_os_distro, sa_kernel) VALUES " +
+            "(?, ?, ?, ?, " +
+            "?, ?, ?, " +
+            "?, ?, ?, " +
+            "?, ?, ?, " +
+            "?, ?, ?);";
+
+        PreparedStatement st = con.prepareStatement(query);
+
+        List<Interface> interfaces = collectedData.getInterfaces().getInterface();
+
+        Inventory inventory = collectedData.getInventory();
+        int diskCount = inventory.getDisks().getQuantity().intValue();
+        List<Disk> disks = inventory.getDisks().getDisk();
+
+        // sa_contact_date = current date
+        Calendar cal = Calendar.getInstance();
+        st.setDate(1, new java.sql.Date(cal.getTimeInMillis()));
+
+        // sa_gesacid
+        st.setInt(2, collectedData.getGesacid().intValue());
+
+        // sa_machine
+        org.postgresql.util.PGobject macaddr = new org.postgresql.util.PGobject();
+        macaddr.setType("macaddr");
+        macaddr.setValue(interfaces.get(0).getMacAddress());
+        st.setObject(3, macaddr);
+
+        // sa_versao
+        st.setString(4, collectedData.getAgentVersion());
+
+        // sa_hd_model
+        st.setString(5, disks.get(0).getModel());
+
+        // sa_hd_size
+        st.setInt(6, disks.get(0).getSize().intValue());
+
+        // sa_hd_used
+        st.setInt(7, disks.get(0).getUsed().intValue());
+
+        if (diskCount > 1) {
+            // sa_hd2_model
+            st.setString(8, disks.get(1).getModel());
+
+            // sa_hd2_size
+            st.setInt(9, disks.get(1).getSize().intValue());
+
+            // sa_hd2_used
+            st.setInt(10, disks.get(1).getUsed().intValue());
+        }
+        else {
+            // sa_hd2_model
+            st.setNull(8, Types.VARCHAR);
+
+            // sa_hd2_size
+            st.setNull(9, Types.INTEGER);
+
+            // sa_hd2_used
+            st.setNull(10, Types.INTEGER);
+        }
+
+        // sa_extra_hds
+        st.setInt(11, (diskCount > 2) ? (diskCount - 2) : 0);
+
+        // sa_memory_size
+        st.setInt(12, inventory.getMemory().intValue());
+
+        // sa_processor
+        st.setString(13, inventory.getProcessor());
+
+        // sa_os_type
+        st.setString(14, inventory.getOs());
+
+        // sa_os_distro
+        st.setString(15, inventory.getDistro());
+
+        // sa_kernel
+        st.setString(16, inventory.getKernel());
+
+        return st;
+    }
 }