lukaszraczylo/go-simple-graphql
1
0
Fork 0
mirror of https://github.com/lukaszraczylo/go-simple-graphql.git synced 2026-10-14 00:58:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

...

Browse Source
This commit is contained in:
lukaszraczylo
2023-04-08 05:44:23 +01:00
parent dfb082ad51
commit aa25c43795
2 changed files with 34 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
base.go
+5
View File
@@ -106,3 +106,8 @@ func (b *BaseClient) enableCache() {
b.Logger.Error(b, "Error while creating cache client;", "error", err.Error()) b.Logger.Error(b, "Error while creating cache client;", "error", err.Error())
} }
} }
func (b *BaseClient) disableCache() {
b.cache.enabled = false
b.cache.client = nil
}
query.go
+29 -26
View File
@@ -73,62 +73,65 @@ func (c *BaseClient) Query(queryContent string, queryVariables interface{}, quer
var queryHash string var queryHash string
var cachedResponse []byte var cachedResponse []byte
cacheBaseClient := c query := c.NewQuery(queryContent, queryVariables)
defer func() {
cacheBaseClient = nil
}()
query := cacheBaseClient.NewQuery(queryContent, queryVariables) localClient := *c
// Check for library specific headers // Check for library specific headers
if len(queryHeaders) > 0 { if len(queryHeaders) > 0 {
queryHeadersModified, cache_enabled := cacheBaseClient.parseQueryHeaders(queryHeaders) queryHeadersModified, cache_enabled := c.parseQueryHeaders(queryHeaders)
// compare if there are any changes
if !reflect.DeepEqual(queryHeadersModified, queryHeaders) { if !reflect.DeepEqual(queryHeadersModified, queryHeaders) {
if queryHeadersModified["gqlcache"] != nil && c.cache.enabled != cache_enabled { if queryHeadersModified["gqlcache"] != nil && c.cache.enabled != cache_enabled {
cacheBaseClient.Logger.Debug(cacheBaseClient, "Switching cache on as per single-request header") c.Logger.Debug(c, "Switching cache on as per single-request header")
cbc := reflect.ValueOf(*c).Interface().(BaseClient)
cacheBaseClient := &cbc // Create a new BaseClient instance with the same settings as c, but with the cache enabled/disabled as required
cacheBaseClient.enableCache() if cache_enabled {
localClient.enableCache()
} else {
localClient.disableCache()
}
} }
} }
} }
if cacheBaseClient.cache.enabled { if localClient.cache.enabled {
queryHash = strutil.Md5(fmt.Sprintf("%s-%+v", query.compiledQuery, queryHeaders)) queryHash = strutil.Md5(fmt.Sprintf("%s-%+v", query.compiledQuery, queryHeaders))
cacheBaseClient.Logger.Debug(cacheBaseClient, "Hash calculated;", "hash:", queryHash) c.Logger.Debug(c, "Hash calculated;", "hash:", queryHash)
cachedResponse = c.cacheLookup(queryHash) cachedResponse = c.cacheLookup(queryHash)
if cachedResponse != nil { if cachedResponse != nil {
cacheBaseClient.Logger.Debug(cacheBaseClient, "Found cached response") c.Logger.Debug(c, "Found cached response")
return cacheBaseClient.decodeResponse(cachedResponse), nil return c.decodeResponse(cachedResponse), nil
} else { } else {
cacheBaseClient.Logger.Debug(cacheBaseClient, "No cached response found") c.Logger.Debug(c, "No cached response found")
} }
} }
response, err := cacheBaseClient.executeQuery(query.compiledQuery, queryHeaders) response, err := localClient.executeQuery(query.compiledQuery, queryHeaders)
if err != nil { if err != nil {
cacheBaseClient.Logger.Error(cacheBaseClient, "Error while executing query;", "error", err.Error()) c.Logger.Error(c, "Error while executing query;", "error", err.Error())
return nil, err return nil, err
} }
jsonData, err := json.Marshal(response) jsonData, err := json.Marshal(response)
if err != nil { if err != nil {
cacheBaseClient.Logger.Error(cacheBaseClient, "Error while converting to json;", "error", err.Error()) c.Logger.Error(c, "Error while converting to json;", "error", err.Error())
return nil, err return nil, err
} }
if cacheBaseClient.cache.enabled && jsonData != nil && queryHash != "" { if localClient.cache.enabled && jsonData != nil && queryHash != "" {
c.cache.client.Set(queryHash, jsonData) err = c.cache.client.Set(queryHash, jsonData)
} else if cacheBaseClient.cache.enabled && jsonData == nil { if err != nil {
cacheBaseClient.Logger.Warn(cacheBaseClient, "Response is empty") c.Logger.Error(c, "Error while setting cache key;", "error", err.Error())
} else if cacheBaseClient.cache.enabled && queryHash == "" { }
cacheBaseClient.Logger.Warn(cacheBaseClient, "Query hash is empty") } else if localClient.cache.enabled && c.cache.enabled && jsonData == nil {
c.Logger.Warn(c, "Response is empty")
} else if localClient.cache.enabled && c.cache.enabled && queryHash == "" {
c.Logger.Warn(c, "Query hash is empty")
} }
return cacheBaseClient.decodeResponse(jsonData), err return c.decodeResponse(jsonData), err
} }
func (c *BaseClient) decodeResponse(jsonData []byte) any { func (c *BaseClient) decodeResponse(jsonData []byte) any {