mirror of
https://github.com/lukaszraczylo/go-simple-graphql.git
synced 2026-10-14 00:58:32 +00:00
Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
7623410bfe
|
||
|
|
8e26d79df7
|
@@ -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
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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
@@ -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)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|||||||
Reference in New Issue
Block a user