com.softlyinspired.jlw.connections.dbConnection.java Source code

Introduction

Here is the source code for com.softlyinspired.jlw.connections.dbConnection.java

Source

/*
 * This code is part of JLW Warehouse Wonder
 * Copyright (c) 2016-  Julie Kenny @ Softly Inspired  All rights reserved. 
 *
 * This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
    
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
    
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
    
 */
package com.softlyinspired.jlw.connections;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.Statement;

import com.mongodb.*;
import com.softlyinspired.jlw.JLWUtilities;
import com.softlyinspired.jlw.mongodb.repoConnection;

public class dbConnection {

    public String connectionName;
    public String connectionDriver;
    public String connectionURL;
    public String connectionUser;
    public String connectionPassword;

    DBCollection connectionColl;
    MongoClient mongoClient;
    DBObject currentConnectionDoc;
    private static String collectionName = new String("connections");

    public int create() {
        /* Create a new connection */
        if (this.checkExisting()) {
            return 1;
        }

        try {
            /* Get the database collection */
            DBCollection connectionColl = repoConnection.getConnectionCollection();
            if (connectionColl == null) {
                return -1;
            }

            // search for the id
            BasicDBObject query = new BasicDBObject();
            query.put("name", connectionName);
            query.put("driver", connectionDriver);
            query.put("url", connectionURL);
            query.put("password", connectionPassword);
            query.put("user", connectionUser);

            try {
                connectionColl.insert(query);
            } catch (Exception e) {
                System.out.println("Error creating concern");
            }

        } catch (Exception e) {
            System.out.println("error in checking script during create");
        }
        return 0;
    }

    public boolean checkExisting() {
        /**
         * sets value of concernExists;
         */
        boolean nameExists;
        try {
            /* Get the database collection */
            DBCollection connectionColl = repoConnection.getConnectionCollection();
            if (connectionColl == null) {
                return false;
            }
            ;
            // search for the id
            BasicDBObject query = new BasicDBObject();
            query.put("name", connectionName);

            DBCursor checkDoc = connectionColl.find(query).limit(1);

            if (checkDoc.count() == 0) {
                nameExists = false;
            } else {
                nameExists = true;
                currentConnectionDoc = checkDoc.next();
            }

        } catch (Exception e) {
            System.out.println("Connection does not exist");
            nameExists = false;
        }
        return nameExists;
    }

    public int updateExisting() {
        /* Update connection details */

        int result = 0;
        DBCollection connectionColl = repoConnection.getConnectionCollection();

        checkExisting();

        try {
            // search for the id
            BasicDBObject updateQuery = new BasicDBObject();
            BasicDBObject searchQuery = new BasicDBObject().append("name", connectionName);
            DBObject changes = new BasicDBObject().append("driver", connectionDriver).append("url", connectionURL)
                    .append("user", connectionUser).append("password", connectionPassword);
            updateQuery.append("$set", changes);
            connectionColl.update(searchQuery, updateQuery);
            JLWUtilities.scriptInfoMessage("Existing Connection Saved");
        } catch (Exception e) {
            System.out.println("Error updating connection");
        }

        return result;
    }

    public dbConnection fetchDetails(String chkConnectionName) {
        /* Update concern details */
        connectionName = chkConnectionName;
        checkExisting();
        connectionName = chkConnectionName;

        try {
            connectionDriver = currentConnectionDoc.get("driver").toString();
        } catch (Exception e) {
            connectionDriver = "";
        }
        ;

        try {
            connectionURL = currentConnectionDoc.get("url").toString();
        } catch (Exception e) {
            connectionURL = "";
        }

        try {
            connectionUser = currentConnectionDoc.get("user").toString();
        } catch (Exception e) {
            connectionUser = "";
        }

        try {
            connectionPassword = currentConnectionDoc.get("password").toString();
            //connectionPassword = Arrays(currentConnectionDoc.get("password").toCharArray(););
        } catch (Exception e) {
            connectionPassword = null;
        }

        return this;

    }

    public int delete() {
        /* Delete current connection */
        try {
            connectionColl = repoConnection.getConnectionCollection();
            BasicDBObject searchQuery = new BasicDBObject().append("name", this.connectionName);
            connectionColl.remove(searchQuery);
            return 0;
        } catch (Exception E) {
            System.out.println("Error in delete process");
            return 1;
        }

    }

    public boolean test() {
        /* Test the connection */
        // Database credentials
        Connection conn = null;
        Statement stmt = null;
        try {

            String passwordUse = this.connectionPassword.toString();
            conn = DriverManager.getConnection(this.connectionURL, this.connectionUser,
                    String.valueOf(connectionPassword));
            stmt = conn.createStatement();
            JLWUtilities.scriptInfoMessage("Connection Successful");
            return true;

        } catch (Exception e) {
            JLWUtilities.scriptErrorMessage("Connection Failed : " + "\n" + e);
            return false;
        }
    }

}