feat: add logs command

cmd/logs/logs.go

@@ -0,0 +1,54 @@
+package logs
+
+import (
+	"github.com/dream11/odin/cmd"
+	"github.com/dream11/odin/internal/service"
+	logs "github.com/dream11/odin/proto/gen/go/dream11/od/logs/v1"
+	logger "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+)
+
+var traceID string
+var follow bool
+var serviceName string
+var componentName string
+var logsClient = service.Logs{}
+
+var logsCmd = &cobra.Command{
+	Use:   "logs",
+	Short: "Get Logs",
+	Long:  `Get logs for a trace id`,
+	Run: func(cmd *cobra.Command, args []string) {
+		execute(cmd)
+	},
+}
+
+func init() {
+	logsCmd.Flags().StringVarP(&traceID, "trace-id", "t", "", "trace id")
+	logsCmd.Flags().BoolVarP(&follow, "follow", "f", false, "follow logs")
+	logsCmd.Flags().StringVar(&serviceName, "service", "", "service name")
+	logsCmd.Flags().StringVar(&serviceName, "component", "", "component name")
+	cmd.RootCmd.AddCommand(logsCmd)
+}
+
+func execute(cmd *cobra.Command) {
+	ctx := cmd.Context()
+	// Check traceid should be present
+	if traceID == "" {
+		logger.Error("TraceID is required")
+		err := cmd.Help()
+		if err != nil {
+			return
+		}
+	} else {
+		err := logsClient.GetLogs(&ctx, &logs.GetLogsRequest{
+			TraceId:       traceID,
+			Follow:        &follow,
+			ServiceName:   &serviceName,
+			ComponentName: &componentName,
+		})
+		if err != nil {
+			logger.Fatal("Failed to get logs ", err)
+		}
+	}
+}

internal/service/logs.go

@@ -0,0 +1,36 @@
+package service
+
+import (
+	"context"
+	"fmt"
+
+	logs "github.com/dream11/odin/proto/gen/go/dream11/od/logs/v1"
+)
+
+// Logs performs operation on logs like get logs
+type Logs struct{}
+
+// GetLogs Get logs
+func (l *Logs) GetLogs(ctx *context.Context, request *logs.GetLogsRequest) error {
+	// Get logs
+	conn, requestCtx, err := grpcClient(ctx)
+	if err != nil {
+		return err
+	}
+	client := logs.NewLogsServiceClient(conn)
+	stream, err := client.GetLogs(*requestCtx, request)
+	if err != nil {
+		return err
+	}
+	for {
+		response, err := stream.Recv()
+		if err != nil {
+			break
+		}
+		// Print logs
+		for _, logMessage := range response.Logs {
+			fmt.Print(logMessage.Message)
+		}
+	}
+	return nil
+}

main.go

@@ -11,6 +11,7 @@ import (
 	_ "github.com/dream11/odin/cmd/deploy"
 	_ "github.com/dream11/odin/cmd/describe"
 	_ "github.com/dream11/odin/cmd/list"
+	_ "github.com/dream11/odin/cmd/logs"
 	_ "github.com/dream11/odin/cmd/operate"
 	_ "github.com/dream11/odin/cmd/release"
 	_ "github.com/dream11/odin/cmd/set"

proto/dream11/od/logs/v1/logs.proto

@@ -0,0 +1,25 @@
+syntax = "proto3";
+package dream11.od.logs.v1;
+
+option go_package = "github.com/dream11/odin/proto/gen/go/dream11/od/logs/v1";
+
+service LogsService {
+  rpc GetLogs(GetLogsRequest) returns (stream GetLogsResponse) {}
+}
+
+message GetLogsRequest {
+  string trace_id = 1;
+  optional bool follow = 2;
+  optional string service_name = 3;
+  optional string component_name = 4;
+}
+
+message GetLogsResponse {
+  repeated Log logs = 1;
+}
+
+message Log {
+  string message = 1;
+  string level = 2;
+  string timestamp = 3;
+}