| 提交 | 用户 | age | ||
| c64e12 | 1 | package com.hx.mybatis.aes.springbean; |
| e29546 | 2 | |
| c64e12 | 3 | import com.gitee.sunchenbin.mybatis.actable.annotation.Table; |
| e29546 | 4 | import com.hx.common.annotations.MysqlHexAes; |
| C | 5 | import com.hx.util.StringUtils; |
| 6 | import org.springframework.stereotype.Component; | |
| 7 | ||
| 8 | import javax.annotation.PostConstruct; | |
| 9 | import javax.annotation.Resource; | |
| 10 | import java.io.File; | |
| 11 | import java.io.FileFilter; | |
| 12 | import java.io.IOException; | |
| 13 | import java.lang.reflect.Field; | |
| 14 | import java.net.URL; | |
| 15 | import java.net.URLDecoder; | |
| 16 | import java.util.*; | |
| 17 | ||
| 18 | /** | |
| 19 | * 获取指定包里面的AES秘钥 | |
| 20 | */ | |
| 21 | @Component | |
| 22 | public class VariableAesKey { | |
| 23 | ||
| 24 | @Resource | |
| 25 | private ConstantBean constantBean; | |
| 26 | ||
| c64e12 | 27 | /**是否已经启动完*/ |
| C | 28 | public static int isRun = 0; |
| 29 | ||
| 30 | /**存储所有AES的秘钥*/ | |
| e29546 | 31 | public static Map<String,String> aesKeys = new HashMap<>(); |
| c64e12 | 32 | /**根据表明来存储AES秘钥*/ |
| C | 33 | public static Map<String,Map<String,String>> aesKeysTable = new HashMap<>(); |
| 34 | ||
| 35 | /**固定的aes秘钥*/ | |
| 36 | public static String AES_KEY = null; | |
| e29546 | 37 | |
| C | 38 | /**存储AES秘钥*/ |
| 39 | public static void setAesKey(String aesKeyFild,String aesKey){ | |
| 40 | aesKeys.put(aesKeyFild,aesKey); | |
| 41 | } | |
| 42 | /**获取AES秘钥*/ | |
| 43 | public static String getAesKey(String aesKeyFild){ | |
| c64e12 | 44 | if(aesKeyFild == null){ |
| C | 45 | return AES_KEY; |
| 46 | } | |
| 47 | if(StringUtils.isEmpty(aesKeys.get(aesKeyFild))){ | |
| 48 | return AES_KEY; | |
| 49 | }else { | |
| 50 | return aesKeys.get(aesKeyFild); | |
| 51 | } | |
| e29546 | 52 | } |
| C | 53 | |
| 54 | /** | |
| 55 | * 项目启动就执行后就执行该方法 | |
| 56 | */ | |
| 57 | @PostConstruct | |
| 58 | public void VariableAesKey(){ | |
| c64e12 | 59 | |
| C | 60 | isRun = 1; |
| e29546 | 61 | //项目启动的时候填入 |
| C | 62 | System.err.println("扫描获取AES:" + constantBean.getPackPath()); |
| c64e12 | 63 | AES_KEY = constantBean.getFixedAesKey(); |
| e29546 | 64 | if(StringUtils.noNull(constantBean.getPackPath())){ |
| C | 65 | Set<Class<?>> classes = classData(constantBean.getPackPath()); |
| c64e12 | 66 | |
| C | 67 | Map<String,String> aesKeysFild = new HashMap<>(); |
| 68 | boolean isAes = false; | |
| 69 | String tableName = null; | |
| 70 | ||
| e29546 | 71 | for(Class<?> cl:classes){ |
| c64e12 | 72 | //表名称 |
| C | 73 | Table table = cl.getAnnotation(Table.class); |
| 74 | if(table == null){ | |
| 75 | continue; | |
| 76 | } | |
| 77 | tableName = table.name(); | |
| 78 | ||
| 79 | aesKeysFild = new HashMap<>(); | |
| 80 | isAes = false; | |
| 81 | ||
| e29546 | 82 | // 取得本类的全部属性 |
| C | 83 | Field[] fields = cl.getDeclaredFields(); |
| 84 | fields = getPatentFields(fields,cl); | |
| 85 | for (Field field:fields) { | |
| 86 | // 判断方法中是否有指定注解类型的注解 | |
| 87 | boolean hasAnnotation = field.isAnnotationPresent(MysqlHexAes.class); | |
| 88 | if (hasAnnotation) { | |
| 89 | // 根据注解类型返回方法的指定类型注解 | |
| 90 | MysqlHexAes mysqlHexAes = field.getAnnotation(MysqlHexAes.class); | |
| 91 | ||
| c64e12 | 92 | //String aesKeyField = mysqlHexAes.aesKeyField(); |
| e29546 | 93 | String aesKey = mysqlHexAes.aesKey(); |
| C | 94 | |
| 95 | if(StringUtils.isEmpty(aesKey)){ | |
| c64e12 | 96 | aesKey = constantBean.getFixedAesKey(); |
| C | 97 | if(StringUtils.isEmpty(aesKey)){ |
| 98 | throw new RuntimeException("mysql的AES秘钥不能为空:"+field.getName()); | |
| e29546 | 99 | } |
| c64e12 | 100 | } |
| C | 101 | String key = aesKeys.get(field.getName()); |
| 102 | if(StringUtils.isEmpty(key)){ | |
| 103 | aesKeys.put(field.getName(),aesKey); | |
| 104 | aesKeysFild.put(field.getName(),aesKey); | |
| 105 | isAes = true; | |
| e29546 | 106 | }else{ |
| c64e12 | 107 | isAes = true; |
| C | 108 | aesKeysFild.put(field.getName(),aesKey); |
| 109 | if(!aesKey.equals(key)){ | |
| 110 | throw new RuntimeException("字段/定义的AES秘钥字段【"+field.getName()+"】多个一样，但是AES秘钥不一样"); | |
| e29546 | 111 | } |
| C | 112 | } |
| 113 | } | |
| c64e12 | 114 | } |
| C | 115 | if(isAes){ |
| 116 | aesKeysTable.put(tableName,aesKeysFild); | |
| e29546 | 117 | } |
| C | 118 | } |
| 119 | } | |
| 120 | } | |
| 121 | ||
| 122 | /**获取包下面的所有文件*/ | |
| 123 | public static Set<Class<?>> classData(String packPath){ | |
| 124 | Set<Class<?>> classes = new LinkedHashSet<Class<?>>(); | |
| c64e12 | 125 | |
| C | 126 | //截取 |
| 127 | String[] packPaths = packPath.split(";|,"); | |
| 128 | for( String packageName : packPaths){ | |
| 129 | // 是否循环迭代 | |
| 130 | boolean recursive = true; | |
| 131 | // 获取包的名字 并进行替换 | |
| 132 | String packageDirName = packageName.replace('.', '/'); | |
| 133 | // 定义一个枚举的集合 并进行循环来处理这个目录下的things | |
| 134 | Enumeration<URL> dirs; | |
| 135 | try{ | |
| 136 | dirs = Thread.currentThread().getContextClassLoader().getResources(packageDirName); | |
| 137 | // 循环迭代下去 | |
| 138 | while (dirs.hasMoreElements()){ | |
| 139 | // 获取下一个元素 | |
| 140 | URL url = dirs.nextElement(); | |
| 141 | // 得到协议的名称 | |
| 142 | String protocol = url.getProtocol(); | |
| 143 | // 如果是以文件的形式保存在服务器上 | |
| 144 | if ("file".equals(protocol)) { | |
| 145 | // 获取包的物理路径 | |
| 146 | String filePath = URLDecoder.decode(url.getFile(), "UTF-8"); | |
| 147 | // 以文件的方式扫描整个包下的文件 并添加到集合中 | |
| 148 | findAndAddClassesInPackageByFile(packageName, filePath, recursive, classes); | |
| 149 | } | |
| e29546 | 150 | } |
| c64e12 | 151 | }catch (IOException e){ |
| C | 152 | e.printStackTrace(); |
| e29546 | 153 | } |
| C | 154 | } |
| 155 | return classes; | |
| 156 | } | |
| 157 | ||
| 158 | /** | |
| 159 | * 以文件的形式来获取包下的所有Class | |
| 160 | * | |
| 161 | * @param packageName | |
| 162 | * @param packagePath | |
| 163 | * @param recursive | |
| 164 | * @param classes | |
| 165 | */ | |
| 166 | public static void findAndAddClassesInPackageByFile( | |
| 167 | String packageName, | |
| 168 | String packagePath, | |
| 169 | final boolean recursive, | |
| 170 | Set<Class<?>> classes){ | |
| 171 | // 获取此包的目录 建立一个File | |
| 172 | File dir = new File(packagePath); | |
| 173 | // 如果不存在或者 也不是目录就直接返回 | |
| 174 | if (!dir.exists() || !dir.isDirectory()) { | |
| 175 | // log.warn("用户定义包名 " + packageName + " 下没有任何文件"); | |
| 176 | return; | |
| 177 | } | |
| 178 | // 如果存在 就获取包下的所有文件 包括目录 | |
| 179 | File[] dirfiles = dir.listFiles(new FileFilter(){ | |
| 180 | // 自定义过滤规则 如果可以循环(包含子目录) 或则是以.class结尾的文件(编译好的java类文件) | |
| 181 | @Override | |
| 182 | public boolean accept(File file){ | |
| 183 | return (recursive && file.isDirectory()) || (file.getName().endsWith(".class")); | |
| 184 | } | |
| 185 | }); | |
| 186 | // 循环所有文件 | |
| 187 | for (File file : dirfiles){ | |
| 188 | // 如果是目录 则继续扫描 | |
| 189 | if (file.isDirectory()) { | |
| 190 | findAndAddClassesInPackageByFile(packageName + "." + file.getName(), file.getAbsolutePath(), recursive, classes); | |
| 191 | }else{ | |
| 192 | // 如果是java类文件 去掉后面的.class 只留下类名 | |
| 193 | String className = file.getName().substring(0, file.getName().length() - 6); | |
| 194 | try{ | |
| 195 | // 添加到集合中去 | |
| 196 | // classes.add(Class.forName(packageName + '.' + | |
| 197 | // className)); | |
| 198 | // 经过回复同学的提醒，这里用forName有一些不好，会触发static方法，没有使用classLoader的load干净 | |
| 199 | classes.add(Thread.currentThread().getContextClassLoader().loadClass(packageName + '.' + className)); | |
| 200 | }catch (ClassNotFoundException e){ | |
| 201 | // log.error("添加用户自定义视图类错误 找不到此类的.class文件"); | |
| 202 | e.printStackTrace(); | |
| 203 | } | |
| 204 | } | |
| 205 | } | |
| 206 | } | |
| 207 | ||
| 208 | /** | |
| 209 | * 获取父类的字段 | |
| 210 | * @param fields | |
| 211 | * @param clas | |
| 212 | * @return | |
| 213 | */ | |
| 214 | public static Field[] getPatentFields(Field[] fields,Class<?> clas){ | |
| 215 | if (clas.getSuperclass() != null) { | |
| 216 | Class clsSup = clas.getSuperclass(); | |
| 217 | List<Field> fieldList = new ArrayList<Field>(); | |
| 218 | fieldList.addAll(Arrays.asList(fields)); | |
| 219 | fieldList.addAll(Arrays.asList(clsSup.getDeclaredFields())); | |
| 220 | fields = new Field[fieldList.size()]; | |
| 221 | int i = 0; | |
| 222 | for (Object field : fieldList.toArray()) { | |
| 223 | fields[i] = (Field) field; | |
| 224 | i++; | |
| 225 | } | |
| 226 | fields = getPatentFields(fields,clsSup); | |
| 227 | } | |
| 228 | return fields; | |
| 229 | } | |
| 230 | ||
| 231 | ||
| 232 | } | |
