Implement MigrateDatabase RPC (#5497)

* Add UpdateDeployTargets method to ApplicationStore interface Signed-off-by: Shinnosuke Sawada-Dazai <[email protected]> * Add MigrateDatabase RPC Signed-off-by: Shinnosuke Sawada-Dazai <[email protected]> * Log error when application migration fails in MigrateDatabase RPC Signed-off-by: Shinnosuke Sawada-Dazai <[email protected]> * Fix build error Signed-off-by: Shinnosuke Sawada-Dazai <[email protected]> * Run make gen/code Signed-off-by: Shinnosuke Sawada-Dazai <[email protected]> --------- Signed-off-by: Shinnosuke Sawada-Dazai <[email protected]>
pipe-cd · Jan 21, 2025 · 1fef133 · 1fef133
1 parent 7f7d7ec
commit 1fef133
Show file tree

Hide file tree

Showing 3 changed files with 50 additions and 0 deletions.
diff --git a/pkg/app/server/grpcapi/api.go b/pkg/app/server/grpcapi/api.go
@@ -47,6 +47,7 @@ type apiApplicationStore interface {
 	Disable(ctx context.Context, id string) error
 	UpdateConfigFilename(ctx context.Context, id, filename string) error
 	UpdateConfiguration(ctx context.Context, id, pipedID, platformProvider, configFilename string) error
+	UpdateDeployTargets(ctx context.Context, id string, targets []string) error
 }
 
 type apiDeploymentStore interface {
@@ -1022,6 +1023,33 @@ func (a *API) Encrypt(ctx context.Context, req *apiservice.EncryptRequest) (*api
 	}, nil
 }
 
+func (a *API) MigrateDatabase(ctx context.Context, req *apiservice.MigrateDatabaseRequest) (*apiservice.MigrateDatabaseResponse, error) {
+	if _, err := requireAPIKey(ctx, model.APIKey_READ_WRITE, a.logger); err != nil {
+		return nil, err
+	}
+
+	switch { //nolint:gocritic // we plan to add more cases
+	case req.GetApplication() != nil:
+		if err := a.migrateApplication(ctx, req.GetApplication()); err != nil {
+			a.logger.Error("failed to migrate application", zap.Error(err))
+			return nil, err
+		}
+		return &apiservice.MigrateDatabaseResponse{}, nil
+	}
+	return nil, status.Error(codes.Unimplemented, "Not implemented")
+}
+
+func (a *API) migrateApplication(ctx context.Context, app *apiservice.MigrateDatabaseRequest_Application) error {
+	application, err := getApplication(ctx, a.applicationStore, app.ApplicationId, a.logger)
+	if err != nil {
+		return gRPCStoreError(err, "get application")
+	}
+	if err := a.applicationStore.UpdateDeployTargets(ctx, app.ApplicationId, []string{application.PlatformProvider}); err != nil {
+		return gRPCStoreError(err, "update application")
+	}
+	return nil
+}
+
 // requireAPIKey checks the existence of an API key inside the given context
 // and ensures that it has enough permissions for the give role.
 func requireAPIKey(ctx context.Context, role model.APIKey_Role, logger *zap.Logger) (*model.APIKey, error) {

diff --git a/pkg/datastore/applicationstore.go b/pkg/datastore/applicationstore.go
@@ -198,6 +198,7 @@ type ApplicationStore interface {
 	UpdateBasicInfo(ctx context.Context, id, name, description string, labels map[string]string) error
 	UpdateConfiguration(ctx context.Context, id, pipedID, platformProvider, configFilename string) error
 	UpdatePlatformProvider(ctx context.Context, id string, provider string) error
+	UpdateDeployTargets(ctx context.Context, id string, targets []string) error
 }
 
 type applicationStore struct {
@@ -377,3 +378,10 @@ func (s *applicationStore) UpdatePlatformProvider(ctx context.Context, id string
 		return nil
 	})
 }
+
+func (s *applicationStore) UpdateDeployTargets(ctx context.Context, id string, targets []string) error {
+	return s.update(ctx, id, func(app *model.Application) error {
+		app.DeployTargets = targets
+		return nil
+	})
+}
diff --git a/pkg/datastore/datastoretest/datastore.mock.go b/pkg/datastore/datastoretest/datastore.mock.go