com.litongjava.db.activerecord.builder.TimestampProcessedModelBuilder Maven / Gradle / Ivy
package com.litongjava.db.activerecord.builder;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.sql.Types;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import com.litongjava.db.activerecord.CPI;
import com.litongjava.db.activerecord.Model;
import com.litongjava.db.activerecord.ModelBuilder;
/**
* TimestampProcessedModelBuilder
* 时间戳被处理过的 ModelBuilder
* oracle 从 Connection 中取值时需要调用具体的 getTimestamp(int) 来取值
*/
public class TimestampProcessedModelBuilder extends ModelBuilder {
public static final TimestampProcessedModelBuilder me = new TimestampProcessedModelBuilder();
@Override
@SuppressWarnings({"rawtypes"})
public List build(ResultSet rs, Class modelClass) throws SQLException, ReflectiveOperationException {
return build(rs, modelClass, null);
}
@Override
@SuppressWarnings({"rawtypes", "unchecked"})
public List build(ResultSet rs, Class modelClass, Function func) throws SQLException, ReflectiveOperationException {
List result = new ArrayList();
ResultSetMetaData rsmd = rs.getMetaData();
int columnCount = rsmd.getColumnCount();
String[] labelNames = new String[columnCount + 1];
int[] types = new int[columnCount + 1];
buildLabelNamesAndTypes(rsmd, labelNames, types);
while (rs.next()) {
Model ar = modelClass.newInstance();
Map attrs = CPI.getAttrs(ar);
for (int i=1; i<=columnCount; i++) {
Object value;
if (types[i] < Types.DATE) {
value = rs.getObject(i);
} else {
if (types[i] == Types.TIMESTAMP) {
value = rs.getTimestamp(i);
} else if (types[i] == Types.DATE) {
value = rs.getDate(i);
} else if (types[i] == Types.CLOB) {
value = handleClob(rs.getClob(i));
} else if (types[i] == Types.NCLOB) {
value = handleClob(rs.getNClob(i));
} else if (types[i] == Types.BLOB) {
value = handleBlob(rs.getBlob(i));
} else {
value = rs.getObject(i);
}
}
attrs.put(labelNames[i], value);
}
if (func == null) {
result.add((T)ar);
} else {
if ( ! func.apply((T)ar) ) {
break ;
}
}
}
return result;
}
}