hotfix: ignore transaction

traP-jp · Jan 26, 2025 · e15821e · e15821e
1 parent 7873c4d
commit e15821e
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 0 deletions.
diff --git a/template/driver.go b/template/driver.go
@@ -120,6 +120,8 @@ func (c *cacheConn) Prepare(rawQuery string) (driver.Stmt, error) {
 			PurgeAllCaches()
 		}
 		return c.inner.Prepare(rawQuery)
+	} else if strings.Contains(normalizedQuery, "FOR UPDATE") {
+		return c.inner.Prepare(rawQuery)
 	}
 
 	if queryInfo.Type == domains.CachePlanQueryType_SELECT && !queryInfo.Select.Cache {

diff --git a/template/driver.tmpl b/template/driver.tmpl
@@ -119,6 +119,8 @@ func (c *cacheConn) Prepare(rawQuery string) (driver.Stmt, error) {
 			PurgeAllCaches()
 		}
 		return c.inner.Prepare(rawQuery)
+	} else if strings.Contains(normalizedQuery, "FOR UPDATE") {
+		return c.inner.Prepare(rawQuery)
 	}
 
 	if queryInfo.Type == domains.CachePlanQueryType_SELECT && !queryInfo.Select.Cache {

diff --git a/template/stmt.go b/template/stmt.go
@@ -389,6 +389,8 @@ func (c *cacheConn) QueryContext(ctx context.Context, rawQuery string, nvargs []
 	queryInfo, ok := queryMap[normalizedQuery]
 	if !ok {
 		return inner.QueryContext(ctx, rawQuery, nvargs)
+	} else if strings.Contains(normalizedQuery, "FOR UPDATE") {
+		return inner.QueryContext(ctx, rawQuery, nvargs)
 	}
 	if queryInfo.Type != domains.CachePlanQueryType_SELECT || !queryInfo.Select.Cache {
 		return inner.QueryContext(ctx, rawQuery, nvargs)

diff --git a/template/stmt.tmpl b/template/stmt.tmpl
@@ -389,6 +389,8 @@ func (c *cacheConn) QueryContext(ctx context.Context, rawQuery string, nvargs []
 	queryInfo, ok := queryMap[normalizedQuery]
 	if !ok {
 		return inner.QueryContext(ctx, rawQuery, nvargs)
+	} else if strings.Contains(normalizedQuery, "FOR UPDATE") {
+		return inner.QueryContext(ctx, rawQuery, nvargs)
 	}
 	if queryInfo.Type != domains.CachePlanQueryType_SELECT || !queryInfo.Select.Cache {
 		return inner.QueryContext(ctx, rawQuery, nvargs)