手撸Mybatis（五）——连接数据库进行insert，update和delete

Ostrich2024-05-03 18:02

本专栏源码：yang-mybatis: 手撸简陋版的mybatis (gitee.com)

引言

在上一章中，我们成功实现了数据库的连接，以及单个字段的查询、resultType映射查询、resultMap映射查询。在本章，我们将讲解关于增加、修改和删除操作。

insert操作

首先，我们修改IUserMapper类，添加insertUser接口

json 复制代码 
package com.yang.mybatis.test;

public interface IUserMapper {
    String queryUserName(Integer id);

    Integer queryUserAge(Integer id);

    User queryUserById(Integer id);

    IdUserNameVO queryIdUserNameVOById(Integer id);

    void insertUser(User user);
}

修改UserMapper.xml，加上insertUser相关的xml块

json 复制代码 
 <insert id="insertUser" parameterType="com.yang.mybatis.test.User">
        insert into user( user_name, password, age, create_time)
        values( #{userName}, #{password}, #{age}, #{createTime})
    </insert>

我们注意到，对于每一个接口方法，其对应的xml块，类型有select，insert，update和delete这几种类型，此外，还有一个parameterType参数，因此，我们需要再次修改MybatisSqlStatement类，加上operateType和parameterType参数

json 复制代码 
public class MybatisSqlStatement implements Serializable {
    private String namespace;

    private String id;

    private String sql;

    private String resultType;

    private String resultMap;

    private String operateType;

    private String parameterType;
  
    ...省略getter 和setter
}

接着修改XmlMybatisMapperParser的parseStatement方法，在解析mapper.xml的时候，设置对应的值

json 复制代码 
 private void parseStatement(MybatisMapperXmlConfiguration mybatisMapperXmlConfiguration, List<Element> elements) {
        if (elements == null || elements.isEmpty()) {
            return;
        }
        String namespace = mybatisMapperXmlConfiguration.getMapperName();
        for (Element element : elements) {
            String id = element.attributeValue("id");
            String resultType = element.attributeValue("resultType");
            String resultMap = element.attributeValue("resultMap");
            String parameterType = element.attributeValue("parameterType");
            String sql = element.getText().trim();

            MybatisSqlStatement mybatisSqlStatement = new MybatisSqlStatement();
            mybatisSqlStatement.setNamespace(namespace);
            mybatisSqlStatement.setId(id);
            mybatisSqlStatement.setSql(sql);
            mybatisSqlStatement.setResultType(resultType);
            mybatisSqlStatement.setResultMap(resultMap);
            mybatisSqlStatement.setParameterType(parameterType);
            mybatisSqlStatement.setOperateType(element.getName().toLowerCase());

            mybatisMapperXmlConfiguration.addMybatisSqlStatement(mybatisSqlStatement);
        }
    }

定义IMybatisPreparedStatementBuilder接口，用于接收请求参数并设置参数值到preparedStatement

json 复制代码 
package com.yang.mybatis.execute;



import com.yang.mybatis.execute.request.MybatisPreparedStatementBuilderRequest;

import java.sql.PreparedStatement;
import java.sql.SQLException;

public interface IMybatisPreparedStatementBuilder {
    PreparedStatement buildPreparedStatement(MybatisPreparedStatementBuilderRequest mybatisPreparedStatementBuilderRequest) throws SQLException;
}

其中，MybatisPreparedStatementBuilderRequest的定义如下：

json 复制代码 
package com.yang.mybatis.execute.request;

import com.yang.mybatis.config.MybatisSqlStatement;

import java.io.Serializable;
import java.sql.Connection;

public class MybatisPreparedStatementBuilderRequest implements Serializable {
    private Connection connection;
    private Object[] parameters;
    private MybatisSqlStatement mybatisSqlStatement;
    private String operateType;

    public Connection getConnection() {
        return connection;
    }

    public void setConnection(Connection connection) {
        this.connection = connection;
    }

    public Object[] getParameters() {
        return parameters;
    }

    public void setParameters(Object[] parameters) {
        this.parameters = parameters;
    }

    public MybatisSqlStatement getMybatisSqlStatement() {
        return mybatisSqlStatement;
    }

    public void setMybatisSqlStatement(MybatisSqlStatement mybatisSqlStatement) {
        this.mybatisSqlStatement = mybatisSqlStatement;
    }

    public String getOperateType() {
        return operateType;
    }

    public void setOperateType(String operateType) {
        this.operateType = operateType;
    }
}

该接口的默认实现类为：

json 复制代码 
package com.yang.mybatis.execute;

import com.yang.mybatis.config.MybatisSqlStatement;
import com.yang.mybatis.execute.request.MybatisPreparedStatementBuilderRequest;
import org.apache.commons.lang3.StringUtils;

import java.lang.reflect.Field;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class DefaultMybatisPreparedStatementBuilder implements IMybatisPreparedStatementBuilder {
    @Override
    public PreparedStatement buildPreparedStatement(MybatisPreparedStatementBuilderRequest mybatisPreparedStatementBuilderRequest) throws SQLException {
        Connection connection = mybatisPreparedStatementBuilderRequest.getConnection();
        Object[] parameters = mybatisPreparedStatementBuilderRequest.getParameters();

        MybatisSqlStatement mybatisSqlStatement = mybatisPreparedStatementBuilderRequest.getMybatisSqlStatement();
        String rawSql = mybatisSqlStatement.getSql();

        List<String> parameterNameList = new ArrayList<>();
        String sql = extractRawSql(rawSql, parameterNameList);
        PreparedStatement preparedStatement = connection.prepareStatement(sql);

        String parameterType = mybatisSqlStatement.getParameterType();
        if (StringUtils.isEmpty(parameterType)) {
            if (parameterNameList.size() != parameters.length) {
                throw new RuntimeException("SQL语句参数个数不匹配====");
            }
            int index = 1;
            for (Object o : parameters) {
                preparedStatement.setObject(index ++, o);
            }
            return preparedStatement;
        }

        try {
            Object parameter = parameters[0];
            Class<?> aClass = Class.forName(parameterType);
            Field[] fields = aClass.getDeclaredFields();
            Map<String, Field> filedName2FieldMap = new HashMap<>();
            for (Field field : fields) {
                filedName2FieldMap.put(field.getName(), field);
            }

            List<Object> parameterValueList = new ArrayList<>();
            for (String parameterName : parameterNameList) {
                Field field = filedName2FieldMap.get(parameterName);
                if (field == null) {
                    throw new RuntimeException("SQL参数不存在");
                }
                field.setAccessible(true);
                Object value = field.get(parameter);
                parameterValueList.add(value);
            }

            int index = 1;
            for (Object o : parameterValueList) {
                preparedStatement.setObject(index++, o);
            }
            return preparedStatement;
        } catch (ClassNotFoundException | IllegalAccessException e) {
            e.printStackTrace();
            throw new RuntimeException(e);
        }
    }

    private String extractRawSql(String rawSql, List<String> parameterNameList) {
        StringBuilder sqlBuilder = new StringBuilder();
        int start = 0;
        int end = -1;
        while ((end = rawSql.indexOf("#", start)) != -1) {
            sqlBuilder.append(rawSql.substring(start, end - 1))
                    .append(" ? ");
            int parameterStart = end + 2;
            int parameterEnd = rawSql.indexOf("}", parameterStart);
            parameterNameList.add(rawSql.substring(parameterStart, parameterEnd));
            start = parameterEnd + 1;
        }
        sqlBuilder.append(rawSql.substring(start));

        return sqlBuilder.toString();
    }
}

然后，我们修改DefaultMybatisSqlSession类，在execute方法中我们先使用IMybatisPreparedStatementBuilder来构建PreparedStatement，然后执行相关操作，最后再通过IMybatisResultParser来对结果进行解析。

json 复制代码 
package com.yang.mybatis.session;

import com.yang.mybatis.config.MybatisConfiguration;
import com.yang.mybatis.config.MybatisEnvironment;
import com.yang.mybatis.config.MybatisMapperXmlConfiguration;
import com.yang.mybatis.config.MybatisSqlStatement;
import com.yang.mybatis.execute.DefaultMybatisPreparedStatementBuilder;
import com.yang.mybatis.execute.DefaultMybatisResultParser;
import com.yang.mybatis.execute.IMybatisPreparedStatementBuilder;
import com.yang.mybatis.execute.IMybatisResultParser;
import com.yang.mybatis.execute.request.MybatisPreparedStatementBuilderRequest;
import com.yang.mybatis.execute.request.MybatisResultParserRequest;
import com.yang.mybatis.proxy.MapperProxyFactory;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Map;


public class DefaultMybatisSqlSession implements IMybatisSqlSession {
    private MapperProxyFactory mapperProxyFactory;
    private MybatisConfiguration mybatisConfiguration;

    private IMybatisResultParser iMybatisResultParser;

    private IMybatisPreparedStatementBuilder iMybatisPreparedStatementBuilder;

    public DefaultMybatisSqlSession(MapperProxyFactory mapperProxyFactory, IMybatisPreparedStatementBuilder iMybatisPreparedStatementBuilder,
                                    IMybatisResultParser iMybatisResultParser) {
        this.mapperProxyFactory = mapperProxyFactory;
        this.mybatisConfiguration = mapperProxyFactory.getMybatisConfiguration();
        this.iMybatisPreparedStatementBuilder = iMybatisPreparedStatementBuilder;
        this.iMybatisResultParser = iMybatisResultParser;
    }

    @Override
    public <T> T execute(String method, Object parameter) {
        Map<String, MybatisSqlStatement> mapperMethod2SqlStatementsMap = mapperProxyFactory.getMybatisConfiguration().getMapperMethod2SqlStatementsMap();
        MybatisSqlStatement mybatisSqlStatement = mapperMethod2SqlStatementsMap.get(method);

        MybatisEnvironment defaultMybatisEnvironment = this.mybatisConfiguration.getDefaultMybatisEnvironment();

        return new TransactionInvoke<T>() {
            @Override
            public T execute(Connection connection) throws SQLException {
                Object[] parameters = (Object[]) parameter;
                String operateType = mybatisSqlStatement.getOperateType();

                MybatisPreparedStatementBuilderRequest mybatisPreparedStatementBuilderRequest = new MybatisPreparedStatementBuilderRequest();
                mybatisPreparedStatementBuilderRequest.setConnection(connection);
                mybatisPreparedStatementBuilderRequest.setMybatisSqlStatement(mybatisSqlStatement);
                mybatisPreparedStatementBuilderRequest.setParameters(parameters);
                mybatisPreparedStatementBuilderRequest.setOperateType(operateType);
                PreparedStatement preparedStatement = iMybatisPreparedStatementBuilder.buildPreparedStatement(mybatisPreparedStatementBuilderRequest);

                ResultSet resultSet = null;
                if ("select".equals(operateType)) {
                    resultSet = preparedStatement.executeQuery();
                } else {
                    preparedStatement.execute();
                }

                String mapperName = mybatisSqlStatement.getNamespace();
                MybatisMapperXmlConfiguration mybatisMapperXmlConfiguration = mybatisConfiguration.getMybatisMapperXmlConfiguration(mapperName);

                MybatisResultParserRequest mybatisResultParserRequest = new MybatisResultParserRequest();
                mybatisResultParserRequest.setResultSet(resultSet);
                mybatisResultParserRequest.setMybatisSqlStatement(mybatisSqlStatement);
                mybatisResultParserRequest.setMybatisMapperXmlConfiguration(mybatisMapperXmlConfiguration);
                mybatisResultParserRequest.setOperateType(operateType);
                T result = iMybatisResultParser.parseResult(mybatisResultParserRequest);
                if (resultSet != null) {
                    resultSet.close();
                }
                return result;
            }
        }.invoke(defaultMybatisEnvironment.getMybatisDataSource());
    }

    @Override
    public <T> T getMapper(Class<T> type) {
        return (T) mapperProxyFactory.newInstance(type, this);
    }
}

这里还需要修改DefaultMybatisSqlSessionFactory类，将IMybatisResultParser和IMybatisPreparedStatementBuilder的构建职责迁移于此，从而避免每一个SqlSession都需要重复创建这些类。

json 复制代码 
package com.yang.mybatis.session;

import com.yang.mybatis.execute.DefaultMybatisPreparedStatementBuilder;
import com.yang.mybatis.execute.DefaultMybatisResultParser;
import com.yang.mybatis.execute.IMybatisPreparedStatementBuilder;
import com.yang.mybatis.execute.IMybatisResultParser;
import com.yang.mybatis.proxy.MapperProxyFactory;

public class DefaultMybatisSqlSessionFactory implements IMybatisSqlSessionFactory {
    private MapperProxyFactory mapperProxyFactory;
    private IMybatisPreparedStatementBuilder iMybatisPreparedStatementBuilder = new DefaultMybatisPreparedStatementBuilder();
    private IMybatisResultParser iMybatisResultParser = new DefaultMybatisResultParser();

    public DefaultMybatisSqlSessionFactory(MapperProxyFactory mapperProxyFactory) {
        this.mapperProxyFactory = mapperProxyFactory;
    }

    @Override
    public IMybatisSqlSession openSession() {
        return new DefaultMybatisSqlSession(mapperProxyFactory, iMybatisPreparedStatementBuilder, iMybatisResultParser);
    }
}

添加测试代码，进行测试：

json 复制代码 
package com.yang.mybatis.test;

import com.yang.mybatis.config.parser.XmlMybatisConfigurationParser;
import com.yang.mybatis.config.parser.XmlMybatisMapperParser;
import com.yang.mybatis.session.IMybatisSqlSession;
import com.yang.mybatis.session.IMybatisSqlSessionFactory;
import com.yang.mybatis.session.MybatisSqlSessionFactoryBuilder;

import java.time.LocalDateTime;


public class Main {
    public static void main(String[] args) {
        String configPath = "mybatis-config.xml";
        IMybatisSqlSessionFactory mybatisSqlSessionFactory = new MybatisSqlSessionFactoryBuilder()
                .setMybatisMapperParser(new XmlMybatisMapperParser())
                .setMybatisConfigurationParser(new XmlMybatisConfigurationParser())
                .setConfigPath(configPath)
                .buildSqlSessionFactory();
        IMybatisSqlSession mybatisSqlSession = mybatisSqlSessionFactory.openSession();
        IUserMapper userMapper = mybatisSqlSession.getMapper(IUserMapper.class);

        User user = new User();
        user.setUserName("test");
        user.setPassword("test");
        user.setAge(4);
        user.setCreateTime(LocalDateTime.now());
        userMapper.insertUser(user);
    }

}

运行上述方法，再次查看数据库，结果如下：

update操作

首先，我们在IUserMapper接口上，添加一个updateUserById方法

json 复制代码 
    void updateUserById(User user);

然后修改UserMapper.xml，添加上updateUserById相关的xml块

json 复制代码 
 <update id="updateUserById" parameterType="com.yang.mybatis.test.User">
        update user
        set user_name = #{userName},
            password = #{password},
            age = #{age}
        where id = #{id}
    </update>

添加测试方法，进行测试

json 复制代码 
   String configPath = "mybatis-config.xml";
        MybatisSqlSessionFactory mybatisSqlSessionFactory = new MybatisSqlSessionFactoryBuilder()
                .setMybatisMapperParser(new XmlMybatisMapperParser())
                .setMybatisConfigurationParser(new XmlMybatisConfigurationParser())
                .setConfigPath(configPath)
                .buildSqlSessionFactory();
        IMybatisSqlSession mybatisSqlSession = mybatisSqlSessionFactory.openSession();
        IUserMapper userMapper = mybatisSqlSession.getMapper(IUserMapper.class);
        User user = userMapper.queryUserById(4);
        user.setUserName("cxy1");
        user.setPassword("1234");
        user.setAge(10);
        userMapper.updateUserById(user);

运行方法后，查看数据库，结果如下：

delete操作

首先，修改IUserMapper，添加deleteUserById方法

json 复制代码 
    void deleteUserById(Integer id);

然后在UserMapper.xml中添加上对应的xml块

json 复制代码 
<delete id="deleteUserById" parameterType="int">
        delete from user
        where id = #{id}
    </delete>

添加测试方法，进行测试：

json 复制代码 
 public static void main(String[] args) {
        String configPath = "mybatis-config.xml";
        IMybatisSqlSessionFactory mybatisSqlSessionFactory = new MybatisSqlSessionFactoryBuilder()
                .setMybatisMapperParser(new XmlMybatisMapperParser())
                .setMybatisConfigurationParser(new XmlMybatisConfigurationParser())
                .setConfigPath(configPath)
                .buildSqlSessionFactory();
        IMybatisSqlSession mybatisSqlSession = mybatisSqlSessionFactory.openSession();
        IUserMapper userMapper = mybatisSqlSession.getMapper(IUserMapper.class);
        userMapper.deleteUserById(5);
    }

结果如下:

可以看出，这里因为我们使用的是parameterType是int而不是java.lang.Integer，所以会报错，但是一般情况下，像int，float，long等小写，一般我们也会使用的，而且用的频率比java.lang.Integer的频率更大，那么，我们其实可以添加一个假名，将int这些小写的方式和对应的包装类关联起来。我们修改DefaultMybatisPreparedStatementBuilder，修改内容如下：

json 复制代码 
package com.yang.mybatis.execute;

import com.yang.mybatis.config.MybatisSqlStatement;
import com.yang.mybatis.execute.request.MybatisPreparedStatementBuilderRequest;
import org.apache.commons.lang3.StringUtils;

import java.lang.reflect.Field;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class DefaultMybatisPreparedStatementBuilder implements IMybatisPreparedStatementBuilder {
    private Map<String, String> baseAliasMap = new HashMap<>();

    {
        baseAliasMap.put("int", Integer.class.getName());
        baseAliasMap.put("float", Float.class.getName());
        baseAliasMap.put("double", Double.class.getName());
        baseAliasMap.put("long", Long.class.getName());
        baseAliasMap.put("byte", Byte.class.getName());
        baseAliasMap.put("short", Short.class.getName());
        baseAliasMap.put("string", String.class.getName());
        baseAliasMap.put("String", String.class.getName());
        baseAliasMap.put("char", Character.class.getName());
    }
    @Override
    public PreparedStatement buildPreparedStatement(MybatisPreparedStatementBuilderRequest mybatisPreparedStatementBuilderRequest) throws SQLException {
        Connection connection = mybatisPreparedStatementBuilderRequest.getConnection();
        Object[] parameters = mybatisPreparedStatementBuilderRequest.getParameters();

        MybatisSqlStatement mybatisSqlStatement = mybatisPreparedStatementBuilderRequest.getMybatisSqlStatement();
        String rawSql = mybatisSqlStatement.getSql();

        List<String> parameterNameList = new ArrayList<>();
        String sql = extractRawSql(rawSql, parameterNameList);
        PreparedStatement preparedStatement = connection.prepareStatement(sql);

        String parameterType = mybatisSqlStatement.getParameterType();
        if (StringUtils.isEmpty(parameterType)) {
            if (parameterNameList.size() != parameters.length) {
                throw new RuntimeException("SQL语句参数个数不匹配====");
            }
            int index = 1;
            for (Object o : parameters) {
                preparedStatement.setObject(index ++, o);
            }
            return preparedStatement;
        }

        try {
            Object parameter = parameters[0];
            if (baseAliasMap.containsKey(parameterType)) {
                if (parameters.length != 1) {
                    throw new RuntimeException("SQL语句有误, 只能有一个parameterType参数");
                }
                preparedStatement.setObject(1, parameter);
                return preparedStatement;
            }
            Class<?> aClass = Class.forName(parameterType);
            Field[] fields = aClass.getDeclaredFields();
            Map<String, Field> filedName2FieldMap = new HashMap<>();
            for (Field field : fields) {
                filedName2FieldMap.put(field.getName(), field);
            }

            List<Object> parameterValueList = new ArrayList<>();
            for (String parameterName : parameterNameList) {
                Field field = filedName2FieldMap.get(parameterName);
                if (field == null) {
                    throw new RuntimeException("SQL参数不存在");
                }
                field.setAccessible(true);
                Object value = field.get(parameter);
                parameterValueList.add(value);
            }

            int index = 1;
            for (Object o : parameterValueList) {
                preparedStatement.setObject(index++, o);
            }
            return preparedStatement;
        } catch (ClassNotFoundException | IllegalAccessException e) {
            throw new RuntimeException(e);
        }
    }

    private String extractRawSql(String rawSql, List<String> parameterNameList) {
        StringBuilder sqlBuilder = new StringBuilder();
        int start = 0;
        int end = -1;
        while ((end = rawSql.indexOf("#", start)) != -1) {
            sqlBuilder.append(rawSql.substring(start, end - 1))
                    .append(" ? ");
            int parameterStart = end + 2;
            int parameterEnd = rawSql.indexOf("}", parameterStart);
            parameterNameList.add(rawSql.substring(parameterStart, parameterEnd));
            start = parameterEnd + 1;
        }
        sqlBuilder.append(rawSql.substring(start));

        return sqlBuilder.toString();
    }
}

再次运行测试方法，这次没有报错了，然后我们查看数据库，结果如下，说明删除成功了。

相关推荐
代码小鑫1 分钟前
A031-基于SpringBoot的健身房管理系统设计与实现
java·开发语言·数据库·spring boot·后端
Json____6 分钟前
学法减分交管12123模拟练习小程序源码前端和后端和搭建教程
前端·后端·学习·小程序·uni-app·学法减分·驾考题库
monkey_meng26 分钟前
【Rust类型驱动开发 Type Driven Development】
开发语言·后端·rust
落落落sss34 分钟前
MQ集群
java·服务器·开发语言·后端·elasticsearch·adb·ruby
大鲤余1 小时前
Rust,删除cargo安装的可执行文件
开发语言·后端·rust
她说彩礼65万1 小时前
Asp.NET Core Mvc中一个视图怎么设置多个强数据类型
后端·asp.net·mvc
陈随易1 小时前
农村程序员-关于小孩教育的思考
前端·后端·程序员
_江南一点雨1 小时前
SpringBoot 3.3.5 试用CRaC，启动速度提升3到10倍
java·spring boot·后端
转转技术团队2 小时前
空间换时间-将查询数据性能提升100倍的计数系统实践
java·后端·架构
r0ad2 小时前
SpringCloud2023实战之接口服务测试工具SpringBootTest
spring boot·后端·spring cloud
热门推荐
01玄机平台应急响应—webshell查杀02Ubuntu 20.04使用Livox mid 360 测试 FAST_LIO03【HarmonyOS】HUAWEI DevEco Studio 下载地址汇总04新版微信小程序获取用户手机号05文件或文件夹名称中有空格如何批量去除06组基轨迹建模 GBTM的介绍与实现（Stata 或 R）072024最新 无人机 数据集（不定期更新）08优化手机性能，解决卡顿问题：关闭这3个微信开关，释放内存空间09基于YOLOv10深度学习的CT扫描图像肾结石智能检测系统【python源码+Pyqt5界面+数据集+训练代码】深度学习实战、目标检测10【TC3xx芯片】TC3xx芯片电源管理系统PMS详解