Refactor query_builder

core-go · Jul 13, 2024 · 1cf3cfa · 1cf3cfa
1 parent d616dd1
commit 1cf3cfa
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/query/builder/builder.go b/query/builder/builder.go
@@ -132,7 +132,14 @@ func Build(filter interface{}, resultModelType reflect.Type, arrFields []string,
 			} else if key == "like" {
 				query = append(query, bson.E{Key: bsonName, Value: primitive.Regex{Pattern: fmt.Sprintf("\\w*%v\\w*", psv)}})
 			} else {
-				query = append(query, bson.E{Key: bsonName, Value: primitive.Regex{Pattern: fmt.Sprintf("^%v", psv)}})
+				opr, ok2 := Operators[key]
+				if ok2 {
+					dQuery := bson.M{}
+					dQuery[opr] = psv
+					query = append(query, bson.E{Key: bsonName, Value: dQuery})
+				} else {
+					query = append(query, bson.E{Key: bsonName, Value: primitive.Regex{Pattern: fmt.Sprintf("^%v", psv)}})
+				}
 			}
 		} else if kind == reflect.Slice {
 			if field.Len() > 0 {