Speed up DoltHub pending PR reads

Author: codex

internal/hosted/authservice_resolver.go

@@ -259,6 +259,7 @@ func (wr *AuthServiceWorkspaceResolver) buildClient(
 
 	db := backend.NewRemoteDBWithClient(proxyClient, upOrg, upDB, wl.ForkOrg, wl.ForkDB, mode)
 	provider := remote.NewDoltHubProviderWithClient(proxyClient)
+	pendingCache := wr.getOrCreatePendingCache(session.ConnectionID, provider, upOrg, upDB)
 
 	branchURL := func(branch string) string {
 		return fmt.Sprintf("https://www.dolthub.com/repositories/%s/%s/data/%s",
@@ -322,9 +323,9 @@ func (wr *AuthServiceWorkspaceResolver) buildClient(
 			}
 			return provider.ClosePR(upOrg, upDB, prID)
 		},
-		ListPendingItems: wr.getOrCreatePendingCache(session.ConnectionID, provider, upOrg, upDB).Get,
+		ListPendingItems: pendingCache.Get,
 		ListPendingItemsContext: func(ctx context.Context) (map[string][]sdk.PendingItem, error) {
-			return wr.getOrCreatePendingCache(session.ConnectionID, provider, upOrg, upDB).GetContext(ctx)
+			return pendingCache.GetContext(ctx)
 		},
 		BranchURL: branchURL,
 		Signing:   wl.Signing,

internal/hosted/authservice_resolver_test.go

@@ -1,10 +1,20 @@
 package hosted
 
 import (
+	"context"
+	"net/http"
 	"testing"
 	"time"
+
+	"github.com/gastownhall/wasteland/internal/dolthubauth"
 )
 
+type roundTripFunc func(*http.Request) (*http.Response, error)
+
+func (fn roundTripFunc) RoundTrip(req *http.Request) (*http.Response, error) {
+	return fn(req)
+}
+
 func TestAuthServiceWorkspaceResolver_InvalidateConnectionClearsPendingCaches(t *testing.T) {
 	resolver := NewAuthServiceWorkspaceResolver(nil, NewSessionStore())
 	cache1 := newPendingUpstreamCache(nil, "hop", "wl-commons", time.Hour)
@@ -24,3 +34,37 @@ func TestAuthServiceWorkspaceResolver_InvalidateConnectionClearsPendingCaches(t
 		t.Fatal("expected unrelated pending cache to remain")
 	}
 }
+
+func TestAuthServiceWorkspaceResolver_BuildClientWarmsPendingCache(t *testing.T) {
+	resolver := NewAuthServiceWorkspaceResolver(nil, NewSessionStore())
+	session := &UserSession{
+		SubjectID:    "subject-1",
+		ConnectionID: "conn-1",
+	}
+	conn := &dolthubauth.ConnectionResponse{
+		ConnectionID: "conn-1",
+		SubjectID:    "subject-1",
+		RigHandle:    "alice",
+	}
+	wl := dolthubauth.WastelandConfig{
+		Upstream: "hop/wl-commons",
+		ForkOrg:  "alice",
+		ForkDB:   "wl-commons",
+		Mode:     "pr",
+	}
+
+	client := &http.Client{Transport: roundTripFunc(func(*http.Request) (*http.Response, error) {
+		return nil, context.Canceled
+	})}
+
+	if _, err := resolver.buildClient(session, conn, client, wl); err != nil {
+		t.Fatalf("buildClient() error = %v", err)
+	}
+
+	key := "conn-1:hop/wl-commons"
+	cache, ok := resolver.pendingCache[key]
+	if !ok {
+		t.Fatalf("expected pending cache %q to be created during build", key)
+	}
+	cache.Stop()
+}