用Lambda武装你的Java: 灵活的事务

651 查看

当你采用Spring之类的框架,用了声明式事务,难道每一段需要事务的代码都必须写成一个bean method,再标上@Transactional?

未免太麻烦了,不信你瞧。假如你写了类似这样的Controller和Service (伪代码):

class UserController {
  @Autowired UserService us;

  String updateUser(long userId) {
    User user = us.authorize(userId);
    Event e = us.update(user);
    publishToMQ(e);
    return "user-page";
  }
}

@Transactional
class UserService {
  User authorize(long userId) {...}
  void update(User user) {...}
}

问题来了,authorize和update分别用了两个分开的事务,如果你用了Hibernate或JPA,并且user是lazy-loading的,这就没法运行。你需要让两次调用运行在同一个事务里。通常的办法是把UserController.updateUser也标成@Transactional。可是这么一来,下一句publishToMQ(e);虽然不需要事务,却也被包在事务里了。

我们可以做得更好!用Java 8做一个Transactor,任何代码块可随时包在事务中!
然后controller可以重写为:

String updateUser(long userId) {
  Event e = Transactor.get().apply(() -> {
    User user = us.authorize(userId);
    return us.update(user);
  });
  publishToMQ(e);
  return "user-page";
}

Transactor的实现:

@Component
@Transactional
public class Transactor {
  public static Transactor get() {
    return instance;
  }

  public <R> R apply(Supplier<R> f) {
    return f.get(); // 有返回值的代码块
  }

  public void run(Runnable f) {
    f.run(); // 无返回值的代码块
  }

  @Autowired
  private ApplicationContext applicationContext;
  @PostConstruct
  void setup() {
    instance = applicationContext.getBean(Transactor.class); //不能写instance=this
  }
  private static Transactor instance;
}

代码中透出四个字:灵活,简洁!

2015/7/5 Update: 可惜的是,有时会抛出org.hibernate.HibernateException: Could not obtain transaction-synchronized Session for current thread (虽然stacktrace中是有代理类的)
目前实测改用Spring的TransactionTemplate是可以的。