Fix/service set update

cmd/deploy/service-set.go

@@ -2,8 +2,10 @@
 
 import (
 	"encoding/json"
+	"fmt"
 	"os"
 
+	"github.com/dream11/odin/internal/ui"
 	"github.com/dream11/odin/pkg/config"
 	serviceDto "github.com/dream11/odin/proto/gen/go/dream11/od/dto/v1"
 	serviceProto "github.com/dream11/odin/proto/gen/go/dream11/od/service/v1"
@@ -23,6 +25,10 @@
 		executeDeployServiceSet(cmd)
 	},
 }
+const (
+	Yes = "y"
+	No  = "n"
+)
 
 func init() {
 
@@ -61,8 +67,57 @@
 		deployServiceSetRequest.Name = serviceSetName
 	}
 
+	conflictingServicesRequest := &serviceProto.GetConflictingServicesRequest{
+		EnvName: env,
+		Name:    deployServiceSetRequest.Name,
+		Services: deployServiceSetRequest.Services,
+	}
+
+	services, errs := serviceClient.GetConflictingServices(&ctx, conflictingServicesRequest)
+	if errs != nil {
+		log.Fatal(fmt.Sprintf("Failed to list services with conflicting versions: %s", errs.Error()))
+		return
+	}
+	var serviceNames []string
+	for _, service := range services.Services {
+
+		allowedInputsSlice := []string{Yes, No}
+		allowedInputs := make(map[string]struct{}, len(allowedInputsSlice))
+		for _, input := range allowedInputsSlice {
+			allowedInputs[input] = struct{}{}
+		}
+		message := "Service already deployed with different version.Do you want to deploy service " + service.Name + " ? (y/n)"
+		inputHandler := ui.Input{}
+		val, err := inputHandler.AskWithConstraints(message, allowedInputs)
+
+		if err != nil {
+			log.Fatal(fmt.Sprintf("An error occurred while processing input: %s", err.Error()))
+		}
+
+		if val != Yes {
+			log.Info(fmt.Sprintf("Skipping service %s from deploy", service.Name))
+			serviceNames = append(serviceNames, service.Name)
+		}
+		// Remove services from deployServiceSetRequest
+		var updatedServices []*serviceProto.ServiceIdentifier
+		for _, svc := range deployServiceSetRequest.Services {
+			shouldRemove := false
+			for _, name := range serviceNames {
+				if svc.ServiceName == name {
+					shouldRemove = true
+					break
+				}
+			}
+			if !shouldRemove {
+				updatedServices = append(updatedServices, svc)
+			}
+		}
+		deployServiceSetRequest.Services = updatedServices
+
+	}
+
 	err := serviceClient.DeployServiceSet(&ctx, &deployServiceSetRequest)
 	if err != nil {
-		log.Fatal("Failed to deploy service ", err)
+		log.Fatal("Failed to deploy service set. ", err)
 	}
 }

internal/service/service.go

@@ -1,6 +1,7 @@
 package service
 
 import (
+	"bytes"
 	"context"
 	"errors"
 	"fmt"
@@ -11,6 +12,7 @@ import (
 	"github.com/dream11/odin/pkg/util"
 	serviceDto "github.com/dream11/odin/proto/gen/go/dream11/od/dto/v1"
 	serviceProto "github.com/dream11/odin/proto/gen/go/dream11/od/service/v1"
+	"github.com/olekukonko/tablewriter"
 	log "github.com/sirupsen/logrus"
 )
 
@@ -51,6 +53,7 @@ func (e *Service) DeployService(ctx *context.Context, request *serviceProto.Depl
 		}
 
 		if response != nil {
+
 			message = util.GenerateResponseMessage(response.GetServiceResponse())
 			logFailedComponentMessagesOnce(response.GetServiceResponse())
 			spinnerInstance.Prefix = fmt.Sprintf(" %s  ", message)
@@ -79,6 +82,7 @@ func logFailedComponentMessagesOnce(response *serviceProto.ServiceResponse) {
 func (e *Service) DeployServiceSet(ctx *context.Context, request *serviceProto.DeployServiceSetRequest) error {
 	conn, requestCtx, err := grpcClient(ctx)
 	if err != nil {
+		log.Errorf("TraceID: %s", (*requestCtx).Value(constant.TraceIDKey))
 		return err
 	}
 	client := serviceProto.NewServiceServiceClient(conn)
@@ -97,7 +101,7 @@ func (e *Service) DeployServiceSet(ctx *context.Context, request *serviceProto.D
 	var message string
 	for {
 		response, err := stream.Recv()
-		spinnerInstance.Stop()
+
 		if err != nil {
 			if errors.Is(err, context.Canceled) || err == io.EOF {
 				break
@@ -107,17 +111,34 @@ func (e *Service) DeployServiceSet(ctx *context.Context, request *serviceProto.D
 		}
 
 		if response != nil {
-			message = ""
+			spinnerInstance.Stop()
+			var buf bytes.Buffer
+			table := tablewriter.NewWriter(&buf)
+			table.SetHeader([]string{"Service Name", "Version", "Tags", "Action", "Status", "Error"})
 			for _, serviceResponse := range response.GetServices() {
-				logFailedComponentMessagesOnce(serviceResponse.GetServiceResponse())
-				message += util.GenerateResponseMessage(serviceResponse.GetServiceResponse())
+				var errorMessage string
+				if serviceResponse.ServiceResponse.ServiceStatus.ServiceStatus == "FAILED" {
+					traceID := (*requestCtx).Value(constant.TraceIDKey)
+					errorMessage += fmt.Sprintf("[%s] TraceID: %s \n", serviceResponse.ServiceResponse.ServiceStatus.Error, traceID)
+				}
+				row := []string{
+					serviceResponse.ServiceIdentifier.ServiceName,
+					serviceResponse.ServiceIdentifier.ServiceVersion,
+					serviceResponse.ServiceIdentifier.Tags,
+					serviceResponse.ServiceResponse.ServiceStatus.ServiceAction,
+					serviceResponse.ServiceResponse.ServiceStatus.ServiceStatus,
+					errorMessage,
+				}
+				table.Append(row)
 			}
+
+			table.Render()
+			message = buf.String()
 			spinnerInstance.Prefix = fmt.Sprintf(" %s  ", message)
 			spinnerInstance.Start()
 		}
 	}
-
-	log.Info(message)
+	fmt.Println(message)
 	return err
 }
 
@@ -318,8 +339,9 @@ func (e *Service) ConvertToDeployServiceSetRequest(serviceSet *serviceDto.Servic
 	var services []*serviceProto.ServiceIdentifier
 	for _, service := range serviceSet.Services {
 		services = append(services, &serviceProto.ServiceIdentifier{
-			ServiceName:    service.ServiceName,
-			ServiceVersion: service.ServiceVersion,
+			ServiceName:    service.Name,
+			ServiceVersion: service.Version,
+			Tags:           service.Tags,
 		})
 	}
 
@@ -345,3 +367,17 @@ func (e *Service) DescribeService(ctx *context.Context, request *serviceProto.De
 
 	return response, nil
 }
+
+// GetConflictingServices deploys service
+func (e *Service) GetConflictingServices(ctx *context.Context, request *serviceProto.GetConflictingServicesRequest) (*serviceProto.GetConflictingServicesResponse, error) {
+	conn, requestCtx, err := grpcClient(ctx)
+	if err != nil {
+		return &serviceProto.GetConflictingServicesResponse{}, err
+	}
+	client := serviceProto.NewServiceServiceClient(conn)
+	response, err := client.GetConflictingServices(*requestCtx, request)
+	if err != nil {
+		log.Errorf("TraceID: %s", (*requestCtx).Value(constant.TraceIDKey))
+	}
+	return response, err
+}

pkg/util/util.go

@@ -11,7 +11,9 @@ import (
 
 	v1 "github.com/dream11/odin/proto/gen/go/dream11/od/service/v1"
 	"github.com/google/uuid"
+	"github.com/olekukonko/tablewriter"
 	"gopkg.in/yaml.v2"
+
 )
 
 // SplitProviderAccount splits string into list of cloud provider accounts
@@ -37,6 +39,27 @@ func GenerateResponseMessage(response *v1.ServiceResponse) string {
 	return message
 }
 
+// GenerateServiceSetResponseMessage generate response message from ServiceSetResponse
+func GenerateServiceSetResponseMessage(response *v1.DeployServiceSetServiceResponse) string {
+
+	message := fmt.Sprintf("\n Service %s %s %s %s", response.ServiceIdentifier.ServiceName, response.ServiceIdentifier.ServiceVersion, response.ServiceResponse.ServiceStatus.ServiceAction, response.ServiceResponse.ServiceStatus)
+	var tableData [][]string
+	row := []string{
+		response.ServiceIdentifier.ServiceName,
+		response.ServiceIdentifier.ServiceVersion,
+		response.ServiceResponse.ServiceStatus.ServiceAction,
+		response.ServiceResponse.ServiceStatus.ServiceStatus,
+	}
+	tableData = append(tableData, row)
+
+	table := tablewriter.NewWriter(os.Stdout)
+	table.SetHeader([]string{"Service Name", "Version", "Action", "Status", "Error"})
+	table.AppendBulk(tableData)
+	table.Render()
+	return message
+
+}
+
 // FormatToHumanReadableDuration takes a date-time string representing the last deployment time, and returns a human-readable string representing the duration since the last deployment
 func FormatToHumanReadableDuration(inputDateTime string) string {
 	// Check if the input is a Unix timestamp prefixed by "seconds:"

proto/dream11/od/dto/v1/service.proto

@@ -23,8 +23,9 @@ message ServiceDefinition {
 }
 
 message ServiceIdentifier {
-  string service_name = 1;
-  string service_version = 2;
+  string name = 1;
+  string version = 2;
+  string tags = 3;
 }
 
 message ServiceSet {

proto/dream11/od/service/v1/service.proto

@@ -16,6 +16,7 @@ service ServiceService {
   rpc ListService(ListServiceRequest) returns (ListServiceResponse) {}
   rpc DescribeService(DescribeServiceRequest) returns (DescribeServiceResponse) {}
   rpc OperateComponentDiff(OperateComponentDiffRequest) returns (OperateComponentDiffResponse) {}
+  rpc GetConflictingServices(GetConflictingServicesRequest) returns (GetConflictingServicesResponse) {}
 }
 
 message DeployServiceRequest {
@@ -143,3 +144,12 @@ message OperateComponentDiffResponse {
   google.protobuf.Struct old_values = 1;
   google.protobuf.Struct new_values = 2;
 }
+message GetConflictingServicesRequest {
+  string env_name = 1;
+  string name = 2;
+  repeated ServiceIdentifier services = 3;
+}
+
+message GetConflictingServicesResponse {
+  repeated dto.v1.ServiceMetadata services = 1;
+}