From ab39e6ce7caff8f5a45d0be354700680034a2f12 Mon Sep 17 00:00:00 2001
From: yedf2 <120050102@qq.com>
Date: Mon, 15 Nov 2021 09:54:22 +0800
Subject: [PATCH] add XaSqlTimeoutMs

---
 dtmcli/dtmimp/vars.go |  3 +++
 dtmcli/types.go       | 10 ++++++++++
 2 files changed, 13 insertions(+)

diff --git a/dtmcli/dtmimp/vars.go b/dtmcli/dtmimp/vars.go
index 6dc42c3..ee9bad6 100644
--- a/dtmcli/dtmimp/vars.go
+++ b/dtmcli/dtmimp/vars.go
@@ -18,6 +18,9 @@ var ErrFailure = errors.New("FAILURE")
 // ErrOngoing error of ONGOING
 var ErrOngoing = errors.New("ONGOING")
 
+// XaSqlTimeoutMs milliseconds for Xa sql to timeout
+var XaSqlTimeoutMs = 15000
+
 // MapSuccess HTTP result of SUCCESS
 var MapSuccess = map[string]interface{}{"dtm_result": ResultSuccess}
 
diff --git a/dtmcli/types.go b/dtmcli/types.go
index 29f8575..ccd34d2 100644
--- a/dtmcli/types.go
+++ b/dtmcli/types.go
@@ -37,3 +37,13 @@ func SetCurrentDBType(dbType string) {
 func GetCurrentDBType() string {
 	return dtmimp.GetCurrentDBType()
 }
+
+// SetXaSqlTimeoutMs set XaSqlTimeoutMs
+func XaSqlTimeoutMs(ms int) {
+	dtmimp.XaSqlTimeoutMs = ms
+}
+
+// GetXaSqlTimeoutMs get XaSqlTimeoutMs
+func GetXaSqlTimeoutMs() int {
+	return dtmimp.XaSqlTimeoutMs
+}