Add feature to restart piped via web console (API)

pkg/app/server/grpcapi/piped_api.go

@@ -69,7 +69,7 @@ type pipedApiDeploymentChainStore interface {
 
 type pipedApiPipedStore interface {
 	Get(ctx context.Context, id string) (*model.Piped, error)
-	UpdateMetadata(ctx context.Context, id, version, config string, cps []*model.Piped_CloudProvider, repos []*model.ApplicationGitRepository, se *model.Piped_SecretEncryption, startedAt int64) error
+	UpdateMetadata(ctx context.Context, id, version, config string, cps []*model.Piped_CloudProvider, repos []*model.ApplicationGitRepository, se *model.Piped_SecretEncryption, needRestart bool, startedAt int64) error
 }
 
 type pipedApiEventStore interface {
@@ -179,6 +179,7 @@ func (a *PipedAPI) ReportPipedMeta(ctx context.Context, req *pipedservice.Report
 		req.CloudProviders,
 		req.Repositories,
 		req.SecretEncryption,
+		false,
 		now,
 	); err != nil {
 		return nil, gRPCEntityOperationError(err, fmt.Sprintf("update metadata of piped %s", pipedID))
@@ -864,6 +865,22 @@ func (a *PipedAPI) PutLatestAnalysisResult(ctx context.Context, req *pipedservic
 	return &pipedservice.PutLatestAnalysisResultResponse{}, nil
 }
 
+func (a *PipedAPI) GetNeedRestart(ctx context.Context, _ *pipedservice.GetNeedRestartRequest) (*pipedservice.GetNeedRestartResponse, error) {
+	_, pipedID, _, err := rpcauth.ExtractPipedToken(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	piped, err := getPiped(ctx, a.pipedStore, pipedID, a.logger)
+	if err != nil {
+		return nil, err
+	}
+
+	return &pipedservice.GetNeedRestartResponse{
+		NeedRestart: piped.NeedRestart,
+	}, nil
+}
+
 func (a *PipedAPI) GetDesiredVersion(ctx context.Context, _ *pipedservice.GetDesiredVersionRequest) (*pipedservice.GetDesiredVersionResponse, error) {
 	_, pipedID, _, err := rpcauth.ExtractPipedToken(ctx)
 	if err != nil {

pkg/app/server/grpcapi/web_api.go

@@ -75,6 +75,7 @@ type webApiPipedStore interface {
 	AddKey(ctx context.Context, id, keyHash, creator string, createdAt time.Time) error
 	DeleteOldKeys(ctx context.Context, id string) error
 	UpdateInfo(ctx context.Context, id, name, desc string) error
+	RestartPiped(ctx context.Context, id string) error
 	EnablePiped(ctx context.Context, id string) error
 	DisablePiped(ctx context.Context, id string) error
 	UpdateDesiredVersion(ctx context.Context, id, version string) error
@@ -258,6 +259,13 @@ func (a *WebAPI) DeleteOldPipedKeys(ctx context.Context, req *webservice.DeleteO
 	return &webservice.DeleteOldPipedKeysResponse{}, nil
 }
 
+func (a *WebAPI) RestartPiped(ctx context.Context, req *webservice.RestartPipedRequest) (*webservice.RestartPipedResponse, error) {
+	if err := a.updatePiped(ctx, req.PipedId, a.pipedStore.RestartPiped); err != nil {
+		return nil, err
+	}
+	return &webservice.RestartPipedResponse{}, nil
+}
+
 func (a *WebAPI) EnablePiped(ctx context.Context, req *webservice.EnablePipedRequest) (*webservice.EnablePipedResponse, error) {
 	if err := a.updatePiped(ctx, req.PipedId, a.pipedStore.EnablePiped); err != nil {
 		return nil, err

pkg/app/server/service/pipedservice/service.proto

@@ -153,6 +153,9 @@ service PipedService {
     // GetLatestAnalysisResult updates the most successful analysis result.
     rpc PutLatestAnalysisResult(PutLatestAnalysisResultRequest) returns (PutLatestAnalysisResultResponse) {}
 
+    // GetDesiredVersion returns the desired version of the given Piped.
+    rpc GetNeedRestart(GetNeedRestartRequest) returns (GetNeedRestartResponse) {}
+
     // GetDesiredVersion returns the desired version of the given Piped.
     rpc GetDesiredVersion(GetDesiredVersionRequest) returns (GetDesiredVersionResponse) {}
 
@@ -480,6 +483,13 @@ message PutLatestAnalysisResultRequest {
 message PutLatestAnalysisResultResponse {
 }
 
+message GetNeedRestartRequest {
+}
+
+message GetNeedRestartResponse {
+    bool need_restart = 1;
+}
+
 message GetDesiredVersionRequest {
 }