MongoTemplate使用updateFirst()更新数据时，审计功能失效

使用审计功能

在日常的开发工作中，我们经常需要记录数据的创建时间，更新时间等信息。但如果每次更新的时候都去设置时间的话，又比较麻烦，我们可以直接使用审计功能去帮助我们自动地填充这些字段。要开启审计功能，首先要在启动类或者是配置类上加上@EnableMongoAuditing注解。

package Demo;

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.data.mongodb.config.EnableMongoAuditing;

@SpringBootApplication
@EnableMongoAuditing
public class DemoApplication {
    public static void main(String[] args) {
        SpringApplication.run(DemoApplication.class,args);
        System.out.println("start sucess!!!!!!!!!!!!!!!!");
    }
}

然后在实体类上加上以下的注解：

• @CreatedDate 自动添加创建时间
• @LastModifiedDate 自动添加更新时间

package Demo.Domain;

import com.fasterxml.jackson.annotation.JsonFormat;
import lombok.Data;
import lombok.ToString;
import org.springframework.data.annotation.CreatedDate;
import org.springframework.data.annotation.Id;
import org.springframework.data.annotation.LastModifiedDate;
import org.springframework.data.mongodb.core.mapping.Document;

import java.time.LocalDateTime;

@Data
@Document
@ToString
public class User {
    @Id
    private String id;

    private String userName;

    private String jobName;

    @CreatedDate
    @JsonFormat(pattern = "yyyy-MM-dd  HH:mm:ss",timezone = "GMT+8")
    private LocalDateTime createTime;

    @LastModifiedDate
    @JsonFormat(pattern = "yyyy-MM-dd  HH:mm:ss",timezone = "GMT+8")
    private LocalDateTime updateTime;
}

然后就可以愉快地使用了

package Demo.Service;

import Demo.Domain.User;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;

@Service
public class UserServiceImpl {

    @Resource
    private MongoTemplate mongoTemplate;

    public void saveOrUpdate(User user) {
        User byUserName = getByUserName(user.getUserName());
        if (byUserName == null) {
            mongoTemplate.save(user);
            return;
        }
        Query query = getQuery(user.getUserName());
        Update update = new Update();
        update.set("jobName", user.getJobName());
        mongoTemplate.updateFirst(query, update, User.class);
    }

    public User getByUserName(String userName) {
        Query query = getQuery(userName);
        return mongoTemplate.findOne(query, User.class);
    }

    public void deleteByUserName(String userName){
        mongoTemplate.remove(getQuery(userName),User.class);
    }

    private Query getQuery(String userName) {
        Query query = new Query();
        query.addCriteria(Criteria.where("userName").is(userName));
        return query;
    }
}

测试类为：

package Demo.Service;


import Demo.Domain.User;
import lombok.extern.slf4j.Slf4j;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;

@SpringBootTest
@RunWith(SpringRunner.class)
@Slf4j
public class UserServiceImplTest {

    @Autowired
    private UserServiceImpl userServiceImpl;

    @Test
    public void saveOrUpdate() throws InterruptedException {
        User user =new User();
        user.setUserName("123");
        user.setJobName("job1");
        // 插入数据
        userServiceImpl.saveOrUpdate(user);
        log.info("插入数据:{}",userServiceImpl.getByUserName(user.getUserName()));
        user.setJobName("job2");
        Thread.sleep(3000);
        //更新数据
        userServiceImpl.saveOrUpdate(user);
        log.info("更新数据:{}",userServiceImpl.getByUserName(user.getUserName()));
        //删除数据
        userServiceImpl.deleteByUserName(user.getUserName());
    }
}

发现updateFirst()方法无法自动更新数据的更新时间

但是，在测试的时候我发现：在使用mongoTemplate.save()方法去插入数据的时候，能够正常记录数据的创建时间，但在使用mongoTemplate.updateFirst()方法去更新数据的时候，却无法记录数据的更新时间！！！

具体的原因可能是save()方法做了审计相关的一些增强，但是updateFirst()方法却没有实现相关的增强。

解决方法

直接使用save()去更新

既然这里的更新代码已经查出相应的user对象了，那么直接去更新它相应的字段值，然后再次进行save()就好了。

    public void saveOrUpdate(User user) {
        User byUserName = getByUserName(user.getUserName());
        if (byUserName == null) {
            mongoTemplate.save(user);
            return;
        }
        // 直接在这里更新对应字段的值，因为就算是使用updateFirst()也得去填充update的字段值，不如直接在这里添加算了
        byUserName.setJobName(user.getJobName());
        mongoTemplate.save(byUserName);
    }

在使用updateFirst()的时候顺手将更新时间加进去

反正使用@LastModifiedDate注解应该也是通过类似AOP的方式去实现的增强，从而自动地添加数据的更新时间，那这里的增强失效了。（我也不知道是因为增强失效还是根本就没有增强0.0）。那直接手动去设置更新时间字段即可。updateFirst()方法在使用的时候本就有一种自行挑选需要更新的字段的感觉。

    public void saveOrUpdate(User user) {
        User byUserName = getByUserName(user.getUserName());
        if (byUserName == null) {
            mongoTemplate.save(user);
            return;
        }
        Query query = getQuery(user.getUserName());
        Update update = new Update();
        update.set("jobName", user.getJobName());
        // 自行设置更新字段
        update.set("updateTime", LocalDateTime.now());
        mongoTemplate.updateFirst(query, update, User.class);
    }