首页 > 代码库 > 模拟Hibernate动态生成SQL语句
模拟Hibernate动态生成SQL语句
这里有一个xml配置文件,也就是Hibernate框架中会用到的POJO和数据库的映射文件
1 <?xml version="1.0" encoding="utf-8"?> 2 <!DOCTYPE hibernate-mapping PUBLIC "-//Hibernate/Hibernate Mapping DTD 3.0//EN" 3 "http://www.hibernate.org/dtd/hibernate-mapping-3.0.dtd"> 4 <!-- 5 Mapping file autogenerated by MyEclipse Persistence Tools 6 --> 7 <hibernate-mapping> 8 <class name="hp.pojo.Member" table="member" catalog="hedb"> 9 <id name="mid" type="java.lang.String"> 10 <column name="mid" length="50" /> 11 <generator class="assigned"></generator> 12 </id> 13 <property name="name" type="java.lang.String"> 14 <column name="name" length="50" /> 15 </property> 16 <property name="age" type="java.lang.Integer"> 17 <column name="age" /> 18 </property> 19 <property name="salary" type="java.lang.Double"> 20 <column name="salary" precision="22" scale="0" /> 21 </property> 22 <property name="birthday" type="java.util.Date"> 23 <column name="birthday" length="0" /> 24 </property> 25 <property name="note" type="java.lang.String"> 26 <column name="note" length="65535" /> 27 </property> 28 </class> 29 </hibernate-mapping>
POJO类如下:
1 package hp.pojo; 2 3 import java.util.Date; 4 5 /** 6 * Member entity. @author MyEclipse Persistence Tools 7 */ 8 9 @SuppressWarnings("serial") 10 public class Member implements java.io.Serializable 11 { 12 13 // Fields 14 15 private String mid; 16 private String name; 17 private Integer age; 18 private Double salary; 19 private Date birthday; 20 private String note; 21 22 // Constructors 23 24 /** default constructor */ 25 public Member() 26 { 27 } 28 29 /** minimal constructor */ 30 public Member(String mid) 31 { 32 this.mid = mid; 33 } 34 35 /** full constructor */ 36 public Member(String mid, String name, Integer age, Double salary, Date birthday, String note) 37 { 38 this.mid = mid; 39 this.name = name; 40 this.age = age; 41 this.salary = salary; 42 this.birthday = birthday; 43 this.note = note; 44 } 45 46 // Property accessors 47 48 public String getMid() 49 { 50 return this.mid; 51 } 52 53 public void setMid(String mid) 54 { 55 this.mid = mid; 56 } 57 58 public String getName() 59 { 60 return this.name; 61 } 62 63 public void setName(String name) 64 { 65 this.name = name; 66 } 67 68 public Integer getAge() 69 { 70 return this.age; 71 } 72 73 public void setAge(Integer age) 74 { 75 this.age = age; 76 } 77 78 public Double getSalary() 79 { 80 return this.salary; 81 } 82 83 public void setSalary(Double salary) 84 { 85 this.salary = salary; 86 } 87 88 public Date getBirthday() 89 { 90 return this.birthday; 91 } 92 93 public void setBirthday(Date birthday) 94 { 95 this.birthday = birthday; 96 } 97 98 public String getNote() 99 { 100 return this.note; 101 } 102 103 public void setNote(String note) 104 { 105 this.note = note; 106 } 107 108 }
下面代码演示了利用反射以及XML解析,动态生成SQL语句,并且利用JDBC技术执行SQL语句的过程:
1 package hp.test; 2 3 import java.io.File; 4 import java.lang.reflect.Field; 5 import java.lang.reflect.Method; 6 import java.sql.Connection; 7 import java.sql.DriverManager; 8 import java.util.ArrayList; 9 import java.util.Date; 10 import java.util.List; 11 12 13 14 15 16 import org.dom4j.Document; 17 import org.dom4j.Element; 18 import org.dom4j.io.SAXReader; 19 20 import com.mysql.jdbc.PreparedStatement; 21 22 public class MyHibernate 23 { 24 private class Column 25 { 26 private String name; 27 private String length; 28 } 29 30 private class Generator 31 { 32 private String type; 33 } 34 35 private class Id 36 { 37 private String name; 38 private String type; 39 40 private Column column; 41 private Generator generator; 42 } 43 private class Property 44 { 45 private String name; 46 private String type; 47 private Column column; 48 } 49 50 private class Class 51 { 52 private String name; 53 private String table; 54 private String catalog; 55 56 private Id id; 57 private List<Property> allProperties; 58 } 59 private class KeyValuePair 60 { 61 private String type; 62 private Object value; 63 64 private KeyValuePair(String type,Object value) 65 { 66 this.type = type; 67 this.value =http://www.mamicode.com/ value; 68 } 69 } 70 71 private String configPath ; 72 private Object obj = null; 73 private Document document; 74 private Class voClass ; 75 private String sql; 76 private List<KeyValuePair> allValues; 77 78 79 public MyHibernate(String configPath) 80 { 81 this.configPath = configPath; 82 } 83 84 public int save(Object obj) throws Exception 85 { 86 this.obj = obj; 87 this.load(); 88 this.format(); 89 this.generateSql(); 90 return this.saveToDb(); 91 } 92 93 public void load() throws Exception 94 { 95 SAXReader saxReader = new SAXReader(); 96 this.document = saxReader.read(new File(this.configPath)); 97 } 98 public void format() throws Exception 99 { 100 Class _class = new Class(); 101 org.dom4j.Element root = this.document.getRootElement(); 102 Element classElement = root.element("class"); 103 _class.name = classElement.attributeValue("name"); 104 _class.table = classElement.attributeValue("table"); 105 _class.catalog = classElement.attributeValue("catalog"); 106 107 Element idElement = classElement.element("id"); 108 _class.id = new Id(); 109 _class.id.name = idElement.attributeValue("name"); 110 _class.id.type = idElement.attributeValue("type"); 111 112 Element columnElementInId = idElement.element("column"); 113 _class.id.column = new Column(); 114 _class.id.column.name = columnElementInId.attributeValue("name"); 115 _class.id.column.length = columnElementInId.attributeValue("length"); 116 117 Element generatorElement = idElement.element("generator"); 118 _class.id.generator = new Generator(); 119 _class.id.generator.type = generatorElement.attributeValue("class"); 120 121 List<Property> allProperties = new ArrayList<MyHibernate.Property>(); 122 _class.allProperties = allProperties; 123 124 List<Element> allPropertiesElements = classElement.elements("property"); 125 for(Element item : allPropertiesElements) 126 { 127 Property property = new Property(); 128 property.name = item.attributeValue("name"); 129 property.type = item.attributeValue("type"); 130 131 Element columnElement = item.element("column"); 132 property.column = new Column(); 133 property.column.name = columnElement.attributeValue("name"); 134 property.column.length = columnElement.attributeValue("length"); 135 allProperties.add(property); 136 //System.out.println("name: " + property.name); 137 } 138 this.voClass = _class; 139 } 140 141 public void generateSql() throws Exception 142 { 143 this.allValues = new ArrayList<MyHibernate.KeyValuePair>(); 144 StringBuffer columns = new StringBuffer(); 145 StringBuffer values = new StringBuffer(); 146 StringBuffer sql = new StringBuffer(" INSERT INTO "); 147 sql.append(this.voClass.table).append("( "); 148 149 if("assigned".equals(this.voClass.id.generator.type)) 150 { 151 //需要用户提供主键列 152 columns.append(this.voClass.id.column.name).append(","); 153 values.append("?,"); 154 155 Field field = this.obj.getClass().getDeclaredField(this.voClass.id.name); 156 field.setAccessible(true); 157 this.allValues.add(new KeyValuePair(this.voClass.id.type, field.get(this.obj))); 158 } 159 160 for(Property property : this.voClass.allProperties) 161 { 162 columns.append(property.column.name).append(","); 163 values.append("?,"); 164 165 Field field = this.obj.getClass().getDeclaredField(property.name); 166 field.setAccessible(true); 167 this.allValues.add(new KeyValuePair(property.type, field.get(this.obj))); 168 } 169 170 columns.delete(columns.length()-1, columns.length()); 171 values.delete(values.length()-1, columns.length()); 172 173 sql.append(columns.toString()).append(") VALUES ("); 174 sql.append(values.toString()).append(")"); 175 this.sql = sql.toString(); 176 System.out.println(this.sql); 177 } 178 179 public int saveToDb() throws Exception 180 { 181 java.lang.Class.forName("org.gjt.mm.mysql.Driver"); 182 Connection conn = DriverManager.getConnection("jdbc:mysql://localhost:3306/hedb", "root", "admin"); 183 java.sql.PreparedStatement ps = conn.prepareStatement(this.sql); 184 185 for(int i = 0; i < this.allValues.size(); i++) 186 { 187 //System.out.println("***type:" + this.allValues.get(i).type); 188 if(String.class.getName().toString().equals(this.allValues.get(i).type)) 189 { 190 ps.setString(i+1, (String)this.allValues.get(i).value); 191 } 192 else if(Integer.class.getName().toString().equals(this.allValues.get(i).type)) 193 { 194 ps.setInt(i+1, (Integer)this.allValues.get(i).value); 195 }else if(Double.class.getName().toString().equals(this.allValues.get(i).type)) 196 { 197 ps.setDouble(i+1, (Double)this.allValues.get(i).value); 198 }else if(Date.class.getName().toString().equals(this.allValues.get(i).type)) 199 { 200 ps.setDate(i+1, new java.sql.Date(((Date)this.allValues.get(i).value).getTime())); 201 } 202 203 //System.out.println("设置第" + (i+1) + "个值("+this.allValues.get(i).type+","+this.allValues.get(i).value+")"); 204 } 205 return ps.executeUpdate(); 206 207 } 208 }
测试代码:
1 package hp.test; 2 3 import java.util.Date; 4 5 import org.hibernate.Session; 6 import org.hibernate.SessionFactory; 7 8 import hp.pojo.Member; 9 import hp.utils.HibernateHelper; 10 11 public class Main 12 { 13 14 public static void main(String[] args) 15 { 16 Member vo = new Member(); 17 vo.setMid("admin4"); 18 vo.setAge(24); 19 vo.setBirthday(new Date()); 20 vo.setName("sheldon4"); 21 vo.setNote("a good person4"); 22 vo.setSalary(44444.44); 23 24 try 25 { 26 MyHibernate mh = new MyHibernate("C:\\D\\code\\resource\\mapping.xml"); 27 int count = mh.save(vo); 28 System.out.println("成功插入" + count + "行数据"); 29 } 30 catch(Exception e) 31 { 32 e.printStackTrace(); 33 34 } 35 System.out.println("main done//~"); 36 } 37 38 }
模拟Hibernate动态生成SQL语句
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。