From 07415c1e1846cbb57b46073d897a57b0ce4abf73 Mon Sep 17 00:00:00 2001
From: lacunadream <davidyong12@gmail.com>
Date: Wed, 30 Dec 2020 23:34:09 +0800
Subject: [PATCH] feat: add mysql support for locks

Add pessimistic_write_or_fail and pessimistic_partial_write support for mysql

Closes: #6530
---
 src/query-builder/SelectQueryBuilder.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/query-builder/SelectQueryBuilder.ts b/src/query-builder/SelectQueryBuilder.ts
index 44d45041d0..8fcc1f652b 100644
--- a/src/query-builder/SelectQueryBuilder.ts
+++ b/src/query-builder/SelectQueryBuilder.ts
@@ -1686,14 +1686,14 @@ 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 MysqlDriver) {
                     return " FOR UPDATE SKIP LOCKED";
 
                 } else {
                     throw new LockNotSupportedOnGivenDriverError();
                 }
             case "pessimistic_write_or_fail":
-                if (driver instanceof PostgresDriver) {
+                if (driver instanceof PostgresDriver || driver instanceof MysqlDriver) {
                     return " FOR UPDATE NOWAIT";
                 } else {
                     throw new LockNotSupportedOnGivenDriverError();