Java泛型方法

旧城等待， 2022-10-17 05:29 108阅读 0赞

1、示例代码

请关注方法：encryptObjectStringFiled

import org.apache.commons.collections4.CollectionUtils;
    import org.apache.commons.lang3.StringUtils;
    
    import javax.crypto.*;
    import javax.crypto.spec.SecretKeySpec;
    import java.io.UnsupportedEncodingException;
    import java.lang.reflect.Field;
    import java.lang.reflect.Method;
    import java.security.InvalidKeyException;
    import java.security.NoSuchAlgorithmException;
    import java.security.SecureRandom;
    import java.util.Base64;
    import java.util.HashSet;
    import java.util.Set;
    
    public class PasswordUtil {
    
    //    private static final Logger log = LoggerFactory.getLogger(PasswordUtil.class);
    
        public static final String PASSWORD = "abc892348902423*";
    
        /**
         * 加密
         */
        public static String encrypt(String content) {
            if (StringUtils.isBlank(content)) {
                return null;
            }
            try {
                KeyGenerator kgen = KeyGenerator.getInstance("AES");
                kgen.init(128, new SecureRandom(PASSWORD.getBytes()));
                SecretKey secretKey = kgen.generateKey();
                byte[] enCodeFormat = secretKey.getEncoded();
                SecretKeySpec key = new SecretKeySpec(enCodeFormat, "AES");
                Cipher cipher = Cipher.getInstance("AES");// 创建密码器
                byte[] byteContent = content.getBytes("utf-8");
                cipher.init(Cipher.ENCRYPT_MODE, key);// 初始化
                byte[] result = cipher.doFinal(byteContent);
                String encoded = Base64.getEncoder().encodeToString(result);
                return encoded; // 加密
            } catch (NoSuchAlgorithmException e) {
                e.printStackTrace();
            } catch (NoSuchPaddingException e) {
                e.printStackTrace();
            } catch (InvalidKeyException e) {
                e.printStackTrace();
            } catch (UnsupportedEncodingException e) {
                e.printStackTrace();
            } catch (IllegalBlockSizeException e) {
                e.printStackTrace();
            } catch (BadPaddingException e) {
                e.printStackTrace();
            }
            return null;
        }
    
        /**
         * 解密
         */
        public static String decrypt(String content) {
            if (StringUtils.isBlank(content)) {
                return null;
            }
            try {
                byte[] decoded = Base64.getDecoder().decode(content);
                KeyGenerator kgen = KeyGenerator.getInstance("AES");
                kgen.init(128, new SecureRandom(PASSWORD.getBytes()));
                SecretKey secretKey = kgen.generateKey();
                byte[] enCodeFormat = secretKey.getEncoded();
                SecretKeySpec key = new SecretKeySpec(enCodeFormat, "AES");
                Cipher cipher = Cipher.getInstance("AES");// 创建密码器
                cipher.init(Cipher.DECRYPT_MODE, key);// 初始化
                byte[] result = cipher.doFinal(decoded);
                return new String(result); // 加密
            } catch (NoSuchAlgorithmException e) {
                e.printStackTrace();
            } catch (NoSuchPaddingException e) {
                e.printStackTrace();
            } catch (InvalidKeyException e) {
                e.printStackTrace();
            } catch (IllegalBlockSizeException e) {
                e.printStackTrace();
            } catch (BadPaddingException e) {
                e.printStackTrace();
            }
            return null;
        }
    
        /**
         * 加密对象指定的String类型的属性值
         */
        public static <T> T encryptObjectStringFiled(T obj, Set<String> filedNameSet) {
            if (CollectionUtils.isEmpty(filedNameSet)) {
                return obj;
            }
            Class<?> clz = obj.getClass();
            Field[] fields = clz.getDeclaredFields();
    
            try {
                for (Field field : fields) {
                    String fieldName = field.getName();
                    if (!filedNameSet.contains(fieldName)) {
                        continue;
                    }
                    //GET方法
                    Method methodGet = obj.getClass().getMethod("get" + getMethodName(field.getName()));
                    Object val = methodGet.invoke(obj);
                    String content = null == val ? null : val.toString();
                    //加密
                    String encrypt = encrypt(content);
                    Method methodSet = obj.getClass().getMethod("set" + getMethodName(field.getName()), String.class);
                    methodSet.invoke(obj, encrypt);
    
                }
            } catch (Exception e) {
    //            log.error("对象属性加密失败：{}", e);
            }
    
            return obj;
        }
    
        /**
         * 解密对象指定的String类型的属性值
         */
        public static <T> T decryptObjectStringFiled(T obj, Set<String> filedNameSet) {
            if (CollectionUtils.isEmpty(filedNameSet)) {
                return obj;
            }
            Class<?> clz = obj.getClass();
            Field[] fields = clz.getDeclaredFields();
    
            try {
                for (Field field : fields) {
                    String fieldName = field.getName();
                    if (!filedNameSet.contains(fieldName)) {
                        continue;
                    }
                    Method methodGet = obj.getClass().getMethod("get" + getMethodName(field.getName()));
                    Object val = methodGet.invoke(obj);
                    //解密
                    String content = null == val ? null : val.toString();
                    String encrypt = decrypt(content);
                    Method methodSet = obj.getClass().getMethod("set" + getMethodName(field.getName()), String.class);
                    methodSet.invoke(obj, encrypt);
    
                }
            } catch (Exception e) {
    //            log.error("对象属性解密失败：{}", e);
            }
    
            return obj;
        }
    
        // 把一个字符串的第一个字母大写、效率是最高的、
        private static String getMethodName(String filedName) {
            return filedName.substring(0, 1).toUpperCase() + filedName.substring(1);
        }
    
    
        public static void main(String[] args) {
            String content = "1000004";
            //加密
            System.out.println("加密前：" + content);
            String encryptResult = encrypt(content);
            System.out.println("加密后：" + encryptResult);
            //解密
            String decryptResult = decrypt(encryptResult);
            System.out.println("解密后：" + decryptResult);
    
            TestVO obj = new TestVO();
            Set set = new HashSet();
            set.add("name");
            set.add("age");
            TestVO object = encryptObjectStringFiled(obj, set);
            System.out.println(object.getName());
            System.out.println(object.getAge());
            TestVO object2 = decryptObjectStringFiled(object, set);
    
            System.out.println(object2.getName());
            System.out.println(object2.getAge());
    
            String s = "prZfSL90VWQmK0hT7j3tjQ==";
            String origin = decrypt(s);
            System.out.println(origin);
    
        }
    
    
    }

import lombok.Data;
    
    @Data
    public class TestVO {
    
        private String name = "张**三";
    
        private String age = "12";
    }