ResourceServer: make the resource store the default unified storage backend (#90899)

* make the resource store the default unified storage backend

* add integration tests

* fix test non passing

* Update pkg/storage/unified/sql/test/integration_test.go

Co-authored-by: Ryan McKinley <ryantxu@gmail.com>

* lint

* fix tests

* fix no rows

---------

Co-authored-by: Ryan McKinley <ryantxu@gmail.com>

pkg/storage/unified/sql/service.go

@@ -0,0 +1,134 @@
+package sql
+
+import (
+	"context"
+
+	"github.com/grafana/dskit/services"
+	"github.com/prometheus/client_golang/prometheus"
+	"google.golang.org/grpc/health/grpc_health_v1"
+
+	infraDB "github.com/grafana/grafana/pkg/infra/db"
+	"github.com/grafana/grafana/pkg/infra/log"
+	"github.com/grafana/grafana/pkg/infra/tracing"
+	"github.com/grafana/grafana/pkg/modules"
+	"github.com/grafana/grafana/pkg/services/featuremgmt"
+	"github.com/grafana/grafana/pkg/services/grpcserver"
+	"github.com/grafana/grafana/pkg/services/grpcserver/interceptors"
+	"github.com/grafana/grafana/pkg/setting"
+	"github.com/grafana/grafana/pkg/storage/unified/resource"
+	"github.com/grafana/grafana/pkg/storage/unified/resource/grpc"
+)
+
+var (
+	_ Service = (*service)(nil)
+)
+
+type Service interface {
+	services.NamedService
+}
+
+type service struct {
+	*services.BasicService
+
+	cfg       *setting.Cfg
+	features  featuremgmt.FeatureToggles
+	db        infraDB.DB
+	stopCh    chan struct{}
+	stoppedCh chan error
+
+	handler grpcserver.Provider
+
+	tracing *tracing.TracingService
+
+	authenticator interceptors.Authenticator
+
+	log log.Logger
+}
+
+func ProvideService(
+	cfg *setting.Cfg,
+	features featuremgmt.FeatureToggles,
+	db infraDB.DB,
+	log log.Logger,
+) (*service, error) {
+	tracingCfg, err := tracing.ProvideTracingConfig(cfg)
+	if err != nil {
+		return nil, err
+	}
+	tracingCfg.ServiceName = "unified-storage"
+
+	tracing, err := tracing.ProvideService(tracingCfg)
+	if err != nil {
+		return nil, err
+	}
+
+	authn := &grpc.Authenticator{}
+
+	s := &service{
+		cfg:           cfg,
+		features:      features,
+		stopCh:        make(chan struct{}),
+		authenticator: authn,
+		tracing:       tracing,
+		db:            db,
+		log:           log,
+	}
+
+	// This will be used when running as a dskit service
+	s.BasicService = services.NewBasicService(s.start, s.running, nil).WithName(modules.StorageServer)
+
+	return s, nil
+}
+
+func (s *service) start(ctx context.Context) error {
+	server, err := ProvideResourceServer(s.db, s.cfg, s.features, s.tracing)
+	if err != nil {
+		return err
+	}
+	s.handler, err = grpcserver.ProvideService(s.cfg, s.features, s.authenticator, s.tracing, prometheus.DefaultRegisterer)
+	if err != nil {
+		return err
+	}
+
+	healthService, err := resource.ProvideHealthService(server)
+	if err != nil {
+		return err
+	}
+
+	resource.RegisterResourceStoreServer(s.handler.GetServer(), server)
+	grpc_health_v1.RegisterHealthServer(s.handler.GetServer(), healthService)
+
+	// register reflection service
+	_, err = grpcserver.ProvideReflectionService(s.cfg, s.handler)
+	if err != nil {
+		return err
+	}
+
+	// start the gRPC server
+	go func() {
+		err := s.handler.Run(ctx)
+		if err != nil {
+			s.stoppedCh <- err
+		} else {
+			s.stoppedCh <- nil
+		}
+	}()
+	return nil
+}
+
+// GetAddress returns the address of the gRPC server.
+func (s *service) GetAddress() string {
+	return s.handler.GetAddress()
+}
+
+func (s *service) running(ctx context.Context) error {
+	select {
+	case err := <-s.stoppedCh:
+		if err != nil {
+			return err
+		}
+	case <-ctx.Done():
+		close(s.stopCh)
+	}
+	return nil
+}