feat: support with up in db2

Author: taozhi8833998

pegjs/db2.pegjs

@@ -1046,7 +1046,8 @@ select_stmt_nake
     g:group_by_clause?  __
     h:having_clause?    __
     o:order_by_clause?  __
-    l:limit_clause? {
+    l:limit_clause? 
+    iso:isolation_clause? __ {
       if(f) f.forEach(info => info.table && tableList.add(`select::${info.db}::${info.table}`));
       return {
           with: cte,
@@ -1059,7 +1060,8 @@ select_stmt_nake
           groupby: g,
           having: h,
           orderby: o,
-          limit: l
+          limit: l,
+          isolation: iso,
       };
   }
 
@@ -1341,6 +1343,14 @@ limit_clause
     }
   }
 
+isolation_clause
+  = KW_WITH __ is:('CS'i / 'UR'i / 'RS'i / 'RR'i) {
+    return {
+      type: 'isolation',
+      keyword: 'with',
+      expr: { type: 'origin', value: is },
+    }
+  }
 update_stmt
   = KW_UPDATE    __
     t:table_ref_list __

src/select.js

@@ -3,7 +3,7 @@ import { columnsToSQL } from './column'
 import { limitToSQL } from './limit'
 import { withToSQL } from './with'
 import { tablesToSQL } from './tables'
-import { hasVal, commonOptionConnector, connector, identifierToSql, topToSQL, toUpper } from './util'
+import { hasVal, commonOptionConnector, connector, identifierToSql, topToSQL, toUpper, literalToSQL } from './util'
 import { collateToSQL } from './collate'
 
 function distinctToSQL(distinct) {
@@ -67,6 +67,7 @@ function selectToSQL(stmt) {
     groupby,
     having,
     into = {},
+    isolation,
     limit,
     options,
     orderby,
@@ -100,6 +101,7 @@ function selectToSQL(stmt) {
   clauses.push(orderOrPartitionByToSQL(orderby, 'order by'))
   clauses.push(collateToSQL(collate))
   clauses.push(limitToSQL(limit))
+  if (isolation) clauses.push(commonOptionConnector(isolation.keyword, literalToSQL, isolation.expr))
   clauses.push(toUpper(lockingRead))
   if (position === 'end') clauses.push(intoSQL)
   clauses.push(forXmlToSQL(forXml))

test/db2.spec.js

@@ -27,6 +27,13 @@ describe('db2', () => {
         'SELECT CUSTOMER_NUMBER FROM ORG.CUSTOMER_INFO_1 EXCEPT SELECT CUSTOMER_NUMBER FROM ORG.CUSTOMER_BACKUP'
       ]
     },
+    {
+      title: 'isolation clause',
+      sql: [
+        'SELECT * FROM STAFF WITH UR',
+        'SELECT * FROM STAFF WITH UR'
+      ]
+    },
   ]
   SQL_LIST.forEach(sqlInfo => {
     const { title, sql } = sqlInfo