refactor(api): query and querybatch in go

go/cmd/api-devserver/main.go

@@ -182,6 +182,7 @@ func runBackend(ctx context.Context, port int) {
 	relationsStore := db.NewRelationsStore(dbClient)
 	if err := api.RunServer(ctx, api.ServerOptions{
 		Port:           port,
+		VerboseLogs:    true,
 		VulnStore:      vulnStore,
 		RelationsStore: relationsStore,
 	}); err != nil {

go/cmd/api/main.go

@@ -8,8 +8,10 @@ import (
 	"log/slog"
 	"os"
 	"os/signal"
+	"strings"
 	"syscall"
 
+	"cloud.google.com/go/compute/metadata"
 	"cloud.google.com/go/datastore"
 	"cloud.google.com/go/storage"
 	"github.com/google/osv.dev/go/internal/api"
@@ -36,8 +38,13 @@ func run() error {
 
 	project := os.Getenv("GOOGLE_CLOUD_PROJECT")
 	if project == "" {
-		logger.ErrorContext(ctx, "GOOGLE_CLOUD_PROJECT environment variable is not set")
-		return errors.New("GOOGLE_CLOUD_PROJECT environment variable is not set")
+		// Fallback to metadata server for Cloud Run
+		var err error
+		project, err = metadata.ProjectIDWithContext(ctx)
+		if err != nil {
+			logger.ErrorContext(ctx, "GOOGLE_CLOUD_PROJECT environment variable is not set")
+			return errors.New("GOOGLE_CLOUD_PROJECT environment variable is not set")
+		}
 	}
 	datastoreID := os.Getenv("DATASTORE_DATABASE_ID") // empty string is the (default) database
 	dbClient, err := datastore.NewClientWithDatabase(ctx, project, datastoreID)
@@ -63,8 +70,11 @@ func run() error {
 	})
 	relationsStore := db.NewRelationsStore(dbClient)
 
+	verboseLogs := strings.ToLower(os.Getenv("OSV_VERBOSE_LOGGING")) == "true"
+
 	return api.RunServer(ctx, api.ServerOptions{
 		Port:           *port,
+		VerboseLogs:    verboseLogs,
 		VulnStore:      vulnStore,
 		RelationsStore: relationsStore,
 	})

go/internal/api/get_vuln_by_id_test.go

@@ -3,10 +3,8 @@ package api
 import (
 	"context"
 	"errors"
-	"iter"
 	"strings"
 	"testing"
-	"time"
 
 	"github.com/google/go-cmp/cmp"
 	"github.com/google/osv.dev/go/internal/models"
@@ -19,6 +17,8 @@ import (
 )
 
 type mockVulnerabilityStore struct {
+	models.UnimplementedVulnerabilityStore
+
 	vuln *osvschema.Vulnerability
 	err  error
 }
@@ -34,22 +34,6 @@ func (m *mockVulnerabilityStore) Get(_ context.Context, _ string) (*osvschema.Vu
 	return m.vuln, nil
 }
 
-func (m *mockVulnerabilityStore) ListBySource(_ context.Context, _ string, _ bool) iter.Seq2[*models.VulnSourceRef, error] {
-	panic("unimplemented")
-}
-
-func (m *mockVulnerabilityStore) GetSourceModified(_ context.Context, _ string) (time.Time, error) {
-	panic("unimplemented")
-}
-
-func (m *mockVulnerabilityStore) GetWithMetadata(_ context.Context, _ string) (*osvschema.Vulnerability, *models.VulnSourceRef, error) {
-	panic("unimplemented")
-}
-
-func (m *mockVulnerabilityStore) Write(_ context.Context, _ models.WriteRequest) error {
-	panic("unimplemented")
-}
-
 type mockRelationsStore struct {
 	aliases *models.GetAliasResult
 	err     error