From 8c310afa6edebf41126e717852d796fe920cc2a5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Miko=C5=82aj=20Pieczaba?= <mikolaj@pieczaba.me>
Date: Sun, 31 Jul 2022 21:08:12 +0200
Subject: [PATCH] feat: add 'QueryRow' method to DeleteBuilder

---
 delete.go | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/delete.go b/delete.go
index f3f31e6..bd66e1e 100644
--- a/delete.go
+++ b/delete.go
@@ -183,9 +183,25 @@ func (b DeleteBuilder) Query() (*sql.Rows, error) {
 	return data.Query()
 }
 
+func (b DeleteBuilder) QueryRow() RowScanner {
+	data := builder.GetStruct(b).(deleteData)
+	return data.QueryRow()
+}
+
 func (d *deleteData) Query() (*sql.Rows, error) {
 	if d.RunWith == nil {
 		return nil, RunnerNotSet
 	}
 	return QueryWith(d.RunWith, d)
 }
+
+func (d *deleteData) QueryRow() RowScanner {
+	if d.RunWith == nil {
+		return &Row{err: RunnerNotSet}
+	}
+	queryRower, ok := d.RunWith.(QueryRower)
+	if !ok {
+		return &Row{err: RunnerNotQueryRunner}
+	}
+	return QueryRowWith(queryRower, d)
+}