AJ-456、AJ-457: throw exception when new column's rows doesn't match exits cols' size; add new property colRows.

Author: chengyitian

src/com/xxdb/data/BasicTable.java

@@ -17,6 +17,7 @@ public class BasicTable extends AbstractEntity implements Table{
 	private List<String> colNames = new ArrayList<String>();
 	private Map<String, Integer> colNamesIndex = new HashMap<String, Integer>();
 	private int[] colCompresses = null;
+	private int colRows;
 
 	public BasicTable(ExtendedDataInput in) throws IOException{
 		int rows = in.readInt();
@@ -60,6 +61,7 @@ public BasicTable(ExtendedDataInput in) throws IOException{
 			}
 			if(vector.rows() != rows && vector.rows()!= 1)
 				throw new IOException("The number of rows for column " + colNames.get(i) + " is not consistent with other columns");
+			this.colRows = rows;
 			columns.add(vector);
 		}
 		if(collection != null)
@@ -70,10 +72,11 @@ public BasicTable(final List<String> colNames, final List<Vector> cols) {
 		if(colNames.size() != cols.size()){
 			throw new Error("The length of column name and column data is unequal.");
 		}
-		int rowsCount = cols.get(0).rows();
+
+		this.colRows = cols.get(0).rows();
 		for (int i=0;i<cols.size();i++) {
 			Vector v = cols.get(i);
-			if(v.rows() != rowsCount)
+			if(v.rows() != this.colRows)
 				throw new Error("The length of column " + colNames.get(i) + "  must be the same as the first column length.");
 		}
         this.setColName(colNames);
@@ -165,7 +168,7 @@ public int rows() {
 		if(columns()<=0)
 			return 0;
 		else
-			return columns.get(0).rows();
+			return this.colRows;
 	}
 
 	@Override
@@ -380,9 +383,14 @@ public void addColumn(String colName, Vector col) {
 
 		if (colNames.contains(colName))
 			throw new RuntimeException("The table already contains column '" + colName + "'.");
+
+		if (this.colRows != 0 && col.rows() != this.colRows)
+			throw new RuntimeException("The length of column " + colName + "  must be the same as the first column length: " + this.colRows +".");
+
 		colNames.add(colName);
 		colNamesIndex.put(colName, colNamesIndex.size());
 		columns.add(col);
+		this.colRows = col.rows();
 	}
 
 	@Override

src/com/xxdb/streaming/client/MessageParser.java

@@ -62,10 +62,8 @@ public void run() {
 
             if (this.dBConnectionAndSocket == null ) {
                 throw new Exception("dBConnectionAndSocket is null!");
-            } else
-            {
-                if (this.dBConnectionAndSocket.socket != null)
-                {
+            } else {
+                if (this.dBConnectionAndSocket.socket != null) {
                     if (this.dBConnectionAndSocket.conn != null)
                         throw new Exception("Either conn or socket must be null!");
                     socket = this.dBConnectionAndSocket.socket;