at.ac.tuwien.dsg.depic.external.algorithm.MySqlDataAssetStore.java Source code

Introduction

Here is the source code for at.ac.tuwien.dsg.depic.external.algorithm.MySqlDataAssetStore.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package at.ac.tuwien.dsg.depic.external.algorithm;

import at.ac.tuwien.dsg.depic.common.entity.eda.dataasset.DataAsset;
import at.ac.tuwien.dsg.depic.common.utils.JAXBUtils;
import at.ac.tuwien.dsg.depic.common.utils.MySqlConnectionManager;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.io.StringWriter;
import java.nio.charset.StandardCharsets;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
 *
 * @author Jun
 */
public class MySqlDataAssetStore {

    private MySqlConnectionManager connectionManager;

    public MySqlDataAssetStore() {
        String ip = "localhost";
        String port = "3306";
        String db = "GPS";
        String user = "root";
        String pass = "123";

        connectionManager = new MySqlConnectionManager(ip, port, db, user, pass);
    }

    public void storeDataAsset(DataAsset da) {

        try {
            String daXML = JAXBUtils.marshal(da, DataAsset.class);

            //   Logger.getLogger(MySqlDataAssetStore.class.getName()).log(Level.INFO, "Prepare to Store: " + daXML);
            InputStream daStream = new ByteArrayInputStream(daXML.getBytes(StandardCharsets.UTF_8));
            List<InputStream> listOfInputStreams = new ArrayList<InputStream>();
            listOfInputStreams.add(daStream);

            String sql = "INSERT INTO DataAsset (dataAssetID, dataPartitionID, data) VALUES ('"
                    + da.getDataAssetID() + "'," + da.getPartition() + ",?)";

            connectionManager.ExecuteUpdateBlob(sql, listOfInputStreams);

        } catch (Exception ex) {
            Logger.getLogger(MySqlDataAssetStore.class.getName()).log(Level.SEVERE, ex.toString());
        }

    }

    public void updateDataAsset(String dataAssetID, String dataParitionID, String data) {
        InputStream dataStream = new ByteArrayInputStream(data.getBytes(StandardCharsets.UTF_8));
        List<InputStream> listOfInputStreams = new ArrayList<InputStream>();
        listOfInputStreams.add(dataStream);

        String sql = "UPDATE DataAsset SET data=? WHERE dataAssetID='" + dataAssetID + "' AND dataPartitionID='"
                + dataParitionID + "'";
        connectionManager.ExecuteUpdateBlob(sql, listOfInputStreams);

    }

    public ResultSet getDataAssetByID(String dawID) {

        String sql = "SELECT * FROM DataAsset WHERE dataAssetID='" + dawID + "'";
        ResultSet rs = connectionManager.ExecuteQuery(sql);

        return rs;
    }

    public int getNumberOfPartitionsByDataAssetID(String dataAssetID) {

        String sql = "Select COUNT(dataPartitionID) as counter FROM DataAsset WHERE dataAssetID='" + dataAssetID
                + "'";
        ResultSet rs = connectionManager.ExecuteQuery(sql);
        int counter = 0;

        try {
            while (rs.next()) {
                counter = rs.getInt("counter");
            }
            rs.close();
        } catch (SQLException ex) {
            Logger.getLogger(MySqlDataAssetStore.class.getName()).log(Level.SEVERE, null, ex);
        }

        return counter;
    }

    public String getDataPartition(String dataAssetID, String dataPartitionID) {

        String fileName = dataAssetID + ".data";

        at.ac.tuwien.dsg.depic.common.utils.IOUtils iou = new at.ac.tuwien.dsg.depic.common.utils.IOUtils(
                "/Volumes/DATA/BigData/mydb2");
        String daXML = iou.readData(fileName);

        //        String daXML="";
        //        String sql ="SELECT * FROM DataAsset WHERE dataAssetID='"+dataAssetID+"' AND dataPartitionID='"+dataPartitionID+"'";
        //        ResultSet rs = connectionManager.ExecuteQuery(sql);
        //        
        //        try {
        //            while (rs.next()){
        //                
        //                InputStream inputStream = rs.getBinaryStream("data");
        //
        //                StringWriter writer = new StringWriter();
        //                String encoding = StandardCharsets.UTF_8.name();
        //                org.apache.commons.io.IOUtils.copy(inputStream, writer, encoding);
        //                daXML = writer.toString();
        //            }
        //            
        //            rs.close();
        //        } catch (Exception ex) {
        //            Logger.getLogger(MySqlDataAssetStore.class.getName()).log(Level.SEVERE, ex.toString());
        //        }

        return daXML;
    }

    public void cleanTempStore() {

        String sql = "TRUNCATE TABLE DataAsset";
        System.out.println("cleanTempStore: " + sql);

        connectionManager.ExecuteUpdate(sql);

    }

}