逻辑改进

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");
     }