001/**
002 * Copyright (c) 2022-2023, Mybatis-Flex (fuhai999@gmail.com).
003 * <p>
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 * <p>
008 * http://www.apache.org/licenses/LICENSE-2.0
009 * <p>
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016package com.mybatisflex.core.keygen;
017
018import com.mybatisflex.core.FlexConsts;
019import com.mybatisflex.core.util.CollectionUtil;
020import org.apache.ibatis.executor.Executor;
021import org.apache.ibatis.executor.keygen.KeyGenerator;
022import org.apache.ibatis.mapping.MappedStatement;
023
024import java.sql.Statement;
025import java.util.List;
026import java.util.Map;
027
028/**
029 * 多实体主键生成器，用于批量插入的场景
030 */
031public class MultiEntityKeyGenerator implements KeyGenerator {
032
033    private KeyGenerator keyGenerator;
034
035    public MultiEntityKeyGenerator(KeyGenerator keyGenerator) {
036        this.keyGenerator = keyGenerator;
037    }
038
039    @Override
040    public void processBefore(Executor executor, MappedStatement ms, Statement stmt, Object parameter) {
041        List<Object> entities = (List<Object>) ((Map) parameter).get(FlexConsts.ENTITIES);
042        if (CollectionUtil.isNotEmpty(entities)) {
043            for (Object entity : entities) {
044                ((Map) parameter).put(FlexConsts.ENTITY, entity);
045                keyGenerator.processBefore(executor, ms, stmt, parameter);
046            }
047        }
048    }
049
050
051    @Override
052    public void processAfter(Executor executor, MappedStatement ms, Statement stmt, Object parameter) {
053        // do nothing
054        // 多条数据批量插入的场景下，不支持后设置主键
055        // 比如 INSERT INTO `tb_account`(uuid,name,sex) VALUES (?, ?, ?), (?, ?, ?)
056    }
057}