解决Spring或SpringBoot开启事务以后无法返回自增主键的问题

在Spring或Spring Boot应用程序中，如果开启了事务，那么在执行插入操作时，无法返回自增主键。这是因为在事务中，插入操作并没有真正地提交到数据库中，因此无法获取自增主键。在本文中，我们将介绍如何解决这个问题，并提供两个示例说明。

解决方案

要解决这个问题，我们可以使用JDBC的KeyHolder接口来获取自增主键。KeyHolder接口是Spring JDBC框架提供的一个接口，用于保存自动生成的主键。下面是一个示例代码：

@Autowired
private JdbcTemplate jdbcTemplate;

@Transactional
public Long addUser(User user) {
  KeyHolder keyHolder = new GeneratedKeyHolder();
  jdbcTemplate.update(connection -> {
    PreparedStatement ps = connection.prepareStatement("INSERT INTO users (name, age) VALUES (?, ?)", Statement.RETURN_GENERATED_KEYS);
    ps.setString(1, user.getName());
    ps.setInt(2, user.getAge());
    return ps;
  }, keyHolder);
  return keyHolder.getKey().longValue();
}

在上面的代码中，我们使用@Autowired注解将JdbcTemplate注入到Spring Bean中。在addUser()方法中，我们使用@Transactional注解开启事务。在事务中，我们使用JdbcTemplate的update()方法执行插入操作，并将PreparedStatement的第二个参数设置为Statement.RETURN_GENERATED_KEYS，以便获取自增主键。在update()方法中，我们使用Lambda表达式来设置PreparedStatement的参数，并将KeyHolder对象传递给update()方法。在插入操作完成后，我们使用KeyHolder对象的getKey()方法来获取自增主键。

示例说明

下面是两个示例，演示如何解决无法返回自增主键的问题。

示例1：使用JdbcTemplate获取自增主键

在应用程序中，我们可以定义一个名为UserDao的数据访问对象，并使用JdbcTemplate来执行数据库操作。下面是一个示例代码：

@Repository
public class UserDao {

  @Autowired
  private JdbcTemplate jdbcTemplate;

  @Transactional
  public Long addUser(User user) {
    KeyHolder keyHolder = new GeneratedKeyHolder();
    jdbcTemplate.update(connection -> {
      PreparedStatement ps = connection.prepareStatement("INSERT INTO users (name, age) VALUES (?, ?)", Statement.RETURN_GENERATED_KEYS);
      ps.setString(1, user.getName());
      ps.setInt(2, user.getAge());
      return ps;
    }, keyHolder);
    return keyHolder.getKey().longValue();
  }
}

在上面的代码中，我们定义了一个名为UserDao的数据访问对象，并使用@Repository注解将其声明为Spring Bean。该对象使用@Autowired注解将JdbcTemplate注入到Spring Bean中。在addUser()方法中，我们使用@Transactional注解开启事务，并使用JdbcTemplate的update()方法执行插入操作。在update()方法中，我们使用Lambda表达式来设置PreparedStatement的参数，并将KeyHolder对象传递给update()方法。在插入操作完成后，我们使用KeyHolder对象的getKey()方法来获取自增主键。

示例2：使用MyBatis获取自增主键

在应用程序中，我们可以使用MyBatis框架来执行数据库操作。下面是一个示例代码：

@Repository
public interface UserDao {

  @Insert("INSERT INTO users (name, age) VALUES (#{name}, #{age})")
  @Options(useGeneratedKeys = true, keyProperty = "id")
  void addUser(User user);
}

在上面的代码中，我们定义了一个名为UserDao的数据访问对象，并使用@Repository注解将其声明为Spring Bean。该对象使用@Insert注解定义了一个名为addUser()的方法，用于执行插入操作。在@Insert注解中，我们使用#{name}和#{age}来引用User对象的属性，并使用@Options注解来开启自动生成主键的功能。在@Options注解中，我们将useGeneratedKeys属性设置为true，以便开启自动生成主键的功能，并将keyProperty属性设置为"id"，以便将自动生成的主键设置到User对象的id属性中。