add method AddORFilters for adding multiple filters into one OR statement

Author: Timofey S

main.go

@@ -252,6 +252,35 @@ func (q *Query) AddFilter(name string, m Method, value interface{}) *Query {
 return q
 }
 
+// AddORFilters adds multiple filter into one `OR` statement inside parenteses.
+// E.g. (firstname ILIKE ? OR lastname ILIKE ?)
+func (q *Query) AddORFilters(fn func(query *Query)) *Query {
+_q := New()
+
+fn(_q)
+
+if len(_q.Filters) < 2 {
+return q
+}
+
+firstIdx := 0
+lastIdx := len(_q.Filters) - 1
+
+for i := 0; i < len(_q.Filters); i++ {
+switch i {
+case firstIdx:
+_q.Filters[i].OR = StartOR
+case lastIdx:
+_q.Filters[i].OR = EndOR
+default:
+_q.Filters[i].OR = InOR
+}
+}
+
+q.Filters = append(q.Filters, _q.Filters...)
+return q
+}
+
 // AddFilterRaw adds a filter to Query as SQL condition.
 // This function supports only single condition per one call.
 // If you'd like add more then one conditions you should call this func several times.

main_test.go

@@ -577,3 +577,37 @@ func TestEmptySliceFilterWithAnotherFilter(t *testing.T) {
 q.AddFilter("another_id", EQ, uuid.New().String())
 t.Log(q.SQL("test"))
 }
+
+func TestQuery_AddORFilters(t *testing.T) {
+t.Run("2 OR conditions", func(t *testing.T) {
+q := New().AddFilter("test", EQ, "ok")
+q.AddORFilters(func(query *Query) {
+query.AddFilter("firstname", ILIKE, "*hello*")
+query.AddFilter("lastname", ILIKE, "*hello*")
+})
+out := q.SQL("table")
+t.Log(out)
+assert.Equal(t, `SELECT * FROM table WHERE test = ? AND (firstname ILIKE ? OR lastname ILIKE ?)`, out)
+})
+
+t.Run("3 OR conditions", func(t *testing.T) {
+q := New().AddFilter("test", EQ, "ok")
+q.AddORFilters(func(query *Query) {
+query.AddFilter("firstname", ILIKE, "*hello*")
+query.AddFilter("lastname", ILIKE, "*hello*")
+query.AddFilter("email", ILIKE, "*hello*")
+})
+out := q.SQL("table")
+t.Log(out)
+assert.Equal(t, `SELECT * FROM table WHERE test = ? AND (firstname ILIKE ? OR lastname ILIKE ? OR email ILIKE ?)`, out)
+})
+}
+
+func ExampleQuery_AddORFilters() {
+q := New().AddFilter("test", EQ, "ok")
+q.AddORFilters(func(query *Query) {
+query.AddFilter("firstname", ILIKE, "*hello*")
+query.AddFilter("lastname", ILIKE, "*hello*")
+})
+q.SQL("table") // SELECT * FROM table WHERE test = ? AND (firstname ILIKE ? OR lastname ILIKE ?)
+}