Fetch logs via traceID

cmd/logs/logs.go

@@ -0,0 +1,59 @@
+package logs
+
+import (
+	"context"
+	"fmt"
+	"github.com/dream11/odin/cmd"
+	"github.com/dream11/odin/internal/service"
+	"github.com/dream11/odin/pkg/constant"
+	logs "github.com/dream11/odin/proto/gen/go/dream11/od/logs/v1"
+	"github.com/spf13/cobra"
+)
+
+var traceID string
+var logsClient = service.Logs{}
+
+var logsCmd = &cobra.Command{
+	Use:   "logs",
+	Short: "Fetch logs",
+	Long:  "Fetch logs",
+	Run: func(cmd *cobra.Command, args []string) {
+		execute(cmd)
+	},
+}
+
+func init() {
+	logsCmd.Flags().StringVar(&traceID, "traceid", "", "Trace Id to fetch logs")
+	cmd.RootCmd.AddCommand(logsCmd)
+}
+
+func execute(cmd *cobra.Command) {
+	ctx := cmd.Context()
+
+	contextWithTrace := context.WithValue(ctx, constant.TraceIDKey, traceID)
+
+	streamCtx, cancelFunction := context.WithCancel(context.Background())
+	defer cancelFunction()
+
+	var searchAfterParams []int64
+	var err error
+	fmt.Printf("Fetching logs for traceID: %s\n", traceID)
+	for {
+		select {
+		case <-streamCtx.Done():
+			return
+		default:
+			// Get logs with retry on error
+			follow := true
+			searchAfterParams, err = logsClient.GetLogs(&contextWithTrace, &logs.GetLogsRequest{
+				TraceId:           traceID,
+				Follow:            &follow,
+				SearchAfterParams: searchAfterParams,
+			})
+			if err != nil {
+				continue
+			}
+		}
+	}
+
+}

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"