Compare commits

...
6 Commits
2 changed files with 14 additions and 9 deletions
+13 -8
View File
@@ -73,19 +73,23 @@ func (c *BaseClient) Query(queryContent string, queryVariables interface{}, quer
var queryHash string var queryHash string
var cachedResponse []byte var cachedResponse []byte
cbc := reflect.ValueOf(*c).Interface().(BaseClient) cacheBaseClient := c
cacheBaseClient := &cbc defer func() {
cacheBaseClient = nil
}()
query := cacheBaseClient.NewQuery(queryContent, queryVariables) query := cacheBaseClient.NewQuery(queryContent, queryVariables)
// Check for library specific headers // Check for library specific headers
if len(queryHeaders) > 0 { if len(queryHeaders) > 0 {
queryHeadersModified := cacheBaseClient.parseQueryHeaders(queryHeaders) queryHeadersModified, cache_enabled := cacheBaseClient.parseQueryHeaders(queryHeaders)
// compare if there are any changes // compare if there are any changes
if !reflect.DeepEqual(queryHeadersModified, queryHeaders) { if !reflect.DeepEqual(queryHeadersModified, queryHeaders) {
if cacheBaseClient.cache.enabled != c.cache.enabled && cacheBaseClient.cache.enabled { if queryHeadersModified["gqlcache"] != nil && c.cache.enabled != cache_enabled {
cacheBaseClient.Logger.Debug(cacheBaseClient, "Switching cache on as per single-request header") cacheBaseClient.Logger.Debug(cacheBaseClient, "Switching cache on as per single-request header")
cbc := reflect.ValueOf(*c).Interface().(BaseClient)
cacheBaseClient := &cbc
cacheBaseClient.enableCache() cacheBaseClient.enableCache()
} }
} }
@@ -118,9 +122,9 @@ func (c *BaseClient) Query(queryContent string, queryVariables interface{}, quer
if cacheBaseClient.cache.enabled && jsonData != nil && queryHash != "" { if cacheBaseClient.cache.enabled && jsonData != nil && queryHash != "" {
c.cache.client.Set(queryHash, jsonData) c.cache.client.Set(queryHash, jsonData)
} else if jsonData == nil { } else if cacheBaseClient.cache.enabled && jsonData == nil {
cacheBaseClient.Logger.Warn(cacheBaseClient, "Response is empty") cacheBaseClient.Logger.Warn(cacheBaseClient, "Response is empty")
} else if queryHash == "" { } else if cacheBaseClient.cache.enabled && queryHash == "" {
cacheBaseClient.Logger.Warn(cacheBaseClient, "Query hash is empty") cacheBaseClient.Logger.Warn(cacheBaseClient, "Query hash is empty")
} }
@@ -147,11 +151,12 @@ func (c *BaseClient) decodeResponse(jsonData []byte) any {
} }
} }
func (c *BaseClient) parseQueryHeaders(queryHeaders map[string]interface{}) (returnHeaders map[string]interface{}) { func (c *BaseClient) parseQueryHeaders(queryHeaders map[string]interface{}) (returnHeaders map[string]interface{}, cache_enabled bool) {
returnHeaders = make(map[string]interface{}) returnHeaders = make(map[string]interface{})
for k, v := range queryHeaders { for k, v := range queryHeaders {
if k == "gqlcache" { if k == "gqlcache" {
c.cache.enabled, _ = strconv.ParseBool(fmt.Sprintf("%v", v)) c.cache.enabled, _ = strconv.ParseBool(fmt.Sprintf("%v", v))
cache_enabled = true
continue continue
} }
if k == "gqlretries" { if k == "gqlretries" {
@@ -160,5 +165,5 @@ func (c *BaseClient) parseQueryHeaders(queryHeaders map[string]interface{}) (ret
} }
returnHeaders[k] = v returnHeaders[k] = v
} }
return returnHeaders return returnHeaders, cache_enabled
} }
+1 -1
View File
@@ -334,7 +334,7 @@ func TestBaseClient_parseQueryHeaders(t *testing.T) {
for _, tt := range tests { for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) { t.Run(tt.name, func(t *testing.T) {
c := NewConnection() c := NewConnection()
if gotReturnHeaders := c.parseQueryHeaders(tt.args.queryHeaders); !reflect.DeepEqual(gotReturnHeaders, tt.wantReturnHeaders) { if gotReturnHeaders, _ := c.parseQueryHeaders(tt.args.queryHeaders); !reflect.DeepEqual(gotReturnHeaders, tt.wantReturnHeaders) {
t.Errorf("BaseClient.parseQueryHeaders() = %v, want %v", gotReturnHeaders, tt.wantReturnHeaders) t.Errorf("BaseClient.parseQueryHeaders() = %v, want %v", gotReturnHeaders, tt.wantReturnHeaders)
} }
}) })