diff --git a/src/main/java/top/fjy8018/designpattern/pattern/structural/proxy/staticproxy/OrderServiceStaticProxy.java b/src/main/java/top/fjy8018/designpattern/pattern/structural/proxy/staticproxy/OrderServiceStaticProxy.java index c272c27..a43947f 100644 --- a/src/main/java/top/fjy8018/designpattern/pattern/structural/proxy/staticproxy/OrderServiceStaticProxy.java +++ b/src/main/java/top/fjy8018/designpattern/pattern/structural/proxy/staticproxy/OrderServiceStaticProxy.java @@ -29,8 +29,16 @@ public class OrderServiceStaticProxy { * @return */ public int save(Order order) { - beforeMethod(); + beforeMethod(order); + int result = orderService.save(order); + + afterMethod(); + + return result; + } + + private void beforeMethod(Order order) { // 通过用户ID对2取模分库 Integer userId = order.getUserId(); int dbRouter = userId % 2; @@ -39,12 +47,6 @@ public class OrderServiceStaticProxy { // 设置DataSource DataSourceContextHolder.setDBType(DB + String.valueOf(dbRouter)); - afterMethod(); - - return orderService.save(order); - } - - private void beforeMethod() { log.info("静态代理before code"); }