From bfbab3dee2b366dbd4aefa87cb8830cd79f5d934 Mon Sep 17 00:00:00 2001
From: Mitch Alderson <mitch.alderson@atlas.health>
Date: Tue, 29 Oct 2024 13:27:51 -0700
Subject: [PATCH] adds debugging

---
 cache.go | 28 ++++++++++++++++++++++++----
 1 file changed, 24 insertions(+), 4 deletions(-)

diff --git a/cache.go b/cache.go
index eeac62b..2f7705c 100644
--- a/cache.go
+++ b/cache.go
@@ -2,6 +2,8 @@ package gqlgen_cache
 
 import (
 	"context"
+	"encoding/base64"
+	"encoding/json"
 	"errors"
 	"fmt"
 	"github.com/99designs/gqlgen/graphql"
@@ -107,13 +109,31 @@ func (c *fieldCache) findIdField(obj interface{}) (string, error) {
 
 func (c *fieldCache) generateKey(ctx context.Context, obj interface{}) (uint64, string) {
 
+	var id string
+	var err error
+
 	queryContext := graphql.GetOperationContext(ctx)
 	fieldContext := graphql.GetFieldContext(ctx)
 
-	id, err := c.findIdField(obj)
-	if err != nil {
-		c.logger.Debug("id not found", "error", err)
-		return 0, ""
+	c.logger.Debug("generating key", "object", obj, "queryContext", queryContext.RawQuery)
+
+	if obj != nil {
+		id, err = c.findIdField(obj)
+		if err != nil {
+			c.logger.Debug("id not found", "error", err)
+
+			// If we can't find an ID, we will create a hash based on the object
+			b, err := json.Marshal(obj)
+			if err != nil {
+				c.logger.Debug("failed to marshal object", "error", err)
+				return 0, ""
+			}
+
+			id = base64.StdEncoding.EncodeToString(b)
+		}
+
+	} else {
+		id = base64.StdEncoding.EncodeToString([]byte(queryContext.RawQuery))
 	}
 
 	// Create a struct to hold the relevant data