From d4ce4f91a4695283d4eb8760ce2fe7c24e04cd2f Mon Sep 17 00:00:00 2001
From: Tang Cheng <cheng.tang@supwisdom.com>
Date: Thu, 25 Jul 2019 13:57:44 +0800
Subject: [PATCH] =?utf8?q?=E4=BF=AE=E6=94=B9=E6=9B=B4=E6=96=B0=E5=95=86?=
 =?utf8?q?=E6=88=B7=E4=BD=99=E9=A2=9D=E4=BB=BB=E5=8A=A1=E7=9A=84=E5=A4=9A?=
 =?utf8?q?=E7=A7=9F=E6=88=B7=E5=A4=84=E7=90=86=E6=9C=BA=E5=88=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

---
 .../main/kotlin/com/supwisdom/dlpay/api/async_tasks.kt   | 9 ++++++---
 .../supwisdom/dlpay/api/service/transaction_service.kt   | 2 +-
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/payapi/src/main/kotlin/com/supwisdom/dlpay/api/async_tasks.kt b/payapi/src/main/kotlin/com/supwisdom/dlpay/api/async_tasks.kt
index 5b9745ac..759e0478 100644
--- a/payapi/src/main/kotlin/com/supwisdom/dlpay/api/async_tasks.kt
+++ b/payapi/src/main/kotlin/com/supwisdom/dlpay/api/async_tasks.kt
@@ -4,11 +4,13 @@ import com.supwisdom.dlpay.agent.AgentCode
 import com.supwisdom.dlpay.agent.DtlStatus
 import com.supwisdom.dlpay.agent.citizencard.YnrccUtil
 import com.supwisdom.dlpay.api.domain.TDtlQuery
+import com.supwisdom.dlpay.api.domain.TShopdtl
 import com.supwisdom.dlpay.api.domain.TTransactionMain
 import com.supwisdom.dlpay.api.repositories.ShopaccService
 import com.supwisdom.dlpay.api.service.DtlQueryResultService
 import com.supwisdom.dlpay.api.service.TransactionServiceProxy
 import com.supwisdom.dlpay.framework.service.SystemUtilService
+import com.supwisdom.dlpay.framework.tenant.TenantContext
 import com.supwisdom.dlpay.framework.util.ApplicationUtil
 import com.supwisdom.dlpay.framework.util.TradeDict
 import com.supwisdom.dlpay.util.ConstantUtil
@@ -59,8 +61,9 @@ class ShopAccBalanceAsyncTask {
     private lateinit var shopaccService: ShopaccService
 
     @Async("shopAccBalanceUpdater")
-    fun updateShopBalance(shopdtlRefno: String) {
-        shopaccService.recalcShopBalance(shopdtlRefno, true)
+    fun updateShopBalance(shopdtl: TShopdtl) {
+        TenantContext.setTenantSchema(shopdtl.tenantid)
+        shopaccService.recalcShopBalance(shopdtl.refno, true)
     }
 }
 
@@ -100,7 +103,7 @@ class AgentQueryResultTask {
             }
 
             val service = ApplicationUtil.findAgentPayService(applicationContext,
-                    transaction.sourceType+ "Agent")
+                    transaction.sourceType + "Agent")
 
             logger.info("refno=[${transaction.refno}]开始第" + (qcnt + 1) + "次查询支付结果：")
             val resp = service.query(transaction)
diff --git a/payapi/src/main/kotlin/com/supwisdom/dlpay/api/service/transaction_service.kt b/payapi/src/main/kotlin/com/supwisdom/dlpay/api/service/transaction_service.kt
index 76abfd8b..eb08287b 100644
--- a/payapi/src/main/kotlin/com/supwisdom/dlpay/api/service/transaction_service.kt
+++ b/payapi/src/main/kotlin/com/supwisdom/dlpay/api/service/transaction_service.kt
@@ -87,7 +87,7 @@ class TransactionServiceProxy {
     fun success(refno: String, sourcetypeRefno: String, accdateUpdate: Boolean? = true): TTransactionMain {
         return transactionService.success(refno, sourcetypeRefno, accdateUpdate).also {
             if (it.shop) {
-                shopAccBalanceAsyncTask.updateShopBalance(it.refno)
+                shopAccBalanceAsyncTask.updateShopBalance(it.shopDtl)
             }
         }
     }
-- 
2.17.1