com.taobao.tddl.jdbc.group.integration.CRUDTest.java Source code

Introduction

Here is the source code for com.taobao.tddl.jdbc.group.integration.CRUDTest.java

Source

 /*(C) 2007-2012 Alibaba Group Holding Limited.   
  *This program is free software; you can redistribute it and/or modify   
 *it under the terms of the GNU General Public License version 2 as   
 * published by the Free Software Foundation.   
 * Authors:   
 *   junyu <junyu@taobao.com> , shenxun <shenxun@taobao.com>,   
 *   linxuan <linxuan@taobao.com> ,qihao <qihao@taobao.com>    
 */
 package com.taobao.tddl.jdbc.group.integration;

 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;

 import java.sql.Connection;
 import java.sql.PreparedStatement;
 import java.sql.ResultSet;
 import java.sql.Statement;
 import java.util.ArrayList;
 import java.util.List;

 import javax.sql.DataSource;

 import org.junit.AfterClass;
 import org.junit.Before;
 import org.junit.BeforeClass;
 import org.junit.Test;

 import com.taobao.tddl.interact.rule.bean.DBType;
 import com.taobao.tddl.jdbc.group.DataSourceWrapper;
 import com.taobao.tddl.jdbc.group.TGroupDataSource;
 import com.taobao.tddl.jdbc.group.testutil.DBHelper;
 import com.taobao.tddl.jdbc.group.testutil.DataSourceFactory;

 /**
  * TAtomDataSourceorg.apache.commons.dbcp.BasicDataSourcecrud
  * 
  * @author yangzhu
  *
  */
 public class CRUDTest {
     @BeforeClass
     public static void setUpBeforeClass() throws Exception {
         //DBHelper.deleteAll(); //crud
     }

     @AfterClass
     public static void tearDownAfterClass() {

     }

     @Before
     public void setUp() throws Exception {
         DBHelper.deleteAll();
     }

@Test
public void () throws Exception {
   TGroupDataSource ds = new TGroupDataSource();
   DataSourceWrapper dsw = new DataSourceWrapper("db1", "rw", DataSourceFactory.getMySQLDataSource(), DBType.MYSQL);
   ds.init(dsw);
      
   Connection conn = ds.getConnection();

   //Statementcrud
   Statement stmt = conn.createStatement();
   assertEquals(stmt.executeUpdate("insert into crud(f1,f2) values(10,'str')"), 1);
   assertEquals(stmt.executeUpdate("update crud set f2='str2'"), 1);
   ResultSet rs = stmt.executeQuery("select f1,f2 from crud");
   rs.next();
   assertEquals(rs.getInt(1), 10);
   assertEquals(rs.getString(2), "str2");
   assertEquals(stmt.executeUpdate("delete from crud"), 1);
   rs.close();
   stmt.close();

   //PreparedStatementcrud
   String sql = "insert into crud(f1,f2) values(?,?)";
   PreparedStatement ps = conn.prepareStatement(sql);
   ps.setInt(1, 10);
   ps.setString(2, "str");
   assertEquals(ps.executeUpdate(), 1);
   ps.close();

   sql = "update crud set f2=?";
   ps = conn.prepareStatement(sql);
   ps.setString(1, "str2");
   assertEquals(ps.executeUpdate(), 1);
   ps.close();

   sql = "select f1,f2 from crud";
   ps = conn.prepareStatement(sql);
   rs = ps.executeQuery();
   rs.next();
   assertEquals(rs.getInt(1), 10);
   assertEquals(rs.getString(2), "str2");
   rs.close();
   ps.close();

   sql = "delete from crud";
   ps = conn.prepareStatement(sql);
   assertEquals(ps.executeUpdate(), 1);
   ps.close();

   conn.close();
}

@Test
public void DataSourceWrapper() throws Exception {
   List<DataSourceWrapper> dataSourceWrappers = new ArrayList<DataSourceWrapper>();
   dataSourceWrappers.add(new DataSourceWrapper("dbKey1","rw", DataSourceFactory.getMySQLDataSource(1), DBType.MYSQL));
   dataSourceWrappers.add(new DataSourceWrapper("dbKey2","r", DataSourceFactory.getMySQLDataSource(2), DBType.MYSQL));

   TGroupDataSource ds = new TGroupDataSource();
   ds.setDbGroupKey("myDbGroupKey");
   ds.init(dataSourceWrappers);

   Connection conn = ds.getConnection();

   //Statementcrud
   Statement stmt = conn.createStatement();
   assertEquals(stmt.executeUpdate("insert into crud(f1,f2) values(10,'str')"), 1);
   assertEquals(stmt.executeUpdate("update crud set f2='str2'"), 1);
   ResultSet rs = stmt.executeQuery("select f1,f2 from crud");
   rs.next();
   assertEquals(rs.getInt(1), 10);
   assertEquals(rs.getString(2), "str2");
   assertEquals(stmt.executeUpdate("delete from crud"), 1);
   rs.close();
   stmt.close();

   //PreparedStatementcrud
   String sql = "insert into crud(f1,f2) values(?,?)";
   PreparedStatement ps = conn.prepareStatement(sql);
   ps.setInt(1, 10);
   ps.setString(2, "str");
   assertEquals(ps.executeUpdate(), 1);
   ps.close();

   sql = "update crud set f2=?";
   ps = conn.prepareStatement(sql);
   ps.setString(1, "str2");
   assertEquals(ps.executeUpdate(), 1);
   ps.close();

   sql = "select f1,f2 from crud";
   ps = conn.prepareStatement(sql);
   rs = ps.executeQuery();
   rs.next();
   assertEquals(rs.getInt(1), 10);
   assertEquals(rs.getString(2), "str2");
   rs.close();
   ps.close();

   sql = "delete from crud";
   ps = conn.prepareStatement(sql);
   assertEquals(ps.executeUpdate(), 1);
   ps.close();

   conn.close();
}

     //dbGroup: db1:r10w, db2:r20, db3:r30
@Test
public void _db1_db2db3() throws Exception {
   DataSource ds1 = DataSourceFactory.getMySQLDataSource(1);
   DataSource ds2 = DataSourceFactory.getMySQLDataSource(2);
   DataSource ds3 = DataSourceFactory.getMySQLDataSource(3);
      
      
   //db3db2db1
   TGroupDataSource ds = new TGroupDataSource();
   DataSourceWrapper dsw1 = new DataSourceWrapper("db1", "r10w", ds1, DBType.MYSQL);
   DataSourceWrapper dsw2 = new DataSourceWrapper("db2", "r20", ds2, DBType.MYSQL);
   DataSourceWrapper dsw3 = new DataSourceWrapper("db3", "r30", ds3, DBType.MYSQL);
   ds.init(dsw1, dsw2, dsw3);

   Connection conn = ds.getConnection();

   //Statementcrud
   Statement stmt = conn.createStatement();
   assertEquals(stmt.executeUpdate("insert into crud(f1,f2) values(10,'str')"), 1);
   assertEquals(stmt.executeUpdate("update crud set f2='str2'"), 1);
   ResultSet rs = stmt.executeQuery("select f1,f2 from crud");
   rs.next();
   assertEquals(rs.getInt(1), 10);
   assertEquals(rs.getString(2), "str2");
   assertEquals(stmt.executeUpdate("delete from crud"), 1);
   rs.close();
   stmt.close();

   //PreparedStatementcrud
   String sql = "insert into crud(f1,f2) values(10,'str')";
   PreparedStatement ps = conn.prepareStatement(sql);
   assertEquals(ps.executeUpdate(), 1);
   ps.close();

   sql = "update crud set f2='str2'";
   ps = conn.prepareStatement(sql);
   assertEquals(ps.executeUpdate(), 1);
   ps.close();

   sql = "select f1,f2 from crud";
   ps = conn.prepareStatement(sql);
   rs = ps.executeQuery();
   rs.next();
   assertEquals(rs.getInt(1), 10);
   assertEquals(rs.getString(2), "str2");
   rs.close();
   ps.close();

   sql = "delete from crud";
   ps = conn.prepareStatement(sql);
   assertEquals(ps.executeUpdate(), 1);
   ps.close();

   conn.close();
}

     //dbGroup: db1:w, db2:r20, db3:r30
     @Test
     public void _() throws Exception { //
   DataSource ds1 = DataSourceFactory.getMySQLDataSource(1);
   DataSource ds2 = DataSourceFactory.getMySQLDataSource(2);
   DataSource ds3 = DataSourceFactory.getMySQLDataSource(3);
      
   //db3db2db1
   TGroupDataSource ds = new TGroupDataSource();
   DataSourceWrapper dsw1 = new DataSourceWrapper("db1", "w", ds1, DBType.MYSQL);
   DataSourceWrapper dsw2 = new DataSourceWrapper("db2", "r20", ds2, DBType.MYSQL);
   DataSourceWrapper dsw3 = new DataSourceWrapper("db3", "r30", ds3, DBType.MYSQL);
   ds.init(dsw1, dsw2, dsw3);
   Connection conn = ds.getConnection();

   Statement stmt = conn.createStatement();
   assertEquals(stmt.executeUpdate("insert into crud(f1,f2) values(100,'str')"), 1);
      
   //
   //
   //db2,db3rs.next()false
   ResultSet rs = stmt.executeQuery("select f1,f2 from crud where f1=100");
   assertFalse(rs.next());
   rs.close();

   assertEquals(stmt.executeUpdate("delete from crud where f1=100"), 1);
   stmt.close();

   conn.close();
}
 }