第一次寫編程類博客,不足之處望指正。
最近學mysql數據庫時,同時看了看java和python代碼。一看是看的java代碼不能通用,對比了java和python代碼,想到雖然不能像python一樣簡單,但能不能做成通用的呢?由於學習java時間也不長,就在網上看了看別人寫的代碼,先看到的是csdn上的《JDBC(六) 編寫通用的更新和查詢方法》。但是這個方法只能返回一條,就想能不能一次返回多條或整表,所以又看了看別人的。發現大多類似(或一樣),有的用了格式化,單獨建立了個格式化類,增加大量迭代,類型還受限。所以自己用個半天多的時間,在前人的基礎上搞了一個。個別類筆者也不熟,覺得東西還湊合,拿出了分享一下,歡迎指出問題。關鍵處理了一下hashmap的key不能重復的問題,別的也沒什么,修改前的附在最后吧。
import java.lang.reflect.Field; import java.sql.PreparedStatement; import java.sql.ResultSet; import java.sql.ResultSetMetaData; import java.sql.SQLException; import java.util.ArrayList; import java.util.Iterator; import java.util.LinkedHashMap; import java.util.List; import java.util.Map; import java.util.Map.Entry; import org.apache.logging.log4j.LogManager; import org.apache.logging.log4j.Logger; import com.sql.entity.Student; public class SqlSelect1 { private static Logger logger=LogManager.getLogger(SqlSelect1.class.getName()); public static <T> List<T> query(Class<T> clazz, String sql, Object... args) { T entity = null;
//筆者虛擬機默認裝的mariadb,建立連接與mysql差不多,連接的封裝就不發上來了。 MariaDBConn maria=null; PreparedStatement pstmt = null; ResultSet resultSet = null; List<T> list = new ArrayList<T>(); int colLen=0; try { maria=new MariaDBConn(); pstmt = maria.getConn().prepareStatement(sql); int argsLen=args.length; for (int i = 0; i <argsLen; i++) { pstmt.setObject(i + 1, args[i]); } resultSet = pstmt.executeQuery(); ResultSetMetaData resultSetMetaData = resultSet.getMetaData(); //保證集合有序 Map<String, Object> values = new LinkedHashMap<String, Object> (); while (resultSet.next()) { //使用隨機數使key不重復 double plus=Math.random(); colLen=resultSetMetaData.getColumnCount(); for (int i = 0; i <colLen ; i++) { String columnLable = resultSetMetaData .getColumnLabel(i + 1); Object columnValue = resultSet.getObject(i + 1); values.put(columnLable+plus, columnValue); } } if (values.size() > 0) { Iterator<Entry<String, Object>> it = values.entrySet( ).iterator(); while(it.hasNext()){ int i=0; entity = clazz.newInstance(); while(i<colLen){ Entry<String, Object> entry=it.next(); //剝離隨機數,恢復正常key String fieldName = entry.getKey().toString().split("\\d")[0]; Object value = null; if (entry.getValue() instanceof Long) { value = Integer.parseInt(entry.getValue().toString()); } else { value = entry.getValue(); } Field field = clazz.getDeclaredField(fieldName); field.setAccessible(true); field.set(entity, value); i++; } list.add(entity); } } } catch (Exception e) { logger.error(e); }finally{ try { if(pstmt!=null){ pstmt.close(); } } catch (SQLException e) { logger.error(e); } try { if(resultSet!=null){ resultSet.close(); } } catch (SQLException e) { logger.error(e); } maria.closeConn(); } return list; } //測試: public static void main(String[] args) { List<Student> list=SqlSelect1.query(Student.class, "select id,name,birthday,gender from students"); int size=list.size(); for (int i = 0; i < size; i++) { System.out.println(list.get(i)); } } }
測試類:
import java.sql.Timestamp; public class Student { private int id; private String name; private Timestamp birthday; private boolean gender; public Student() { super(); } public Student(String name, Timestamp birthday, boolean gender, boolean isDelete) { super(); this.name = name; this.birthday = birthday; this.gender = gender; this.isDelete = isDelete; } public Student(int id, String name, Timestamp birthday, boolean gender, boolean isDelete) { super(); this.id = id; this.name = name; this.birthday = birthday; this.gender = gender; this.isDelete = isDelete; } public int getId() { return id; } public String getName() { return name; } public Timestamp getBirthday() { return birthday; } public boolean isGender() { return gender; } public boolean isDelete() { return isDelete; } public void setId(int id) { this.id = id; } public void setName(String name) { this.name = name; } public void setBirthday(Timestamp birthday) { this.birthday = birthday; } public void setGender(boolean gender) { this.gender = gender; } public void setDelete(boolean isDelete) { this.isDelete = isDelete; } boolean isDelete; public String toString() { if(this.gender==false){ return "[Id=" + id + ", name=" + name + ", birthday=" + birthday + ", gender=女" + "]"; }else{ return "[Id=" + id + ", name=" + name + ", birthday=" + birthday + ", gender=男" + "]"; } } }
查詢結果:
[Id=1, name=alice, birthday=1972-01-02 00:00:00.0, gender=女] [Id=2, name=jake, birthday=1988-04-01 00:00:00.0, gender=男] [Id=3, name=zero, birthday=1987-02-01 00:00:00.0, gender=男] [Id=4, name=luffy, birthday=1993-01-09 00:00:00.0, gender=男] [Id=5, name=mike, birthday=1950-01-03 00:00:00.0, gender=男] [Id=6, name=James, birthday=1989-01-09 00:00:00.0, gender=男] [Id=7, name=Ace, birthday=1991-02-10 00:00:00.0, gender=男]
修改前代碼:
public <T> T query(Class<T> clazz, String sql, Object... args) { T entity = null; Connection connection = null; PreparedStatement preparedStatement = null; ResultSet resultSet = null; try { // 獲取resultSet connection = JDBCUtils.getConnection(); preparedStatement = connection.prepareStatement(sql); for (int i = 0; i < args.length; i++) { preparedStatement.setObject(i + 1, args[i]); } resultSet = preparedStatement.executeQuery(); // 獲取resultSetMetaData ResultSetMetaData resultSetMetaData = resultSet.getMetaData(); // 創建一個MAP 鍵 :SQL查詢得到的列的別名,別名和javaBean的屬性名對應 值:對應列的值 Map<String, Object> values = new HashMap<String, Object>(); // 處理結果集,利用resultSetMetaData填充Map while (resultSet.next()) { for (int i = 0; i < resultSetMetaData.getColumnCount(); i++) { String columnLable = resultSetMetaData .getColumnLabel(i + 1); Object columnValue = resultSet.getObject(i + 1); values.put(columnLable, columnValue); } } // 如果map不為空,利用反射創建對象 if (values.size() > 0) { // 反射創建對象 entity = clazz.newInstance(); // 遍歷Map ,利用反射為屬性賦值 for (Map.Entry<String, Object> entry : values.entrySet()) { String fieldName = entry.getKey(); Object value = null; // 因為java中和mysql中數據類型對應不一致 // mysql中int對應JAVA中的long,所以當查詢出int型數據的時候需要轉換 if (entry.getValue() instanceof Long) { value = Integer.parseInt(entry.getValue().toString()); } else { value = entry.getValue(); } // ReflectionUtils.setFieldValue(entity, fieldName, value); Field field = clazz.getDeclaredField(fieldName); field.setAccessible(true); field.set(entity, value); } } } catch (Exception e) { // TODO: handle exception e.printStackTrace(); } return entity; }