Spring Boot中如何实现用户更新并正确返回Optional？

问题根源 #

报错原因是userRepository.save(user)返回的是User实体对象，但updateUser方法声明返回Optional<User>，类型不兼容导致编译错误。我们需要将保存后的User实例包装为Optional类型，同时保持需求逻辑：存在用户则更新返回，不存在则返回空Optional。

修复方案1：修改原if-else代码 #

直接将save方法返回的User用Optional.of()包装，适配方法返回类型：

@RequiredArgsConstructor
@Service
public class UserServiceImpl implements UserService {

    private final ModelMapper modelMapper;
    private final UserRepository userRepository;

    public Optional<User> updateUser(long userId, CreateUpdateUserDto data) {
        Optional<User> optionalUser = userRepository.findById(userId);
        if (optionalUser.isPresent()) {
            User user = optionalUser.get();
            user.setFirstName(data.getFirstName());
            user.setLastName(data.getLastName());
            // 包装save结果为Optional
            return Optional.of(userRepository.save(user));
        } else {
            return optionalUser;
        }
    }
}

修复方案2：用Optional链式调用简化代码 #

利用Optional的map方法替代显式的isPresent()判断，代码更简洁且符合Optional的设计规范：

@RequiredArgsConstructor
@Service
public class UserServiceImpl implements UserService {

    private final ModelMapper modelMapper;
    private final UserRepository userRepository;

    public Optional<User> updateUser(long userId, CreateUpdateUserDto data) {
        return userRepository.findById(userId)
            .map(user -> {
                user.setFirstName(data.getFirstName());
                user.setLastName(data.getLastName());
                return userRepository.save(user);
            });
    }
}

• 逻辑说明：findById返回空Optional时，map方法不会执行lambda表达式，直接返回空Optional；若存在用户，map会执行更新逻辑，将保存后的User自动包装为Optional返回。

进阶优化：利用ModelMapper自动映射属性 #

如果项目中已经引入ModelMapper，可以直接用它将DTO的属性映射到User实体，避免手动调用setter：

@RequiredArgsConstructor
@Service
public class UserServiceImpl implements UserService {

    private final ModelMapper modelMapper;
    private final UserRepository userRepository;

    public Optional<User> updateUser(long userId, CreateUpdateUserDto data) {
        return userRepository.findById(userId)
            .map(user -> {
                modelMapper.map(data, user); // 自动覆盖User的对应属性
                return userRepository.save(user);
            });
    }
}

内容的提问来源于stack exchange，提问作者Альберт Александров