K8s: Add App SDK installer (#107773)

2025-07-21 13:32:15 -04:00
parent 548964f728
commit e5d2f92384
107 changed files with 2772 additions and 1891 deletions
@@ -0,0 +1,95 @@
+package appinstaller
+
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	"k8s.io/apimachinery/pkg/runtime/schema"
+	"k8s.io/apiserver/pkg/registry/generic"
+	genericregistry "k8s.io/apiserver/pkg/registry/generic/registry"
+	genericapiserver "k8s.io/apiserver/pkg/server"
+
+	appsdkapiserver "github.com/grafana/grafana-app-sdk/k8s/apiserver"
+	"github.com/grafana/grafana-app-sdk/logging"
+	grafanaregistry "github.com/grafana/grafana/pkg/apiserver/registry/generic"
+	grafanarest "github.com/grafana/grafana/pkg/apiserver/rest"
+	"github.com/grafana/grafana/pkg/services/apiserver/builder"
+	"github.com/grafana/grafana/pkg/services/apiserver/endpoints/request"
+	grafanaapiserveroptions "github.com/grafana/grafana/pkg/services/apiserver/options"
+	"github.com/grafana/grafana/pkg/storage/legacysql/dualwrite"
+)
+
+var _ appsdkapiserver.GenericAPIServer = (*serverWrapper)(nil)
+
+type serverWrapper struct {
+	ctx context.Context
+	appsdkapiserver.GenericAPIServer
+	installer         appsdkapiserver.AppInstaller
+	restOptionsGetter generic.RESTOptionsGetter
+	storageOpts       *grafanaapiserveroptions.StorageOptions
+	kvStore           grafanarest.NamespacedKVStore
+	lock              serverLock
+	namespaceMapper   request.NamespaceMapper
+	dualWriteService  dualwrite.Service
+	dualWriterMetrics *grafanarest.DualWriterMetrics
+	builderMetrics    *builder.BuilderMetrics
+}
+
+func (s *serverWrapper) InstallAPIGroup(apiGroupInfo *genericapiserver.APIGroupInfo) error {
+	log := logging.FromContext(s.ctx)
+	legacyProvider, ok := s.installer.(LegacyStorageProvider)
+	if !ok {
+		return s.GenericAPIServer.InstallAPIGroup(apiGroupInfo)
+	}
+	for v, storageMap := range apiGroupInfo.VersionedResourcesStorageMap {
+		for storagePath, restStorage := range storageMap {
+			genericStorage, ok := restStorage.(*genericregistry.Store)
+			if !ok {
+				log.Error("Expected generic registry store", "storagePath", storagePath, "version", v)
+				continue
+			}
+			resource, err := getResourceFromStoragePath(storagePath)
+			if err != nil {
+				return err
+			}
+			gr := schema.GroupResource{
+				Group:    s.installer.ManifestData().Group,
+				Resource: resource,
+			}
+			genericStorage.KeyRootFunc = grafanaregistry.KeyRootFunc(gr)
+			genericStorage.KeyFunc = grafanaregistry.NamespaceKeyFunc(gr)
+			genericStorage.UpdateStrategy = &updateStrategyWrapper{
+				RESTUpdateStrategy: genericStorage.UpdateStrategy,
+			}
+
+			dw, err := NewDualWriter(
+				s.ctx,
+				gr,
+				s.storageOpts,
+				legacyProvider.GetLegacyStorage(gr.WithVersion(v)),
+				grafanarest.Storage(genericStorage),
+				s.kvStore,
+				s.lock,
+				s.namespaceMapper,
+				s.dualWriteService,
+				s.dualWriterMetrics,
+				s.builderMetrics,
+			)
+			if err != nil {
+				return err
+			}
+			apiGroupInfo.VersionedResourcesStorageMap[v][storagePath] = dw
+		}
+	}
+
+	return s.GenericAPIServer.InstallAPIGroup(apiGroupInfo)
+}
+
+func getResourceFromStoragePath(storagePath string) (string, error) {
+	parts := strings.Split(storagePath, "/")
+	if len(parts) < 1 {
+		return "", fmt.Errorf("invalid storage path: %s", storagePath)
+	}
+	return parts[0], nil
+}