x7.repository.dao.DaoImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of x7-repository Show documentation
http://x7.xream.io
There is a newer version: 2.2.7.RELEASE
/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package x7.repository.dao;

import org.springframework.beans.factory.annotation.Autowired;
import x7.core.bean.*;
import x7.core.bean.condition.InCondition;
import x7.core.bean.condition.ReduceCondition;
import x7.core.bean.condition.RefreshCondition;
import x7.core.repository.X;
import x7.core.util.*;
import x7.core.web.Direction;
import x7.core.web.Page;
import x7.repository.CriteriaParser;
import x7.core.config.ConfigAdapter;
import x7.repository.util.ResultSetUtil;
import x7.repository.exception.PersistenceException;
import x7.repository.exception.RollbackException;
import x7.repository.mapper.Mapper;
import x7.repository.mapper.MapperFactory;
import x7.repository.util.ResultSortUtil;

import java.io.IOException;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.math.BigDecimal;
import java.sql.*;
import java.util.*;

/**
 * @author Sim
 */
public class DaoImpl implements Dao {

    private static DaoImpl instance;

    public static DaoImpl getInstance() {
        if (instance == null) {
            instance = new DaoImpl();
        }
        return instance;
    }

    private DaoImpl() {
        this.criteriaParser = new SqlCriteriaParser();
    }

    private CriteriaParser criteriaParser;

    @Autowired
    private Mapper.Dialect dialect;

    public void setDialect(Mapper.Dialect dialect) {
        this.dialect = dialect;
        this.criteriaParser.setDialect(dialect);
    }


    /**
     * 放回连接池,

     * 连接池已经重写了关闭连接的方法
     */
    private static void close(Connection conn) {
        RcDataSourceUtil.releaseConnection(conn);
    }

    private static void close(PreparedStatement pstmt) {
        if (pstmt != null) {
            try {
                pstmt.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    @Override
    public boolean createBatch(List objList) {

        if (objList.isEmpty())
            return false;
        Object obj = objList.get(0);
        Class clz = obj.getClass();

        String sql = MapperFactory.getSql(clz, Mapper.CREATE);

        List eles = MapperFactory.getElementList(clz);

        Connection conn = null;
        PreparedStatement pstmt = null;
        try {
            Parsed parsed = Parser.get(clz);

            Long keyOneValue = 0L;
            Field keyOneField = parsed.getKeyField(X.KEY_ONE);
            if (Objects.isNull(keyOneField))
                throw new PersistenceException("No setting of PrimaryKey by @X.Key");
            Class keyOneType = keyOneField.getType();
            if (keyOneType != String.class) {
                keyOneValue = keyOneField.getLong(obj);
            }

            conn = DataSourceUtil.getConnection();

            if (keyOneType != String.class && (keyOneValue == null || keyOneValue == 0)) {
                pstmt = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);
            } else {
                pstmt = conn.prepareStatement(sql);
            }

            for (Object o : objList) {

                int i = 1;
                for (BeanElement ele : eles) {

                    Object value = ele.getMethod.invoke(o);
                    if (value == null) {
                        if (ele.clz.isEnum())
                            throw new PersistenceException(
                                    "ENUM CAN NOT NULL, property:" + clz.getName() + SqlScript.POINT + ele.getProperty());
                        if (ele.clz == Boolean.class || ele.clz == Integer.class || ele.clz == Long.class
                                || ele.clz == Double.class || ele.clz == Float.class || ele.clz == BigDecimal.class
                                || ele.clz == Byte.class || ele.clz == Short.class)
                            value = 0;
                        pstmt.setObject(i++, value);
                    } else {
                        if (ele.isJson) {
                            String str = JsonX.toJson(value);
                            this.dialect.setJSON(i++, str, pstmt);
                        } else if (ele.clz.isEnum()) {
                            String str = value.toString();
                            pstmt.setObject(i++, str);
                        } else {
                            value = this.dialect.filterValue(value);
                            pstmt.setObject(i++, value);
                        }
                    }

                }

                pstmt.addBatch();

            }

            pstmt.executeBatch();

        } catch (Exception e) {
            e.printStackTrace();

            throw new RollbackException("RollbackException: " + ExceptionUtil.getMessage(e) + ", while create: " + obj);

        } finally {
            close(pstmt);
            DataSourceUtil.releaseConnection(conn);
        }

        return true;
    }

    protected boolean remove(Object obj, Connection conn) {

        Class clz = obj.getClass();

        String sql = MapperFactory.getSql(clz, Mapper.REMOVE);

        boolean flag = false;

        PreparedStatement pstmt = null;
        try {

            pstmt = conn.prepareStatement(sql);

            Parsed parsed = Parser.get(clz);

            int i = 1;

            Field keyOneField = parsed.getKeyField(X.KEY_ONE);
            if (Objects.isNull(keyOneField))
                throw new PersistenceException("No setting of PrimaryKey by @X.Key");
            SqlUtil.adpterSqlKey(pstmt, keyOneField, obj, i);

            flag = pstmt.executeUpdate() == 0 ? false : true;

        } catch (Exception e) {
            e.printStackTrace();
            throw new RollbackException("RollbackException: " + e.getMessage());

        } finally {
            close(pstmt);
            DataSourceUtil.releaseConnection(conn);
        }

        return flag;
    }

    protected long create(Object obj, Connection conn) {

        Class clz = obj.getClass();

        String sql = MapperFactory.getSql(clz, Mapper.CREATE);

        List eles = MapperFactory.getElementList(clz);

        long id = -1;
        PreparedStatement pstmt = null;
        try {
            Parsed parsed = Parser.get(clz);
            Long keyOneValue = 0L;
            Field keyOneField = parsed.getKeyField(X.KEY_ONE);
            if (Objects.isNull(keyOneField))
                throw new PersistenceException("No setting of PrimaryKey by @X.Key");
            Class keyOneType = keyOneField.getType();
            if (keyOneType == String.class) {
                keyOneValue = 1L;
            }else{
                keyOneValue = keyOneField.getLong(obj);
            }

            /*
             * 返回自增键
             */

            if (keyOneType != String.class && (keyOneValue == null || keyOneValue == 0)) {
                pstmt = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);
            } else {
                pstmt = conn.prepareStatement(sql);
            }

            int i = 1;
            for (BeanElement ele : eles) {

                Object value = ele.getMethod.invoke(obj);
                if (value == null) {
                    if (ele.clz.isEnum())
                        throw new PersistenceException(
                                "ENUM CAN NOT NULL, property:" + clz.getName() + SqlScript.POINT + ele.getProperty());
                    if (ele.clz == Boolean.class || ele.clz == Integer.class || ele.clz == Long.class
                            || ele.clz == Double.class || ele.clz == Float.class || ele.clz == BigDecimal.class
                            || ele.clz == Byte.class || ele.clz == Short.class)
                        value = 0;
                    pstmt.setObject(i++, value);
                } else {
                    if (ele.isJson) {
                        String str = JsonX.toJson(value);
                        this.dialect.setJSON(i++, str, pstmt);
                    } else if (ele.clz.isEnum()) {
                        String str = value.toString();
                        pstmt.setObject(i++, str);
                    } else {
                        value = this.dialect.filterValue(value);
                        pstmt.setObject(i++, value);
                    }

                }

            }

            pstmt.execute();

            if (keyOneType != String.class && (keyOneValue == null || keyOneValue == 0)) {
                ResultSet rs = pstmt.getGeneratedKeys();
                if (rs.next()) {
                    id = rs.getLong(1);
                }

            } else {
                id = keyOneValue;
            }

        } catch (Exception e) {
            System.out.println("Exception occured, while create: " + obj);
            e.printStackTrace();

            throw new RollbackException("RollbackException occoured: " + ExceptionUtil.getMessage(e) + ", while create " + obj);

        } finally {
            close(pstmt);
            DataSourceUtil.releaseConnection(conn);
        }

        return id;
    }

    protected boolean refresh(Object obj, Connection conn) {

        @SuppressWarnings("rawtypes")
        Class clz = obj.getClass();
        Parsed parsed = Parser.get(clz);

        String tableName = parsed.getTableName();

        StringBuilder sb = new StringBuilder();
        sb.append(SqlScript.UPDATE).append(SqlScript.SPACE).append(tableName).append(SqlScript.SPACE);

        Map refreshMap = BeanUtilX.getRefreshMap(parsed, obj);

        String sql = SqlUtil.concatRefresh(sb, parsed, refreshMap);

        if (ConfigAdapter.isIsShowSql())
            System.out.println("refresh normally: " + sql);

        if (sql.contains("SET  WHERE"))
            return false;

        boolean flag = false;

        PreparedStatement pstmt = null;
        try {

            pstmt = conn.prepareStatement(sql);

            int i = 1;
            for (Object value : refreshMap.values()) {
                value = this.dialect.filterValue(value);
                this.dialect.setObject(i++, value, pstmt);
            }

            /*
             * 处理KEY
             */
            Field keyOneField = parsed.getKeyField(X.KEY_ONE);
            if (Objects.isNull(keyOneField))
                throw new PersistenceException("No setting of PrimaryKey by @X.Key");
            SqlUtil.adpterSqlKey(pstmt, keyOneField, obj, i);

            flag = pstmt.executeUpdate() == 0 ? false : true;

        } catch (Exception e) {
            e.printStackTrace();

            throw new RollbackException("RollbackException: " + e.getMessage());

        } finally {
            close(pstmt);
            DataSourceUtil.releaseConnection(conn);
        }

        return flag;
    }

    @Override
    public long create(Object obj) {

        Connection conn = null;
        try {
            conn = DataSourceUtil.getConnection();
        } catch (Exception e) {
            throw new RuntimeException("NO CONNECTION");
        }
        return create(obj, conn);
    }

    @Override
    public boolean refresh(Object obj) {

        Connection conn = null;
        try {
            conn = DataSourceUtil.getConnection();
        } catch (Exception e) {
            throw new RuntimeException("NO CONNECTION");
        }
        return refresh(obj, conn);
    }

    @Override
    public boolean remove(Object obj) {
        Connection conn = null;
        try {
            conn = DataSourceUtil.getConnection();
        } catch (Exception e) {
            throw new RuntimeException("NO CONNECTION");
        }
        return remove(obj, conn);
    }

    protected  T get(Class clz, long idOne, Connection conn) {

        List list = new ArrayList();

        String sql = MapperFactory.getSql(clz, Mapper.QUERY);
        List eles = MapperFactory.getElementList(clz);

        if (ConfigAdapter.isIsShowSql())
            System.out.println(sql);

        PreparedStatement pstmt = null;
        BeanElement tempEle = null;
        try {
            conn.setAutoCommit(true);
            pstmt = conn.prepareStatement(sql);

            int i = 1;

            pstmt.setObject(i++, idOne);

            ResultSet rs = pstmt.executeQuery();

            if (rs != null) {
                while (rs.next()) {
                    T obj = clz.newInstance();
                    list.add(obj);
                    initObj(obj, rs, tempEle, eles);
                }
            }

        } catch (Exception e) {
            e.printStackTrace();
            throw new RollbackException(
                    "Exception occured by class = " + clz.getName() + ",column："+ tempEle!=null?(tempEle.property+"|"+tempEle.getMapper()):"" + ", message: " + ExceptionUtil.getMessage(e));

        } finally {
            close(pstmt);
            close(conn);
        }

        if (list.isEmpty())
            return null;
        return list.get(0);
    }

    @Override
    public  T get(Class clz, long idOne) {
        Connection conn = RcDataSourceUtil.getConnection();
        return get(clz, idOne, conn);
    }

    protected List> list(Class clz, String sql, List