From b2d01f72e0a008e6b9562731fa52eeaa9288059d Mon Sep 17 00:00:00 2001
From: hjxilinx <hjxilinx@163.com>
Date: Mon, 11 Nov 2019 21:43:48 +0800
Subject: [PATCH] add update check command type

---
 src/client/src/tscUtil.c | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/src/client/src/tscUtil.c b/src/client/src/tscUtil.c
index 9a699b7bb0..547171cffe 100644
--- a/src/client/src/tscUtil.c
+++ b/src/client/src/tscUtil.c
@@ -1824,10 +1824,8 @@ bool tscIsUpdateQuery(STscObj* pObj) {
     return TSDB_CODE_DISCONNECTED;
   }
 
-  SSqlCmd* pCmd = &(pObj->pSql->cmd);
-  if (pCmd->command >= TSDB_SQL_INSERT && pCmd->command <= TSDB_SQL_DROP_DNODE) {
-    return 1;
-  }
+  SSqlCmd* pCmd = &pObj->pSql->cmd;
+  return ((pCmd->command >= TSDB_SQL_INSERT && pCmd->command <= TSDB_SQL_DROP_DNODE) ||
+      TSDB_SQL_USE_DB == pCmd->command) ? 1 : 0;
 
-  return 0;
 }
-- 
GitLab