Compare commits

...
2 Commits
Author SHA1 Message Date
lukaszraczylo 7623410bfe fixup! fixup! ... 2023-04-08 06:20:11 +01:00
lukaszraczylo 8e26d79df7 fixup! ... 2023-04-08 05:58:11 +01:00
3 changed files with 21 additions and 19 deletions
+4 -1
View File
@@ -2,6 +2,7 @@ package gql
import ( import (
"crypto/tls" "crypto/tls"
"fmt"
"time" "time"
"github.com/allegro/bigcache" "github.com/allegro/bigcache"
@@ -103,11 +104,13 @@ func (b *BaseClient) enableCache() {
var err error var err error
b.cache.client, err = bigcache.NewBigCache(bigcache.DefaultConfig(time.Duration(b.cache.ttl) * time.Second)) b.cache.client, err = bigcache.NewBigCache(bigcache.DefaultConfig(time.Duration(b.cache.ttl) * time.Second))
if err != nil { if err != nil {
b.Logger.Error(b, "Error while creating cache client;", "error", err.Error()) fmt.Println(">> Error while creating cache client;", "error", err.Error())
panic(err)
} }
} }
func (b *BaseClient) disableCache() { func (b *BaseClient) disableCache() {
b.Logger.Debug(b, "Disabling cache")
b.cache.enabled = false b.cache.enabled = false
b.cache.client = nil b.cache.client = nil
} }
+16 -17
View File
@@ -3,7 +3,6 @@ package gql
import ( import (
"context" "context"
"fmt" "fmt"
"reflect"
"strconv" "strconv"
"github.com/lukaszraczylo/go-simple-graphql/utils/helpers" "github.com/lukaszraczylo/go-simple-graphql/utils/helpers"
@@ -75,22 +74,22 @@ func (c *BaseClient) Query(queryContent string, queryVariables interface{}, quer
query := c.NewQuery(queryContent, queryVariables) query := c.NewQuery(queryContent, queryVariables)
localClient := *c localClientlocal := *c
localClient := &localClientlocal
// Check for library specific headers // Check for library specific headers
if len(queryHeaders) > 0 { if len(queryHeaders) > 0 {
queryHeadersModified, cache_enabled := c.parseQueryHeaders(queryHeaders) queryHeadersModified, cache_enabled, headers_modified := c.parseQueryHeaders(queryHeaders)
if !reflect.DeepEqual(queryHeadersModified, queryHeaders) { if headers_modified {
if queryHeadersModified["gqlcache"] != nil && c.cache.enabled != cache_enabled { queryHeaders = queryHeadersModified
c.Logger.Debug(c, "Switching cache on as per single-request header") c.Logger.Debug(c, fmt.Sprintf("Switching cache %s as per single-request header", strconv.FormatBool(cache_enabled)))
if cache_enabled {
// Create a new BaseClient instance with the same settings as c, but with the cache enabled/disabled as required // this doesn't make sense - cache is created but its ttl is longer than query execution so it'll be disappeared before it's used
if cache_enabled { localClient.enableCache()
localClient.enableCache() } else {
} else { localClient.disableCache()
localClient.disableCache()
}
} }
} }
} }
@@ -120,7 +119,7 @@ func (c *BaseClient) Query(queryContent string, queryVariables interface{}, quer
return nil, err return nil, err
} }
if localClient.cache.enabled && jsonData != nil && queryHash != "" { if localClient.cache.enabled && c.cache.client != nil && jsonData != nil && queryHash != "" {
err = c.cache.client.Set(queryHash, jsonData) err = c.cache.client.Set(queryHash, jsonData)
if err != nil { if err != nil {
c.Logger.Error(c, "Error while setting cache key;", "error", err.Error()) c.Logger.Error(c, "Error while setting cache key;", "error", err.Error())
@@ -154,12 +153,12 @@ func (c *BaseClient) decodeResponse(jsonData []byte) any {
} }
} }
func (c *BaseClient) parseQueryHeaders(queryHeaders map[string]interface{}) (returnHeaders map[string]interface{}, cache_enabled bool) { func (c *BaseClient) parseQueryHeaders(queryHeaders map[string]interface{}) (returnHeaders map[string]interface{}, cache_enabled bool, headers_modified 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)) cache_enabled, _ = strconv.ParseBool(fmt.Sprintf("%v", v))
cache_enabled = true headers_modified = true
continue continue
} }
if k == "gqlretries" { if k == "gqlretries" {
@@ -168,5 +167,5 @@ func (c *BaseClient) parseQueryHeaders(queryHeaders map[string]interface{}) (ret
} }
returnHeaders[k] = v returnHeaders[k] = v
} }
return returnHeaders, cache_enabled return returnHeaders, cache_enabled, headers_modified
} }
+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)
} }
}) })