手写一个orm框架-5

手写一个orm框架-5

生成sql：insert

思路

新增的sql还是比较好实现的，我们需要的大致就是：

1. 构建一个对象 User。
2. 调用新增的方法，将User作为参数传入方法。
3. 通过上一篇的解析结果，拿到所有的字段名称，与要保存的值。生成sql。
4. 通过JdbcTemplate执行sql，插入数据库。

实现

首先我们要根据User.java拿到所有的表的字段个名称，和对应的值。就是上一篇写到的：EntityTableRowMapper

1. 拿到字段和class属性的值 Map<String, Field> columnFieldMapper = entityTableRowMapper.getColumnFieldMapper(); insertColumns = new ArrayList(columnFieldMapper.size()); for (Map.Entry<String, Field> stringFieldEntry : columnFieldMapper.entrySet()) { Field field = stringFieldEntry.getValue(); Object value = EntityUtils.getValue(entity, field); if (value == null) { continue; } insertColumns.add(stringFieldEntry.getKey()); insertColumnValues.add(value); } 这里有两个变量： insertColumns：sql中的字段名。 insertColumnValues：sql中的字段对应的值。
2. 生成插入的sql： StringBuilder builder = new StringBuilder(); int size = insertColumns.size(); builder.append("INSERT INTO ").append(getTableName()).append(StringUtils.SPACE); builder.append(StringUtils.append("( ", StringUtils.join(insertColumns, ", "), " ) ")); builder.append("VALUES "); for (int i = 0; i < insertCount; i++) { builder.append("( "); String[] repeat = StringUtils.repeat("?", size); builder.append(StringUtils.join(Arrays.asList(repeat), ", ")); builder.append(" )"); if (i != insertCount - 1) { builder.append(StringUtils.COMMA); } } builder.append(";");
3. 生成的结果： //user User user = new User(); user.setId(10); user.setCreateDate(new Date()); user.setAge(20); user.setMark("ceshi"); user.setName("heiheihei"); //sql INSERT INTO user ( name, id, create_date, age, mark, status ) VALUES ( ?, ?, ?, ?, ?, ? ); //value [heiheihei, 10, Tue Jan 22 16:33:00 CST 2019, 20, ceshi, 0]
4. 现在可以拿着生成的sql和值去执行啦～ jdbcTemplate.update(sql, insertColumnValues.toArray());

​