Jdbc模板,實現動態生成SQL...
阿新 • • 發佈:2019-02-11
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.util.ArrayList;
import java.util.List;
import org.ddang.util.DbUtil;
/**
* 規定:
* 1.資料表名 “tb_” + “物件名”
* 2.資料庫欄位順序與對應實體屬性順序一致(不是必須的,但是ID必須是第一個屬性)
* 功能:通過反射實現CRUD
* 使用:繼承該類,呼叫對應方法,傳入物件型別
DbUtil.為自定義操作資料庫工具類
* @author Administrator
*
*/
public class JdbcTemplate {
public final static String TABLE_PREFIX = "tb_";
private String tableName = "";
private Field[] fields;
private List getFields(Object obj) {
Class c = obj.getClass();
// 獲得類名
String className = obj.getClass().getSimpleName();
// 類名 + 字首 = 表名
tableName = TABLE_PREFIX + className.toLowerCase();
// 獲得所有的屬性
fields = c.getDeclaredFields();
Method[] method = c.getMethods();
List list = new ArrayList();
for (int j = 0; j < fields.length; j++) {
String m = "get" + fields[j].getName().toUpperCase().charAt(0)
+ fields[j].getName().substring(1 );
for (int i = 0; i < method.length; i++) {
if (method[i].getName().endsWith(m)) {
// method[i].invoke("", null);
try {
list.add(method[i].invoke(obj, null));
System.out.println("---------"
+ method[i].invoke(obj, null));
} catch (Exception e) {
e.printStackTrace();
return null;
}
}
}
}
return list;
}
private String getInsertSQL(Object obj) {
// 拼SQL語句
StringBuffer sql = new StringBuffer();
sql.append("INSERT INTO ");
sql.append("`" + tableName + "`");
sql.append(" (");
for (int i = 0; i < fields.length; i++) {
sql.append("`" + fields[i].getName() + "`");
if (i < fields.length - 1) {
sql.append(",");
}
}
sql.append(") ");
sql.append(" VALUES(");
for (int i = 0; i < fields.length; i++) {
sql.append("?");
if (i < fields.length - 1) {
sql.append(",");
}
}
sql.append(") ");
return sql.toString();
}
private String getUpdateSQL(Object obj) {
// 拼SQL語句
StringBuffer sql = new StringBuffer();
sql.append("UPDATE ");
sql.append("`" + tableName + "` ");
sql.append(" SET ");
for (int i = 1; i < fields.length; i++) {
sql.append("`" + fields[i].getName() + "`");
sql.append("=?");
if (i < fields.length - 1) {
sql.append(",");
}
}
sql.append(" WHERE ");
sql.append(fields[0].getName());
sql.append("=?");
return sql.toString();
}
private String getSelectSQL(Object obj) {
// 拼SQL語句
StringBuffer sql = new StringBuffer();
sql.append("SELECT * FROM ");
sql.append("`" + tableName + "` ");
return sql.toString();
}
private String getDeleteSQL(Object obj) {
// 拼SQL語句
StringBuffer sql = new StringBuffer();
sql.append("DELETE FROM ");
sql.append("`" + tableName + "` ");
sql.append(" WHERE ");
sql.append(fields[0].getName());
sql.append("=?");
return sql.toString();
}
public boolean save(Object obj) throws Exception {
// 獲取obj的屬性的值
List list = getFields(obj);
// 獲取sql
String sql = getInsertSQL(obj);
System.out.println(sql);
// 通過DbUtil//連線資料庫
Connection conn = DbUtil.getConn();
PreparedStatement ps = conn.prepareStatement(sql);
for (int i = 1; i <= list.size(); i++) {
ps.setObject(i, list.get(i-1));
}
boolean flag = ps.executeUpdate() > 0 ? true : false;
DbUtil.close(conn);
return flag;
}
public boolean update(Object obj) throws Exception {
// 獲取obj的屬性的值
List list = getFields(obj);
// 獲取sql
String sql = getUpdateSQL(obj);
// 通過DbUtil
Connection conn = DbUtil.getConn();
PreparedStatement ps = conn.prepareStatement(sql);
for (int i = 1; i < list.size(); i++) {
ps.setObject(i, list.get(i));
}
ps.setInt(list.size(), (Integer) list.get(0));
boolean flag = ps.executeUpdate() > 0 ? true : false;
DbUtil.close(conn);
return flag;
}
public List select(Object obj) throws Exception {
return select(obj,null,null);
}
public List select(Object obj,String fieldName,Object fieldValue) throws Exception {
// 獲取obj的屬性的值
List lists = getFields(obj);
// 獲取sql
String sql = getSelectSQL(obj);
if(fieldName != null)
sql += " WHERE `"+ fieldName +"` =?";
// 通過DbUtil
System.out.println("SQL:"+sql);
Connection conn = DbUtil.getConn();
PreparedStatement ps = conn.prepareStatement(sql);
if(fieldName != null)
ps.setObject(1, fieldValue);
ResultSet rs = ps.executeQuery();
Class c = obj.getClass();
Method[] method = c.getMethods();
List list = new ArrayList();
while (rs.next()) {
Object o = c.newInstance();
// 獲得所有的屬性
for (int j = 0; j < fields.length; j++) {
String m = "set" + fields[j].getName().toUpperCase().charAt(0)
+ fields[j].getName().substring(1);
for (int i = 0; i < method.length; i++) {
if (method[i].getName().endsWith(m)) {
//System.out.println("dd"+rs.getObject(j));
try{
method[i].invoke(o, rs.getObject(j+1));
} catch (Exception e) {
e.printStackTrace();
}
}
}
}
list.add(o);
}
DbUtil.close(conn);
return list;
}
public boolean delete(Object obj,Integer id) throws Exception {
// 獲取obj的屬性的值
List list = getFields(obj);
// 獲取sql
String sql = getDeleteSQL(obj);
// 通過DbUtil
Connection conn = DbUtil.getConn();
PreparedStatement ps = conn.prepareStatement(sql);
ps.setInt(1, id);
boolean flag = ps.executeUpdate() > 0 ? true : false;
DbUtil.close(conn);
return flag;
}
}