org.dromara.trans.extend.MybatisPlusTransableAdapter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of easy-trans-mybatis-plus-extend Show documentation

There is a newer version: 3.1.1

Show newest version

package org.dromara.trans.extend;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.enums.SqlMethod;
import com.baomidou.mybatisplus.core.toolkit.GlobalConfigUtils;
import com.baomidou.mybatisplus.extension.toolkit.SqlHelper;
import org.dromara.core.trans.vo.VO;
import org.dromara.trans.service.AutoTransable;
import org.apache.ibatis.session.SqlSession;
import org.mybatis.spring.SqlSessionUtils;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class MybatisPlusTransableAdapter implements AutoTransable {

    private Class voClass;

    public MybatisPlusTransableAdapter(Class voClass) {
        this.voClass = voClass;
    }


    @Override
    public List findByIds(List ids) {
        Map map = new HashMap<>(1);
        map.put("coll", ids);
        SqlSession sqlSession = this.sqlSession();
        try {
            return sqlSession.selectList(this.sqlStatement(SqlMethod.SELECT_BATCH_BY_IDS), map);
        } finally {
            this.closeSqlSession(sqlSession);
        }
    }

    @Override
    public List select() {
        Map map = new HashMap<>(1);
        map.put("ew", new LambdaQueryWrapper<>());
        SqlSession sqlSession = this.sqlSession();
        try {
            return sqlSession.selectList(this.sqlStatement(SqlMethod.SELECT_LIST), map);
        } finally {
            this.closeSqlSession(sqlSession);
        }
    }

    @Override
    public VO selectById(Object primaryValue) {
        SqlSession sqlSession = this.sqlSession();
        VO result;
        try {
            result = (VO) sqlSession.selectOne(this.sqlStatement(SqlMethod.SELECT_BY_ID), primaryValue);
        } finally {
            this.closeSqlSession(sqlSession);
        }
        return result;
    }

    protected SqlSession sqlSession() {
        return SqlHelper.sqlSession(this.voClass);
    }

    protected String sqlStatement(SqlMethod sqlMethod) {
        return this.sqlStatement(sqlMethod.getMethod());
    }

    protected String sqlStatement(String sqlMethod) {
        return SqlHelper.table(voClass).getSqlStatement(sqlMethod);
    }

    protected void closeSqlSession(SqlSession sqlSession) {
        SqlSessionUtils.closeSqlSession(sqlSession, GlobalConfigUtils.currentSessionFactory(this.voClass));
    }
}