From 6728918242f689ad50599345e586dd0e7748b33e Mon Sep 17 00:00:00 2001
From: warren <thwjd2717@gmail.com>
Date: Sun, 1 Sep 2024 22:38:16 +0900
Subject: [PATCH] refactor: fix signiture

---
 finisher_api.go | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/finisher_api.go b/finisher_api.go
index f97571ed..7b3dd234 100644
--- a/finisher_api.go
+++ b/finisher_api.go
@@ -1,6 +1,7 @@
 package gorm
 
 import (
+	"context"
 	"database/sql"
 	"errors"
 	"fmt"
@@ -522,7 +523,7 @@ func (db *DB) Rows() (*sql.Rows, error) {
 }
 
 // Scan scans selected value to the struct dest
-func (db *DB) Scan(dest interface{}) (tx *DB) {
+func (db *DB) Scan(ctx context.Context, dest interface{}) (tx *DB) {
 	config := *db.Config
 	currentLogger, newLogger := config.Logger, logger.Recorder.New()
 	config.Logger = newLogger
@@ -530,6 +531,10 @@ func (db *DB) Scan(dest interface{}) (tx *DB) {
 	tx = db.getInstance()
 	tx.Config = &config
 
+	if ctx != nil {
+		tx = tx.WithContext(ctx)
+	}
+
 	if rows, err := tx.Rows(); err == nil {
 		if rows.Next() {
 			tx.ScanRows(rows, dest)
@@ -544,7 +549,7 @@ func (db *DB) Scan(dest interface{}) (tx *DB) {
 		return newLogger.SQL, tx.RowsAffected
 	}, tx.Error)
 	tx.Logger = currentLogger
-	return
+	return tx
 }
 
 // Pluck queries a single column from a model, returning in the slice dest. E.g.: