add update check command type

b2d01f72 · hjxilinx · acce71db · b2d01f72
隐藏空白更改
内联并排

Showing with 3 addition and 5 deletion

src/client/src/tscUtil.c src/client/src/tscUtil.c +3 -5

未找到文件。
--- 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;
 }