import groovy.sql.Sql; //导入依赖的package包/类
@Override public Connection connectionFrom(Object source) {
  if (DataSource.class.isAssignableFrom(source.getClass())) {
    return DATA_SOURCE_CONNECTOR.connectionFrom(source);
  } else if (Sql.class.isAssignableFrom(source.getClass())) {
    return GROOVY_SQL_CONNECTOR.connectionFrom(source);
  } else if (Connection.class.isAssignableFrom(source.getClass())) {
    return CONNECTION_CONNECTOR.connectionFrom(source);
  }
  throw new IllegalArgumentException(
    format(
      "Unable to obtain a database connection from an instance of %s",
      source.getClass()
    )
  );
}
 

import groovy.sql.Sql; //导入依赖的package包/类
@Override protected Connection apply(Sql source) throws SQLException {
  if (source.getConnection() != null) {
    return CONNECTION_CONNECTOR.connectionFrom(source.getConnection());
  } else {
    return DATA_SOURCE_CONNECTOR.connectionFrom(source.getDataSource());
  }
}
 

import groovy.sql.Sql; //导入依赖的package包/类
/**
 * Creates a new ManagedHikariDataSource
 * @param configuration the data source configuration
 * @param listeners Optional close listeners
 */
public ManagedHikariDataSource(final HikariConfig configuration, final DataSourceListener... listeners) {
	super(configuration);
	poolName = getPoolName();
	groovySql = new Sql(this);
	dsCacheKey = CACHE_DS_PREFIX + poolName;
	groovydsCacheKey = CACHE_GSQL_PREFIX + poolName;
	if(listeners!=null) {
		for(DataSourceListener listener: listeners) {
			if(listener==null) continue;
			this.listeners.add(listener);
		}
	}
}
 

import groovy.sql.Sql; //导入依赖的package包/类
public static int executeInsert(Connection conn, String sql) throws SQLException {
    System.out.println(sql);
    Sql client = new Sql(conn);
    Integer result = (Integer) client.executeInsert(sql).get(0).get(0);
    conn.close();
    return result;
}
 

import groovy.sql.Sql; //导入依赖的package包/类
public static int executeUpdate(Connection conn, String sql) throws SQLException {
    System.out.println(sql);
    Sql client = new Sql(conn);
    int result = client.executeUpdate(sql);
    conn.close();
    return result;
}
 

import groovy.sql.Sql; //导入依赖的package包/类
@SuppressWarnings("unchecked")
public static List<List<Object>> executeQuery(Connection conn, String sql) throws SQLException {
     System.out.println(sql);
     Sql client = new Sql(conn);
     List<GroovyRowResult> rows = client.rows(sql);

     List<List<Object>> result = new ArrayList<List<Object>>();

     for (GroovyRowResult row : rows) {
         result.add(Lists.newArrayList(row.values()));
     }
     conn.close();
     return result;
 }
 

import groovy.sql.Sql; //导入依赖的package包/类
@BeforeClass
public static void init() throws SQLException {
	GroupDataSource ds = createDs();

	Sql sql = new Sql(ds.getConnection());
	sql.execute("CREATE TABLE Persons\n" + "(\n" + "Id int,\n" + "LastName varchar(255),\n"
	      + "FirstName varchar(255),\n" + "Address varchar(255),\n" + "City varchar(255)\n" + ")");
	sql.execute("insert into persons (id,lastname,firstname,address,city) values (1,'','','','')");
}
 

import groovy.sql.Sql; //导入依赖的package包/类
@Test(expected = ZebraException.class, timeout = 30000)
public void test_connect_fail() throws SQLException {
	GroupDataSource ds = new GroupDataSource();
	ds.setConfigManagerType(Constants.CONFIG_MANAGER_TYPE_LOCAL);
	ds.setJdbcRef("sample.ds.error");
	ds.setFilter("!wall,cat");
	ds.init();
	new Sql(ds.getConnection()).execute("select 1");
}
 

import groovy.sql.Sql; //导入依赖的package包/类
@Test
public void test_sql_success1() throws SQLException {
	GroupDataSource ds = createDs();

	DaoContextHolder.setSqlName("testPreparedStatementQuery");

	new Sql(ds.getConnection()).execute(Constants.SQL_FORCE_WRITE_HINT + "select * from Persons");
}
 

import groovy.sql.Sql; //导入依赖的package包/类
@Test(expected = SQLException.class)
public void test_sql_rejected_by_flow_control() throws SQLException {
	GroupDataSource ds = new GroupDataSource();
	ds.setConfigManagerType(Constants.CONFIG_MANAGER_TYPE_LOCAL);
	ds.setJdbcRef("sample.ds.v2");
	ds.setFilter("wall,cat");
	ds.init();

	DaoContextHolder.setSqlName("test");

	new Sql(ds.getConnection()).execute("select 1", new Object[0]);
}
 

import groovy.sql.Sql; //导入依赖的package包/类
@Test
public void test_sql_success() throws SQLException {
	GroupDataSource ds = createDs();

	new Sql(ds.getConnection()).execute(Constants.SQL_FORCE_WRITE_HINT + "select * from Persons");
}
 

import groovy.sql.Sql; //导入依赖的package包/类
@Test(expected = Exception.class)
public void test_sql_fail_on_slave() throws SQLException {
	GroupDataSource ds = createDs();

	new Sql(ds.getConnection()).execute("select * from xxx");
}
 

import groovy.sql.Sql; //导入依赖的package包/类
/**
 * @return sql - {return content description}
 */
public static Sql getSql() {
	return sql;
}