SpringBoot 实现JPA的save方法不更新null属性

小灰灰 2022-02-04 12:15 237阅读 0赞

转载自：[https://www.jianshu.com/p/4931fbc52ea1][https_www.jianshu.com_p_4931fbc52ea1]

直接调用原生Save方法会导致null属性覆盖到数据库，使用起来十分不方便。本文提供便捷方法解决此问题。

# 核心思路 #

如果现在保存某User对象，首先根据主键查询这个User的最新对象，然后将此User对象的非空属性覆盖到最新对象。

# 核心代码 #

直接修改通用JpaRepository的实现类，然后在启动类标记此实现类即可。

主要是实现一个类：

package com.tianyalei.giftmall.global.base;
    
    import org.springframework.beans.BeanUtils;
    import org.springframework.beans.BeanWrapper;
    import org.springframework.beans.BeanWrapperImpl;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.data.jpa.repository.support.JpaEntityInformation;
    import org.springframework.data.jpa.repository.support.SimpleJpaRepository;
    import org.springframework.transaction.annotation.Transactional;
    import org.springframework.util.StringUtils;
    
    import javax.persistence.EntityManager;
    import java.beans.PropertyDescriptor;
    import java.util.HashSet;
    import java.util.Optional;
    import java.util.Set;
    
    /**
     * @author wuweifeng wrote on 2018/12/4.
     */
    public class SimpleJpaRepositoryImpl<T, ID> extends SimpleJpaRepository<T, ID> {
    
        private final JpaEntityInformation<T, ?> entityInformation;
        private final EntityManager em;
    
        @Autowired
        public SimpleJpaRepositoryImpl(JpaEntityInformation<T, ?> entityInformation, EntityManager entityManager) {
            super(entityInformation, entityManager);
            this.entityInformation = entityInformation;
            this.em = entityManager;
        }
    
        /**
         * 通用save方法 ：新增/选择性更新
         */
        @Override
        @Transactional(rollbackFor = Exception.class)
        public <S extends T> S save(S entity) {
            //获取ID
            ID entityId = (ID) entityInformation.getId(entity);
            Optional<T> optionalT;
            if (StringUtils.isEmpty(entityId)) {
                //String uuid = UUID.randomUUID().toString();
                防止UUID重复
                //if (findById((ID) uuid).isPresent()) {
                //    uuid = UUID.randomUUID().toString();
                //}
                若ID为空 则设置为UUID
                //new BeanWrapperImpl(entity).setPropertyValue(entityInformation.getIdAttribute().getName(), uuid);
                //标记为新增数据
                optionalT = Optional.empty();
            } else {
                //若ID非空 则查询最新数据
                optionalT = findById(entityId);
            }
            //获取空属性并处理成null
            String[] nullProperties = getNullProperties(entity);
            //若根据ID查询结果为空
            if (!optionalT.isPresent()) {
                //新增
                em.persist(entity);
                return entity;
            } else {
                //1.获取最新对象
                T target = optionalT.get();
                //2.将非空属性覆盖到最新对象
                BeanUtils.copyProperties(entity, target, nullProperties);
                //3.更新非空属性
                em.merge(target);
                return entity;
            }
        }
    
        /**
         * 获取对象的空属性
         */
        private static String[] getNullProperties(Object src) {
            //1.获取Bean
            BeanWrapper srcBean = new BeanWrapperImpl(src);
            //2.获取Bean的属性描述
            PropertyDescriptor[] pds = srcBean.getPropertyDescriptors();
            //3.获取Bean的空属性
            Set<String> properties = new HashSet<>();
            for (PropertyDescriptor propertyDescriptor : pds) {
                String propertyName = propertyDescriptor.getName();
                Object propertyValue = srcBean.getPropertyValue(propertyName);
                if (StringUtils.isEmpty(propertyValue)) {
                    srcBean.setPropertyValue(propertyName, null);
                    properties.add(propertyName);
                }
            }
            return properties.toArray(new String[0]);
        }
    }

然后在application类上加个注解即可

@SpringBootApplication
    @EnableJpaRepositories(repositoryBaseClass = SimpleJpaRepositoryImpl.class)
    public class GiftmallApplication {
    
        public static void main(String[] args) {
            SpringApplication.run(GiftmallApplication.class, args);
        }
    
        @Bean
        public ObjectMapper objectMapper() {
            return new ObjectMapper().disable(SerializationFeature.FAIL_ON_EMPTY_BEANS);
        }
    }

这样，所有的save方法都会进入到上面类的逻辑里，进行null字段的排除后更新。

[https_www.jianshu.com_p_4931fbc52ea1]: https://www.jianshu.com/p/4931fbc52ea1