merage

kimi

2020-05-27 c007f0ca1785db093d48f4846cda82fe8e955765

 src/main/java/com/highdatas/mdm/util/DbUtils.java

@@ -1,23 +1,39 @@
package com.highdatas.mdm.util;


import com.baomidou.mybatisplus.annotations.TableName;
import com.highdatas.mdm.entity.Maintain;
import com.highdatas.mdm.entity.MaintainField;
import com.highdatas.mdm.entity.TUser;
import com.highdatas.mdm.entity.TableSchemaResult;
import com.alibaba.fastjson.JSONObject;
import com.google.common.collect.Maps;
import com.highdatas.mdm.entity.Character;
import com.highdatas.mdm.entity.*;
import com.highdatas.mdm.mapper.MaintainDetailMapper;
import com.highdatas.mdm.mapper.TableInfoMapper;
import com.highdatas.mdm.pojo.MasterAuthorType;
import com.highdatas.mdm.pojo.Segment;
import com.highdatas.mdm.service.*;
import com.highdatas.mdm.util.pool.MqClient;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.ParameterMapping;
import org.apache.ibatis.mapping.ParameterMode;
import org.apache.ibatis.reflection.MetaObject;
import org.apache.ibatis.reflection.property.PropertyTokenizer;
import org.apache.ibatis.scripting.xmltags.ForEachSqlNode;
import org.apache.ibatis.session.Configuration;
import org.apache.ibatis.session.SqlSessionFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cglib.beans.BeanMap;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.security.MessageDigest;
import java.text.MessageFormat;
import java.text.SimpleDateFormat;
import java.util.*;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;

/**
 * @author kimi
@@ -26,7 +42,103 @@
 */

@Slf4j
@Component
public class DbUtils {

    private static UserRoleClient userRoleClient;
    public static MaintainDetailMapper maintainDetailMapper;
    public static IMaintainService maintainService;
    public static IMaintainDetailService maintainDetailService;
    public static ISysFieldService fieldService;
    public static IMaintainFieldService maintainFieldService;
    public static TableInfoMapper tableInfoMapper;
    public static ISysViewService viewService;
    public static MasterDataService masterDataService;
    public static IMasterAuthorService masterAuthorService;
    public static ISysMenuService menuService;
    public static MqClient mqClient;

    public static ISysDispenseLogsService dispenseLogsService;
    public static DispenseService dispenseService;

    public static AntianaphylaxisClient antianaphylaxisClient;


    @Autowired
    public void setAntianaphylaxisClient(AntianaphylaxisClient antianaphylaxisClient){
        DbUtils.antianaphylaxisClient = antianaphylaxisClient;
    }



    @Autowired
    public void setDispenseService(DispenseService dispenseService){
        DbUtils.dispenseService = dispenseService;
    }
    @Autowired
    public void setISysDispenseLogsService(ISysDispenseLogsService dispenseLogsService){
        DbUtils.dispenseLogsService = dispenseLogsService;
    }

    @Autowired
    public void setMqClient(MqClient mqClient){
        DbUtils.mqClient = mqClient;
    }

    @Autowired
    public void setISysMenuService(ISysMenuService menuService){
        DbUtils.menuService = menuService;
    }
    @Autowired
    public void setIMasterAuthorService(IMasterAuthorService masterAuthorService){
        DbUtils.masterAuthorService = masterAuthorService;
    }

    @Autowired
    public void setMasterDataService(MasterDataService masterDataService){
        DbUtils.masterDataService = masterDataService;
    }

    @Autowired
    public void setISysViewService(ISysViewService viewService){
        DbUtils.viewService = viewService;
    }

    @Autowired
    public void setTableInfoMapper(TableInfoMapper tableInfoMapper){
        DbUtils.tableInfoMapper = tableInfoMapper;
    }

    @Autowired
    public void setIMaintainFieldService(IMaintainFieldService maintainFieldService){
        DbUtils.maintainFieldService = maintainFieldService;
    }

    @Autowired
    public void setISysFieldService(ISysFieldService fieldService){
        DbUtils.fieldService = fieldService;
    }

    @Autowired
    public void setIMaintainDetailService(IMaintainDetailService maintainDetailService){
        DbUtils.maintainDetailService = maintainDetailService;
    }

    @Autowired
    public void setIMaintainService(IMaintainService maintainService){
        DbUtils.maintainService = maintainService;
    }
    @Autowired
    public void setMaintainDetailMapper(MaintainDetailMapper maintainDetailMapper){
        DbUtils.maintainDetailMapper = maintainDetailMapper;
    }


    @Autowired
    public void setUserRoleClient(UserRoleClient userRoleClient){
        DbUtils.userRoleClient = userRoleClient;
    }

    private static final char HEX_DIGITS[] = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F' };

    public static String versionAddSub (String baseVersion) {
@@ -71,28 +183,25 @@
        return MessageFormat.format("V{0}.{1}",bigVersion, 0);
    }

    public static Integer getSubVersion(String baseVersion) {
        if (StringUtils.isEmpty(baseVersion)) {
            return null;
    public static <T> Map<String, Object> beanToMap(T bean) {
        Map<String, Object> map = Maps.newHashMap();
        if (bean != null) {
            BeanMap beanMap = BeanMap.create(bean);
            for (Object key : beanMap.keySet()) {
                map.put(key+"", beanMap.get(key));
            }
        }

        String[] split = baseVersion.split(Constant.Dot);
        if (split.length == 1) {
            return null;
        }
        return  Integer.valueOf(split[1]);
        return map;
    }
    public static Integer getBigVersion(String baseVersion) {
        if (StringUtils.isEmpty(baseVersion)) {
            return null;
        }

        String[] split = baseVersion.split(Constant.Dot);
        if (split.length == 0 || split[0].length() < 2) {
            return null;
        }
    public static HashMap<String, Object> JsonObjectToHashMap(JSONObject jsonObj){
        HashMap<String, Object> data = new HashMap<String, Object>();

        return  Integer.valueOf(split[0].substring(1));
        Set<String> keySet = jsonObj.keySet();
        for (String key : keySet) {
            data.put(key, jsonObj.get(key));
        }
        return data;
    }


@@ -110,29 +219,6 @@
        return segment.toString();
    }

    public static String getTableNameFromEntity(Object object) {
        Field[] fields = object.getClass().getDeclaredFields();
        for (Field field : fields) {
            field.setAccessible(true);
            // 获取表名
            TableName table = object.getClass().getAnnotation(TableName.class);
            if (table != null) {
                String tableName = table.value();
                return tableName;
            }
        }
        return null;
    }

    public static Object toSqlToJava(String value, TableSchemaResult tableSchema) throws ClassNotFoundException, NoSuchMethodException, IllegalAccessException, InstantiationException {
        String dbType = tableSchema.getDbType();
        String javaType = toSqlToJava(dbType);
        Class<?> baseWapperClaz = Class.forName("java.lang." + javaType);
        Object baseWapper = baseWapperClaz.newInstance();
        Method valueOfMethod = baseWapperClaz.getDeclaredMethod("valueOf");
        return null;
//        valueOfMethod.invoke(baseWapper, )
    }

    public static String toSqlToJava(String sqlType) {
        if( sqlType == null || sqlType.trim().length() == 0 ) return sqlType;
@@ -172,12 +258,6 @@
        return sqlType;
    }

    public static void sqlValueToJava(Object value, String dbType) throws Exception {
        String javaType = toSqlToJava(dbType);
        Class<?> baseWapperClaz = Class.forName("java.lang." + javaType);
        Object baseWapper = baseWapperClaz.newInstance();
        Method valueOfMethod = baseWapperClaz.getDeclaredMethod("valueOf");
    }

    public static String quotedStr(Object o) {
       return quotedStr(String.valueOf(o));
@@ -199,6 +279,14 @@

    public static String StrJoin(String... objects){
        ContentBuilder builder = new ContentBuilder(Constant.EMPTY);
        for (String object : objects) {
            builder.append(object);
        }
        return builder.toString();
    }

    public static String StrJoinLink(String link,String... objects){
        ContentBuilder builder = new ContentBuilder(link);
        for (String object : objects) {
            builder.append(object);
        }
@@ -238,6 +326,29 @@

    public static int getOrderNoAdd(int orderNo) {
        return orderNo + 1;
    }

    public static Date getDateByEnd() {
        Calendar calendar = Calendar.getInstance();
        calendar.setTime(new Date());
        calendar.set(Calendar.HOUR_OF_DAY, 23);
        calendar.set(Calendar.MINUTE, 59);
        calendar.set(Calendar.SECOND, 59);
        Date zero = calendar.getTime();
        return zero;
    }

    public static long getDistanceTime2Long(Date startTime, Date endTime) {
        long time1 = startTime.getTime();
        long time2 = endTime.getTime();

        long diff;
        if (time1 < time2) {
            diff = time2 - time1;
        } else {
            diff = time1 - time2;
        }
        return diff;
    }

    public static final String getChineseOrEnglishOrNumber(String str) {
@@ -332,14 +443,240 @@

    public static List<String> split(String redisVal) {
        if (StringUtils.isEmpty(redisVal)) {
            return null;
            return new ArrayList<>();
        }
        String[] split = redisVal.split(Constant.SEMICOLON);
        List<String> strings = Arrays.asList(split);
        return  new ArrayList<String>(strings);
    }

    public static List<String> split(String redisVal, String spilt) {
        if (StringUtils.isEmpty(redisVal)) {
            return new ArrayList<>();
        }
        String[] split = redisVal.split(spilt);
        List<String> strings = Arrays.asList(split);
        return strings;
    }

    public static TUser getUser(HttpServletRequest request) {
        return (TUser) request.getSession().getAttribute(Constant.USER);
        String userId = request.getParameter("userId");
        if (StringUtils.isEmpty(userId)) {
            return (TUser) request.getSession().getAttribute(Constant.USER);
        } else {
            return getUserById(userId);
        }
    }

    public static Character getCharacter(HttpServletRequest request) {
        String characterType = request.getParameter("characterType");
        if (StringUtils.isEmpty(characterType)) {
            return (TUser) request.getSession().getAttribute(Constant.USER);
        }else {
            String characterId = request.getParameter("characterId");
            MasterAuthorType type = MasterAuthorType.valueOf(characterType);
            switch (type){
                case role:
                    return DbUtils.getRoleById(characterId);
                case user:
                    return DbUtils.getUserById(characterId);
                case groupInfo:
                    return DbUtils.getGroupById(characterId);
            }
        }
        return null;
    }

    private static Character getGroupById(String characterId) {
        return new TUserGroup().setGroupId(characterId);
    }

    private static Character getRoleById(String roleId) {
        return userRoleClient.getRoleByRoleId(roleId);
    }

    public static boolean addOneRole(TUser user, List<String> roleIds) {
        if (user == null) {
            return false;
        }
        return userRoleClient.addRoleByUser(user.getUserId(), roleIds, user.getUserName());
    }

    public static boolean groupAddOneRole(String userId, String groupId, List<String> roleIds) {

        return userRoleClient.addRoleByGroupId(userId, roleIds, groupId);
    }

    public static boolean delOneRole(String userId, List<String> roleIds) {
        return userRoleClient.deleteRoleByUser(userId, roleIds);
    }
    public static boolean groupDelOneRole(String groupId, List<String> roleIds) {
        return userRoleClient.deleteRoleByGroupId(roleIds, groupId);
    }
    public static TUser getUserById(String userId) {
        TUser tUserById = userRoleClient.getTUserById(userId);

        return tUserById;
    }

    public static Set<String> getRoleByGroup(String groupId) {
        List<TRole> tRoleListByUserId = userRoleClient.getRoleListByGroupId(groupId);
        if (tRoleListByUserId == null || tRoleListByUserId.isEmpty()){
            return new HashSet<>();
        }
        Set<String> collect = tRoleListByUserId.stream().map(tRole -> tRole.getRoleId()).collect(Collectors.toSet());
        return collect;
    }

    public static Set<String> getRoleByUser(String userId) {
        List<TRole> tRoleListByUserId = userRoleClient.getTRoleListByUserId(userId);
        if (tRoleListByUserId == null || tRoleListByUserId.isEmpty()){
            return new HashSet<>();
        }
        Set<String> collect = tRoleListByUserId.stream().map(tRole -> tRole.getRoleId()).collect(Collectors.toSet());
        return collect;
    }

    public static Set<TUser> getAllUser() {
        List<TUser> tRoleListByUserId = userRoleClient.listUserRoleGroup();
        if (tRoleListByUserId == null) {
            return new HashSet<>();
        }
        return new HashSet<TUser>(tRoleListByUserId);
    }


    public static Set<String> getUserByRole(String roleId) {
        List<String> userListByRole = userRoleClient.getUserListByRole(roleId);
        if (userListByRole == null) {
            return new HashSet<>();
        }
        HashSet<String> result = new HashSet<>(userListByRole);
        return result;
    }

    public static Set<String> getGroupByUser(String userId) {
        List<TUserGroup> tRoleListByUserId = userRoleClient.getUserGroupByUserId(userId);
        if (tRoleListByUserId == null || tRoleListByUserId.isEmpty()){
            return new HashSet<>();
        }
        Set<String> collect = tRoleListByUserId.stream().map(tRole -> tRole.getGroupId()).collect(Collectors.toSet());
        return collect;
    }


    public static MyBatisSql getMyBatisSql(String id, Map<String, Object> parameterMap, SqlSessionFactory sqlSessionFactory) {
        MyBatisSql ibatisSql = new MyBatisSql();
        MappedStatement ms = sqlSessionFactory.getConfiguration().getMappedStatement(id);
        BoundSql boundSql = ms.getBoundSql(parameterMap);
        ibatisSql.setSql(boundSql.getSql());
        List<ParameterMapping> parameterMappings = boundSql.getParameterMappings();
        Configuration configuration = sqlSessionFactory.getConfiguration();
        if (parameterMappings != null) {
            Object[] parameterArray = new Object[parameterMappings.size()];
            ParameterMapping parameterMapping = null;
            Object value = null;
            Object parameterObject = null;
            MetaObject metaObject = null;
            PropertyTokenizer prop = null;
            String propertyName = null;
            String[] names = null;
            for (int i = 0; i < parameterMappings.size(); i++) {
                parameterMapping = parameterMappings.get(i);
                if (parameterMapping.getMode() != ParameterMode.OUT) {
                    propertyName = parameterMapping.getProperty();
                    names = propertyName.split("\\.");
                    if (propertyName.indexOf(".") != -1 && names.length == 2) {
                        parameterObject = parameterMap.get(names[0]);
                        propertyName = names[1];
                    } else if (propertyName.indexOf(".") != -1 && names.length == 3) {
                        parameterObject = parameterMap.get(names[0]); // map
                        if (parameterObject instanceof Map) {
                            parameterObject = ((Map) parameterObject).get(names[1]);
                        }
                        propertyName = names[2];
                    } else {
                        parameterObject = parameterMap.get(propertyName);
                    }
                    metaObject = parameterMap == null ? null : MetaObject.forObject(parameterObject,configuration.getObjectFactory(),configuration.getObjectWrapperFactory(), configuration.getReflectorFactory());
                    prop = new PropertyTokenizer(propertyName);
                    if (parameterObject == null) {
                        value = null;
                    } else if (ms.getConfiguration().getTypeHandlerRegistry().hasTypeHandler(parameterObject.getClass())) {
                        value = parameterObject;
                    } else if (boundSql.hasAdditionalParameter(propertyName)) {
                        value = boundSql.getAdditionalParameter(propertyName);
                    } else if (propertyName.startsWith(ForEachSqlNode.ITEM_PREFIX) && boundSql.hasAdditionalParameter(prop.getName())) {
                        value = boundSql.getAdditionalParameter(prop.getName());
                        if (value != null) {
                            value = MetaObject.forObject(value,configuration.getObjectFactory(),configuration.getObjectWrapperFactory(), configuration.getReflectorFactory()).getValue(propertyName.substring(prop.getName().length()));
                        }
                    } else {
                        value = metaObject == null ? null : metaObject.getValue(propertyName);
                    }
                    parameterArray[i] = value;
                }
            }
            ibatisSql.setParameters(parameterArray);
        }
        return ibatisSql;
    }

    public static String getNowDateStr() {
        SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
        String time = format.format(new Date());
        return time;
    }

    public static Set<String> getRoleByUser(String userId, String uuid) {

        Date start = new Date();
        List<TRole> tRoleListByUserId = userRoleClient.getTRoleListByUserId(userId);
        Date end = new Date();
        log.info(MessageFormat.format("master tag:{0} form userId select role :{1} ms",uuid, (end.getTime() - start.getTime())));


        if (tRoleListByUserId == null || tRoleListByUserId.isEmpty()){
            return new HashSet<>();
        }
        Set<String> collect = tRoleListByUserId.stream()
                .filter(tRole -> tRole != null).map(tRole -> tRole.getRoleId()).collect(Collectors.toSet());


        return collect;
        
    }

    public static Set<String> getGroupByUser(String userId, String uuid) {

        Date start = new Date();
        List<TUserGroup> tRoleListByUserId = userRoleClient.getUserGroupByUserId(userId);
        Date end = new Date();
        log.info(MessageFormat.format("master tag:{0} form userId select group :{1} ms",uuid, (end.getTime() - start.getTime())));

        if (tRoleListByUserId == null || tRoleListByUserId.isEmpty()){
            return new HashSet<>();
        }
        Set<String> collect = tRoleListByUserId.stream().map(tRole -> tRole.getGroupId()).collect(Collectors.toSet());


        return collect;
    }

    public static Set<String> getRoleByGroup(String groupId, String uuid) {

        Date start = new Date();
        List<TRole> tRoleListByUserId = userRoleClient.getRoleListByGroupId(groupId);
        Date end = new Date();
        log.info(MessageFormat.format("master tag:{0} form groupId select role :{1} ms",uuid, (end.getTime() - start.getTime())));

        if (tRoleListByUserId == null || tRoleListByUserId.isEmpty()){
            return new HashSet<>();
        }
        Set<String> collect = tRoleListByUserId.stream().map(tRole -> tRole.getRoleId()).collect(Collectors.toSet());


        return collect;
    }

}