package com.wordcount.example;

import java.io.Serializable;
//STEP 1. Import required packages
import java.sql.*;

public class VerticaCON implements Serializable{
	// JDBC driver name and database URL
	static final String JDBC_DRIVER = "com.vertica.jdbc.Driver";
	
	static final String DB_URL = "jdbc:vertica://16.155.190.158:5433/VMart";
	
	static final String USER = "dbadmin";
	static final String PASS = "password";

	public Connection getConnection() {
		Connection conn = null;
		// Statement stmt = null;
		try {
			// STEP 2: Register JDBC driver
			Class.forName("com.vertica.jdbc.Driver");
			// STEP 3: Open a connection
			System.out.println("Connecting to database...");
			conn = DriverManager.getConnection(DB_URL, USER, PASS);
			System.out.println("Connected!!!!");
			// conn.close();
		} catch (SQLException se) {
			// Handle errors for JDBC
			se.printStackTrace();
		} catch (Exception e) {
			// Handle errors for Class.forName
			e.printStackTrace();
		}
		return conn;
	}

	public void closeConnection(Connection conn) throws SQLException {
		conn.close();
		System.out.println("Close Connection");
	}
}