Fescar - RM InsertExecutor介绍

简介: 开篇 这篇文章的目的是讲解RM Executor模块当中一些通用的方法,这些方法在各个Executor的父类当中实现的,各个子类Executor模块都会复用,因此抽取出来统一的进行讲解。 个人是认为抽取通用的内容放在一篇文章讲解完后可以针对每类Executor讲解特有的功能,这样能够有更好的理解。

开篇

 这篇文章的目的是讲解RM Executor模块当中一些通用的方法,这些方法在各个Executor的父类当中实现的,各个子类Executor模块都会复用,因此抽取出来统一的进行讲解。

 个人是认为抽取通用的内容放在一篇文章讲解完后可以针对每类Executor讲解特有的功能,这样能够有更好的理解。这篇文章讲解Executor的实现类InsertExecutor。


类依赖图


说明:

  • 着重讲解InsertExecutor实现类。


InsertExecutor方法介绍

public class InsertExecutor<T, S extends Statement> extends AbstractDMLBaseExecutor<T, S> {
 
   protected TableRecords beforeImage() throws SQLException {
        return TableRecords.empty(getTableMeta());
    }

}

public class TableRecords {

    public static TableRecords empty(TableMeta tableMeta) {
        return new TableRecords(tableMeta) {
            @Override
            public int size() {
                return 0;
            }

            @Override
            public List<Field> pkRows() {
                return new ArrayList<>();
            }

            @Override
            public void add(Row row) {
                throw new UnsupportedOperationException("xxx");
            }

            @Override
            public TableMeta getTableMeta() {
                throw new UnsupportedOperationException("xxx");
            }
        };
    }
}
AI 代码解读

说明:

  • Insert操作的执行前镜像顾名思义是空的,所以beforeImage()方法返回空记录
  • TableRecords.empty返回的是一个没有记录的TableRecords对象。


public class InsertExecutor<T, S extends Statement> extends AbstractDMLBaseExecutor<T, S> {

    protected TableRecords afterImage(TableRecords beforeImage) throws SQLException {
        SQLInsertRecognizer recogizier = (SQLInsertRecognizer)sqlRecognizer;
        List<String> insertColumns = recogizier.getInsertColumns();
        TableMeta tmeta = getTableMeta();
        TableRecords afterImage = null;
        if (tmeta.containsPK(insertColumns)) {
            // 处理插入数据包含主键的情况
            List<Object> pkValues = null;
            String pk = tmeta.getPkName();
            for (int paramIdx = 0; paramIdx < insertColumns.size(); paramIdx++) {
                if (insertColumns.get(paramIdx).equalsIgnoreCase(pk)) {
                    // 处理PreparedStatement类型,从参数列表中提取
                    if (statementProxy instanceof PreparedStatementProxy) {
                        pkValues = ((PreparedStatementProxy) statementProxy).getParamsByIndex(paramIdx);
                    } else {
                        // 处理Statement类型,直接从SQL语句中提取
                        List<List<Object>> insertRows = recogizier.getInsertRows();
                        pkValues = new ArrayList<>(insertRows.size());
                        
                        for (List<Object> row : insertRows) {
                            pkValues.add(row.get(paramIdx));
                        }
                    }

                    //主键只有一个
                    break;
                }
            }
            if (pkValues == null) {
                throw new ShouldNeverHappenException();
            }
            afterImage = getTableRecords(pkValues);

        } else {
            // 处理主键自动生成的场景,也就是不包含主键的情况。
            Map<String, ColumnMeta> pkMetaMap = getTableMeta().getPrimaryKeyMap();
            if (pkMetaMap.size() != 1) {
                throw new NotSupportYetException();
            }

            // 获取primary key的元数据信息
            ColumnMeta pkMeta = pkMetaMap.values().iterator().next();
            if (!pkMeta.isAutoincrement()) {
                throw new ShouldNeverHappenException();
            }

            // 获取generatedKeys()获取生成的主键
            ResultSet genKeys = null;
            try {
                genKeys = statementProxy.getTargetStatement().getGeneratedKeys();
            } catch (SQLException e) {
               
                if ("S1009".equalsIgnoreCase(e.getSQLState())) {
                    genKeys = statementProxy.getTargetStatement().executeQuery("SELECT LAST_INSERT_ID()");
                } else {
                    throw e;
                }
            }
            List<Object> pkValues = new ArrayList<>();
            while (genKeys.next()) {
                Object v = genKeys.getObject(1);
                pkValues.add(v);
            }

            // 生成执行后镜像
            afterImage = getTableRecords(pkValues);

        }

        if (afterImage == null) {
            throw new SQLException("Failed to build after-image for insert");
        }

        return afterImage;
    }
}
AI 代码解读

说明:

  • afterImage准备执行后的镜像,针对Insert操作就是新增的记录。
  • afterImage处理分为两种场景,分为插入记录包含主键字段和不包含主键字段
  • 记录包含主键字段场景下,区分preparedStatement和Statement两种场景,前者去传入参数中获取,后者从执行的SQL当中解析。
  • 不包含主键字段场景下,通过获取getGeneratedKeys()返回的值获取。
  • 获取主键的值以后pkValues,然后通过getTableRecords()生成执行后镜像


public class InsertExecutor<T, S extends Statement> extends AbstractDMLBaseExecutor<T, S> {

    private TableRecords getTableRecords(List<Object> pkValues) throws SQLException {
        TableRecords afterImage;
        String pk = getTableMeta().getPkName();
        StringBuffer selectSQLAppender = new StringBuffer("SELECT * FROM " + getTableMeta().getTableName() + " WHERE ");
        for (int i = 1; i <= pkValues.size(); i++) {
            selectSQLAppender.append(pk + "=?");
            if (i < pkValues.size()) {
                selectSQLAppender.append(" OR ");
            }
        }
        PreparedStatement ps = null;
        ResultSet rs = null;
        try {
            ps = statementProxy.getConnection().prepareStatement(selectSQLAppender.toString());

            for (int i = 1; i <= pkValues.size(); i++) {
                ps.setObject(i, pkValues.get(i - 1));
            }

            rs = ps.executeQuery();
            afterImage = TableRecords.buildRecords(getTableMeta(), rs);

        } finally {
            if (rs != null) {
                rs.close();
            }
            if (ps != null) {
                ps.close();
            }
        }
        return afterImage;
    }
}
AI 代码解读

说明:

  • getTableRecords通过针对主键信息拼接SELECT的SQL语句拼接成根据主键查询字段信息的SQL
  • 通过buildRecords()方法针对查询结果ResultSet进行处理。
  • 生成执行后镜像信息在buildRecords()内部实现。


public class TableRecords {

    public static TableRecords buildRecords(TableMeta tmeta, ResultSet resultSet) throws SQLException {
        TableRecords records = new TableRecords(tmeta);
        ResultSetMetaData resultSetMetaData = resultSet.getMetaData();
        int columnCount = resultSetMetaData.getColumnCount();

        while (resultSet.next()) {
            List<Field> fields = new ArrayList<>(columnCount);
            for (int i = 1; i <= columnCount; i++) {
                String colName = resultSetMetaData.getColumnName(i);
                ColumnMeta col = tmeta.getColumnMeta(colName);
                Field field = new Field();
                field.setName(col.getColumnName());
                if (tmeta.getPkName().equals(field.getName())) {
                    field.setKeyType(KeyType.PrimaryKey);
                }
                field.setType(col.getDataType());
                field.setValue(resultSet.getObject(i));
                fields.add(field);
            }

            Row row = new Row();
            row.setFields(fields);

            records.add(row);
        }
        return records;
    }
}
AI 代码解读

说明:

  • buildRecords内部遍历查询的结果凭借Row对象。
  • 插入的数据可能有多条,外层while循环获取所有的行数,内层for循环遍历所有的列数。


期待

InsertExecutor生成执行前后镜像的过程分析完后,会接着分析DeleteExecutor。额外的收货就是分析完对于SQL或者说对阿里开源的druid有了一些深入研究的兴趣。

目录
打赏
0
0
0
0
1184
分享
相关文章
java基础:IO流 理论与代码示例(详解、idea设置统一utf-8编码问题)
这篇文章详细介绍了Java中的IO流,包括字符与字节的概念、编码格式、File类的使用、IO流的分类和原理,以及通过代码示例展示了各种流的应用,如节点流、处理流、缓存流、转换流、对象流和随机访问文件流。同时,还探讨了IDEA中设置项目编码格式的方法,以及如何处理序列化和反序列化问题。
240 1
java基础:IO流 理论与代码示例(详解、idea设置统一utf-8编码问题)
如何查看Redis的用户名和密码
【7月更文挑战第29天】
3168 3
西门子S7-1200 CPU型号及模块类型有哪些
上篇文章我们介绍了西门子S7-1200功能特点及应用范围有哪些,今天我为大家简单介绍一下西门子S7-1200的CPU型号及模块类型。西门子S7-1200作为紧凑型自动化产品的新成员,目前有三款CPU,分别是CPU1211C、CPU1212C和CPU1214C。根据电源和输入输出信号的不同,每款CPU各有三种不同的型号,不同型号的CPU,本机自带输入输出数字量的点数有所差异。CPU1211C不支持信号扩展模块,而CPU1212C支持两个,CPU1214C最多支持八个。
西门子S7-1200 CPU型号及模块类型有哪些
码蹄杯语言基础:结构体(C语言)
码蹄杯语言基础:结构体(C语言)
165 0
Android平台GB28181设备接入、RTMP推送模块如何实现高效率的视频编码
我们在做Android平台RTMP推送、轻量级RTSP服务和GB28181设备接入模块的时候,有一个点是逃不掉的:如何高效率的实现视频数据编码?
234 0
高性能数据库
Redis(Remote Dictionary Server)是一个开源的高性能键值对存储系统,它以其快速、灵活、可靠的特性在互联网应用中得到了广泛应用。Redis不仅仅是一个内存数据库,还具备了丰富的数据结构和功能,包括字符串、哈希表、列表、集合、有序集合等。本文将深入探索Redis的背景、特点、功能以及在互联网应用中的重要性。
证照之星XE新版本下载及功能
证照之星XE版本全新上线,新增了500+的证件照规格仓库,100套服装素材等,还有全新的AI算法加持,让证件照制作变得更加简单。
412 0
AI助理
登录插画

登录以查看您的控制台资源

管理云资源
状态一览
快捷访问

你好,我是AI助理

可以解答问题、推荐解决方案等