Merge pull request #506 from uptrace/feat/apply-query-builder

feat: add ApplyQueryBuilder
uptrace · Apr 5, 2022 · 2e4d130 · 2e4d130
2 parents 98d111b + 582eca0
commit 2e4d130
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/query_delete.go b/query_delete.go
@@ -293,11 +293,17 @@ func (q *DeleteQuery) QueryBuilder() QueryBuilder {
 	return &deleteQueryBuilder{q}
 }
 
+func (q *DeleteQuery) ApplyQueryBuilder(fn func(QueryBuilder) QueryBuilder) *DeleteQuery {
+	return fn(q.QueryBuilder()).Unwrap().(*DeleteQuery)
+}
+
 type deleteQueryBuilder struct {
 	*DeleteQuery
 }
 
-func (q *deleteQueryBuilder) WhereGroup(sep string, fn func(QueryBuilder) QueryBuilder) QueryBuilder {
+func (q *deleteQueryBuilder) WhereGroup(
+	sep string, fn func(QueryBuilder) QueryBuilder,
+) QueryBuilder {
 	q.DeleteQuery = q.DeleteQuery.WhereGroup(sep, func(qs *DeleteQuery) *DeleteQuery {
 		return fn(q).(*deleteQueryBuilder).DeleteQuery
 	})

diff --git a/query_select.go b/query_select.go
@@ -926,6 +926,10 @@ func (q *SelectQuery) QueryBuilder() QueryBuilder {
 	return &selectQueryBuilder{q}
 }
 
+func (q *SelectQuery) ApplyQueryBuilder(fn func(QueryBuilder) QueryBuilder) *SelectQuery {
+	return fn(q.QueryBuilder()).Unwrap().(*SelectQuery)
+}
+
 type selectQueryBuilder struct {
 	*SelectQuery
 }

diff --git a/query_update.go b/query_update.go
@@ -493,11 +493,17 @@ func (q *UpdateQuery) QueryBuilder() QueryBuilder {
 	return &updateQueryBuilder{q}
 }
 
+func (q *UpdateQuery) ApplyQueryBuilder(fn func(QueryBuilder) QueryBuilder) *UpdateQuery {
+	return fn(q.QueryBuilder()).Unwrap().(*UpdateQuery)
+}
+
 type updateQueryBuilder struct {
 	*UpdateQuery
 }
 
-func (q *updateQueryBuilder) WhereGroup(sep string, fn func(QueryBuilder) QueryBuilder) QueryBuilder {
+func (q *updateQueryBuilder) WhereGroup(
+	sep string, fn func(QueryBuilder) QueryBuilder,
+) QueryBuilder {
 	q.UpdateQuery = q.UpdateQuery.WhereGroup(sep, func(qs *UpdateQuery) *UpdateQuery {
 		return fn(q).(*updateQueryBuilder).UpdateQuery
 	})