allow lock modes for cockroachdb

fixes typeorm#8249
wodka · Oct 6, 2021 · 11e37c2 · 11e37c2
1 parent 8615733
commit 11e37c2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/query-builder/SelectQueryBuilder.ts b/src/query-builder/SelectQueryBuilder.ts
@@ -1693,7 +1693,7 @@ export class SelectQueryBuilder<Entity> extends QueryBuilder<Entity> implements
                 if (driver instanceof MysqlDriver || driver instanceof AuroraDataApiDriver) {
                     return " LOCK IN SHARE MODE";
 
-                } else if (driver instanceof PostgresDriver) {
+                } else if (driver instanceof PostgresDriver || driver instanceof CockroachDriver) {
                     return " FOR SHARE" + lockTablesClause;
 
                 } else if (driver instanceof OracleDriver) {
@@ -1720,7 +1720,7 @@ export class SelectQueryBuilder<Entity> extends QueryBuilder<Entity> implements
                     throw new LockNotSupportedOnGivenDriverError();
                 }
             case "pessimistic_partial_write":
-                if (driver instanceof PostgresDriver) {
+                if (driver instanceof PostgresDriver || driver instanceof CockroachDriver) {
                     return " FOR UPDATE" + lockTablesClause + " SKIP LOCKED";
 
                 } else if (driver instanceof MysqlDriver) {