✨ only expose the query when asked to

fumeapp · Jul 28, 2023 · 0b18fa3 · 0b18fa3
1 parent cfb77b9
commit 0b18fa3
Showing 1 changed file with 15 additions and 7 deletions.
diff --git a/aws/dynamo.go b/aws/dynamo.go
@@ -9,12 +9,13 @@ import (
 )
 
 type DynamoIndexQuery struct {
-	Region     string
-	Table      string
-	Wheres     DynamoIndexWheres
-	OrderField string
-	Direction  string
-	SetLimit   *int32
+	Region      string
+	Table       string
+	Wheres      DynamoIndexWheres
+	OrderField  string
+	Direction   string
+	SetLimit    *int32
+	ExposeQuery bool
 }
 
 type DynamoIndexWhere struct {
@@ -43,6 +44,11 @@ func DynamoIndex(region string, table string) DynamoIndexQuery {
 	return query
 }
 
+func (q DynamoIndexQuery) ShowQuery(value bool) DynamoIndexQuery {
+	q.ExposeQuery = value
+	return q
+}
+
 func (q DynamoIndexQuery) WhereStr(field string, value string) DynamoIndexQuery {
 	q.Wheres.Add(field, "string", value)
 	return q
@@ -122,7 +128,9 @@ func (q DynamoIndexQuery) Get(out interface{}) (*DynamoIndexResults, error) {
 	var results DynamoIndexResults
 	results.Items = out
 	results.NextToken = p.NextToken
-	results.Query = &query
+	if q.ExposeQuery {
+		results.Query = &query
+	}
 
 	return &results, nil
 }