org.zodiac.mybatisplus.injector.MyBatisPlusSqlInjector Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of zodiac-mybatis-plus Show documentation

Zodiac MyBatis Plus

There is a newer version: 1.6.8

package org.zodiac.mybatisplus.injector;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
import com.baomidou.mybatisplus.extension.injector.methods.InsertBatchSomeColumn;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import org.zodiac.mybatisplus.injector.methods.InsertBatch;
import org.zodiac.mybatisplus.injector.methods.InsertIgnore;
import org.zodiac.mybatisplus.injector.methods.Replace;
import org.zodiac.mybatisplus.injector.methods.SaveOrUpdate;
import org.zodiac.mybatisplus.injector.methods.UpdateBatchById;

public class MyBatisPlusSqlInjector extends DefaultSqlInjector {

    public MyBatisPlusSqlInjector() {
        super();
    }

    @Override
    public List getMethodList(Class mapperClass) {
        List methodArrayList = new ArrayList<>();
        methodArrayList.addAll(super.getMethodList(mapperClass));
        methodArrayList.addAll(
            Stream.of(new InsertBatch(), new InsertIgnore(), new Replace(), new SaveOrUpdate(), new UpdateBatchById(),
                new InsertBatchSomeColumn(i -> i.getFieldFill() != FieldFill.UPDATE)).collect(Collectors.toList()));
        return methodArrayList;
    }

}