com.mars.mj.helper.templete.BaseSelect Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mars-mj Show documentation

There is a newer version: 3.0.3

package com.mars.mj.helper.templete;

import com.alibaba.fastjson.JSONObject;
import com.mars.mj.helper.base.DBHelper;
import com.mars.mj.manager.ConnectionManager;
import com.mars.mj.model.SqlBuilderModel;

import java.sql.Connection;
import java.util.ArrayList;
import java.util.List;

/**
 * 查询父类
 */
public class BaseSelect {

    /**
     * 有参查询列表，指定返回类型
     * @param sql sql语句
     * @param param 参数
     * @param cls 返回类型
     * @param dataSourceName 连接名
     * @return 数据
     */
    protected static  List selectList(String sql, Object param, Class cls,String dataSourceName) throws Exception {
        ConnectionManager connectionManager = BaseJdbcTemplete.getConnection(dataSourceName);
        try {
            List result = select(sql, param, connectionManager.getConnection());
            if (result != null && result.size() > 0) {
                List resultList = new ArrayList<>();
                for (JSONObject item : result) {
                    resultList.add(item.toJavaObject(cls));
                }
                return resultList;
            }
        } catch (Exception e) {
            throw e;
        } finally {
            connectionManager.close();
        }
        return null;
    }

    /**
     * 查询
     *
     * @param args
     * @param connection
     * @return
     * @throws Exception
     */
    protected static List select(String sql, Object args, Connection connection) throws Exception {
        List result = null;
        if (args != null) {
            if (args instanceof Object[]) {
                result = DBHelper.selectList(sql, connection, (Object[]) args);
            } else {
                SqlBuilderModel sqlBuilderModel = BaseJdbcTemplete.builderSql(sql, args);
                result = DBHelper.selectList(sqlBuilderModel.getSql(), connection, sqlBuilderModel.getParams());
            }
        } else {
            result = DBHelper.selectList(sql, connection);
        }
        return result;
    }
}