optimizing badger cache, won a 10-15% improvement in most benchmarks

pkg/dgraph/identity.go

@@ -0,0 +1,44 @@
+package dgraph
+
+import (
+	"fmt"
+
+	"next.orly.dev/pkg/crypto/keys"
+)
+
+// Relay identity methods
+// We use the marker system to store the relay's private key
+
+const relayIdentityMarkerKey = "relay_identity_secret"
+
+// GetRelayIdentitySecret retrieves the relay's identity secret key
+func (d *D) GetRelayIdentitySecret() (skb []byte, err error) {
+	return d.GetMarker(relayIdentityMarkerKey)
+}
+
+// SetRelayIdentitySecret sets the relay's identity secret key
+func (d *D) SetRelayIdentitySecret(skb []byte) error {
+	return d.SetMarker(relayIdentityMarkerKey, skb)
+}
+
+// GetOrCreateRelayIdentitySecret retrieves or creates the relay identity
+func (d *D) GetOrCreateRelayIdentitySecret() (skb []byte, err error) {
+	skb, err = d.GetRelayIdentitySecret()
+	if err == nil {
+		return skb, nil
+	}
+
+	// Generate new identity
+	skb, err = keys.GenerateSecretKey()
+	if err != nil {
+		return nil, fmt.Errorf("failed to generate identity: %w", err)
+	}
+
+	// Store it
+	if err = d.SetRelayIdentitySecret(skb); err != nil {
+		return nil, fmt.Errorf("failed to store identity: %w", err)
+	}
+
+	d.Logger.Infof("generated new relay identity")
+	return skb, nil
+}