net.snowflake.client.TestConnectionUtil Maven / Gradle / Ivy

Go to download
package net.snowflake.client;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Properties;

/**
 * @author Scott Ziegler
 */
public class TestConnectionUtil
{

//  static final int PORT = 7430;
//  static final String HOST = "192.168.87.99";

  public static final int PORT = 8082;
  public static final String HOST = "snowflake.reg.local";

//  public static final int PORT = 8080;
//  public static final String HOST = "localhost";

  private static final String DEFAULT_CONNECT_STR =
      "jdbc:snowflake://" + HOST + ":" + PORT;

  private static final String USER = "snowman";
  private static final String PASSWORD = "test";

  public static final String ACCOUNT = "testaccount";

  public static final String ADMIN_ACCOUNT = "snowflake";
  static final String ADMIN_USER = "admin";

  public static Connection getConnection()
          throws SQLException
  {
    return getConnection(USER, ACCOUNT);
  }

  public static Connection getConnection(String user, String account)
      throws SQLException
  {
    try
    {
      Class.forName("net.snowflake.client.jdbc.SnowflakeDriver");
    }
    catch (ClassNotFoundException ex)
    {
      System.err.println("Driver not found");
    }

    // build connection properties
    Properties properties = new Properties();

    properties.put("user", user);
    properties.put("password", PASSWORD);
    properties.put("account", account);
    properties.put("ssl", "off");
    properties.put("tracing", "FINE");

    // create a new connection
    String connectStr = System.getenv("SF_JDBC_CONNECT_STRING");

    // use the default connection string if it is not set in environment
    if (connectStr == null)
    {
      connectStr = DEFAULT_CONNECT_STR;
    }

    return DriverManager.getConnection(connectStr, properties);
  }
}