JoinSqlInjector.java 1.3 KB

12345678910111213141516171819202122232425262728293031323334
  1. package com.poyee.config;
  2. import com.baomidou.mybatisplus.core.injector.AbstractMethod;
  3. import com.baomidou.mybatisplus.core.injector.AbstractSqlInjector;
  4. import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
  5. import com.baomidou.mybatisplus.core.metadata.TableInfo;
  6. import com.baomidou.mybatisplus.extension.injector.methods.InsertBatchSomeColumn;
  7. import com.github.yulichang.injector.MPJSqlInjector;
  8. import com.github.yulichang.interceptor.MPJInterceptor;
  9. import com.github.yulichang.method.*;
  10. import org.springframework.context.annotation.Configuration;
  11. import org.springframework.stereotype.Component;
  12. import java.util.ArrayList;
  13. import java.util.List;
  14. @Component
  15. public class JoinSqlInjector extends AbstractSqlInjector {
  16. @Override
  17. public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
  18. List<AbstractMethod> methodList = new ArrayList<>();
  19. //添加你的方法
  20. methodList.add(new InsertBatchSomeColumn());
  21. //多表查询sql注入 从连表插件里移植过来的
  22. methodList.add(new SelectJoinOne());
  23. methodList.add(new SelectJoinList());
  24. methodList.add(new SelectJoinPage());
  25. methodList.add(new SelectJoinMap());
  26. methodList.add(new SelectJoinMaps());
  27. methodList.add(new SelectJoinMapsPage());
  28. return methodList;
  29. }
  30. }