From 8906dd9ab56903975861b9818c916ba81388ac69 Mon Sep 17 00:00:00 2001
From: javahongxi <javahongxi@qq.com>
Date: Mon, 18 Dec 2017 03:48:02 +0800
Subject: [PATCH] =?UTF-8?q?=E5=88=86=E5=BA=93=E5=88=86=E8=A1=A8demo?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../src/main/java/com/whatsmars/dbrouter/demo/BaseDO.java   | 6 +++++-
 .../src/main/java/com/whatsmars/dbrouter/demo/Main.java     | 2 --
 .../main/java/com/whatsmars/dbrouter/demo/OrderService.java | 3 +++
 .../src/main/resources/sqlmap/Order-mapper.xml              | 2 +-
 4 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/whatsmars-dbrouter/src/main/java/com/whatsmars/dbrouter/demo/BaseDO.java b/whatsmars-dbrouter/src/main/java/com/whatsmars/dbrouter/demo/BaseDO.java
index ed837f3a..1437daf2 100644
--- a/whatsmars-dbrouter/src/main/java/com/whatsmars/dbrouter/demo/BaseDO.java
+++ b/whatsmars-dbrouter/src/main/java/com/whatsmars/dbrouter/demo/BaseDO.java
@@ -1,5 +1,7 @@
 package com.whatsmars.dbrouter.demo;
 
+import com.whatsmars.dbrouter.DbContext;
+
 /**
  * Created by shenhongxi on 16/7/16.
  */
@@ -28,10 +30,12 @@ public class BaseDO {
     }
 
     public String getTableIndex() {
-        return tableIndex;
+        if (tableIndex != null && !"".equals(tableIndex)) return tableIndex;
+        return DbContext.getTableIndex();
     }
 
     public void setTableIndex(String tableIndex) {
         this.tableIndex = tableIndex;
+        DbContext.setTableIndex(tableIndex);
     }
 }
diff --git a/whatsmars-dbrouter/src/main/java/com/whatsmars/dbrouter/demo/Main.java b/whatsmars-dbrouter/src/main/java/com/whatsmars/dbrouter/demo/Main.java
index fbb95c02..b06e1a9e 100644
--- a/whatsmars-dbrouter/src/main/java/com/whatsmars/dbrouter/demo/Main.java
+++ b/whatsmars-dbrouter/src/main/java/com/whatsmars/dbrouter/demo/Main.java
@@ -1,6 +1,5 @@
 package com.whatsmars.dbrouter.demo;
 
-import com.whatsmars.dbrouter.DbContext;
 import org.springframework.context.ApplicationContext;
 import org.springframework.context.support.ClassPathXmlApplicationContext;
 
@@ -18,6 +17,5 @@ public class Main {
         addOrderReq.setUserId("jd123456789");
         addOrderReq.setAmount(new BigDecimal(1000));
         orderService.addOrder(addOrderReq);
-        System.out.println(DbContext.getDbKey() + ":" + DbContext.getTableIndex());
     }
 }
diff --git a/whatsmars-dbrouter/src/main/java/com/whatsmars/dbrouter/demo/OrderService.java b/whatsmars-dbrouter/src/main/java/com/whatsmars/dbrouter/demo/OrderService.java
index 1a3735b3..7f7a8f88 100644
--- a/whatsmars-dbrouter/src/main/java/com/whatsmars/dbrouter/demo/OrderService.java
+++ b/whatsmars-dbrouter/src/main/java/com/whatsmars/dbrouter/demo/OrderService.java
@@ -1,5 +1,6 @@
 package com.whatsmars.dbrouter.demo;
 
+import com.whatsmars.dbrouter.DbContext;
 import com.whatsmars.dbrouter.DbRoute;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
@@ -19,6 +20,8 @@ public class OrderService {
         order.setUserId(req.getUserId());
         order.setOrderId(order.getUserId() + System.currentTimeMillis());
         order.setAmount(req.getAmount());
+        System.out.println(DbContext.getDbKey() + ":" + DbContext.getTableIndex());
+        order.setTableIndex(DbContext.getTableIndex());
         orderDao.insert(order);
     }
 }
diff --git a/whatsmars-dbrouter/src/main/resources/sqlmap/Order-mapper.xml b/whatsmars-dbrouter/src/main/resources/sqlmap/Order-mapper.xml
index 89e674cc..192b88a6 100644
--- a/whatsmars-dbrouter/src/main/resources/sqlmap/Order-mapper.xml
+++ b/whatsmars-dbrouter/src/main/resources/sqlmap/Order-mapper.xml
@@ -7,7 +7,7 @@
 	</sql>
 
     <select id="insert" parameterType="Order">
-        insert into order_#tableIndex#(userId,orderId,amount) values(#{userId}, #{orderId}, #{amount})
+        insert into order_#{tableIndex} (userId,orderId,amount) values (#{userId}, #{orderId}, #{amount})
     </select>
 
 </mapper>
-- 
GitLab