feat: add tdbg taskqueue get-user-data command

Adds AdminService.GetTaskQueueUserData proto + generated code, and wires
it up as a tdbg command. Accepts --task-queue (required),
--task-queue-type (default: workflow), and --partition-id (default: 0).
The backend handler implementation is in a separate PR.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: 2 people

client/admin/client_gen.go

@@ -182,6 +182,50 @@ func AdminDescribeTaskQueuePartition(c *cli.Context, clientFactory ClientFactory
 	return nil
 }
 
+// AdminGetTaskQueueUserData returns the per-type user data for a task queue partition
+func AdminGetTaskQueueUserData(c *cli.Context, clientFactory ClientFactory) error {
+	namespace, err := getRequiredOption(c, FlagNamespace)
+	if err != nil {
+		return err
+	}
+
+	tqName, err := getRequiredOption(c, FlagTaskQueue)
+	if err != nil {
+		return err
+	}
+
+	tlTypeInt, err := StringToEnum(c.String(FlagTaskQueueType), enumspb.TaskQueueType_value)
+	if err != nil {
+		return fmt.Errorf("invalid task queue type: %w", err)
+	}
+	tqType := enumspb.TaskQueueType(tlTypeInt)
+	if tqType == enumspb.TASK_QUEUE_TYPE_UNSPECIFIED {
+		return errors.New("invalid task queue type") // nolint
+	}
+
+	partitionID := 0
+	if c.IsSet(FlagPartitionID) {
+		partitionID = c.Int(FlagPartitionID)
+	}
+
+	client := clientFactory.AdminClient(c)
+	req := &adminservice.GetTaskQueueUserDataRequest{
+		Namespace:     namespace,
+		TaskQueue:     tqName,
+		TaskQueueType: tqType,
+		PartitionId:   int32(partitionID),
+	}
+
+	ctx, cancel := newContext(c)
+	defer cancel()
+	if response, e := client.GetTaskQueueUserData(ctx, req); e != nil {
+		return fmt.Errorf("unable to get Task Queue User Data: %w", e)
+	} else {
+		prettyPrintJSONObject(c, response)
+	}
+	return nil
+}
+
 // AdminForceUnloadTaskQueuePartition forcefully unloads a task queue partition
 func AdminForceUnloadTaskQueuePartition(c *cli.Context, clientFactory ClientFactory) error {
 	// extracting the namespace

tools/tdbg/task_queue_commands.go

@@ -19,6 +19,7 @@ type (
 		adminservice.AdminServiceClient
 		describeTaskQueuePartitionFn    func(request *adminservice.DescribeTaskQueuePartitionRequest) (*adminservice.DescribeTaskQueuePartitionResponse, error)
 		forceUnloadTaskQueuePartitionFn func(request *adminservice.ForceUnloadTaskQueuePartitionRequest) (*adminservice.ForceUnloadTaskQueuePartitionResponse, error)
+		getTaskQueueUserDataFn          func(request *adminservice.GetTaskQueueUserDataRequest) (*adminservice.GetTaskQueueUserDataResponse, error)
 	}
 )
 
@@ -82,6 +83,10 @@ func (t *testClient) ForceUnloadTaskQueuePartition(_ context.Context, request *a
 	return t.forceUnloadTaskQueuePartitionFn(request)
 }
 
+func (t *testClient) GetTaskQueueUserData(_ context.Context, request *adminservice.GetTaskQueueUserDataRequest, opts ...grpc.CallOption) (*adminservice.GetTaskQueueUserDataResponse, error) {
+	return t.getTaskQueueUserDataFn(request)
+}
+
 func (s *taskQueueCommandTestSuite) SetupTest() {
 	s.Assertions = require.New(s.T())
 	s.controller = gomock.NewController(s.T())
@@ -94,6 +99,9 @@ func (s *taskQueueCommandTestSuite) SetupTest() {
 		forceUnloadTaskQueuePartitionFn: func(request *adminservice.ForceUnloadTaskQueuePartitionRequest) (*adminservice.ForceUnloadTaskQueuePartitionResponse, error) {
 			return &adminservice.ForceUnloadTaskQueuePartitionResponse{}, nil
 		},
+		getTaskQueueUserDataFn: func(request *adminservice.GetTaskQueueUserDataRequest) (*adminservice.GetTaskQueueUserDataResponse, error) {
+			return &adminservice.GetTaskQueueUserDataResponse{}, nil
+		},
 	}
 	s.app = NewCliApp(func(params *Params) {
 		params.ClientFactory = client
@@ -163,3 +171,40 @@ func (s *taskQueueCommandTestSuite) TestForceUnloadTaskQueuePartition() {
 		}
 	}
 }
+
+// TestGetTaskQueueUserData tests that the cli accepts the various arguments for get-user-data.
+func (s *taskQueueCommandTestSuite) TestGetTaskQueueUserData() {
+	baseCommand := []string{"tdbg", "taskqueue", "get-user-data",
+		"--task-queue", "test"}
+
+	// Run shared test cases, skipping sticky-name (not a registered flag on this command).
+	for _, test := range testCases {
+		if len(test.inputFlags) > 0 && test.inputFlags[0] == "--sticky-name" {
+			continue
+		}
+		cliCommand := append(baseCommand, test.inputFlags...)
+		resp := s.app.Run(cliCommand)
+		if resp != nil {
+			s.ErrorContainsf(resp, test.err.Error(), "error present")
+		}
+	}
+
+	// Missing --task-queue is enforced by cli/v2 (Required: true) before the action runs.
+	s.Error(s.app.Run([]string{"tdbg", "taskqueue", "get-user-data"}))
+
+	// No --task-queue-type or --partition-id: both use their defaults and succeed.
+	s.NoError(s.app.Run([]string{"tdbg", "taskqueue", "get-user-data",
+		"--task-queue", "test"}))
+
+	// Matching client returns an error: CLI wraps and returns it.
+	errorClient := &testClient{
+		getTaskQueueUserDataFn: func(request *adminservice.GetTaskQueueUserDataRequest) (*adminservice.GetTaskQueueUserDataResponse, error) {
+			return nil, errors.New("matching unavailable")
+		},
+	}
+	errorApp := NewCliApp(func(params *Params) { params.ClientFactory = errorClient })
+	errorApp.ExitErrHandler = func(context *cli.Context, err error) {}
+	resp := errorApp.Run([]string{"tdbg", "taskqueue", "get-user-data",
+		"--task-queue", "test"})
+	s.ErrorContains(resp, "unable to get Task Queue User Data")
+}

tools/tdbg/task_queue_commands_test.go

@@ -682,6 +682,30 @@ func newAdminTaskQueueCommands(clientFactory ClientFactory) []*cli.Command {
 				return AdminForceUnloadTaskQueuePartition(c, clientFactory)
 			},
 		},
+		{
+			Name:  "get-user-data",
+			Usage: "Get per-type user data stored for a task queue",
+			Flags: []cli.Flag{
+				&cli.StringFlag{
+					Name:     FlagTaskQueue,
+					Usage:    "Task Queue name",
+					Required: true,
+				},
+				&cli.StringFlag{
+					Name:  FlagTaskQueueType,
+					Value: "TASK_QUEUE_TYPE_WORKFLOW",
+					Usage: "Task Queue type: TASK_QUEUE_TYPE_WORKFLOW, TASK_QUEUE_TYPE_ACTIVITY, TASK_QUEUE_TYPE_NEXUS",
+				},
+				&cli.Int64Flag{
+					Name:  FlagPartitionID,
+					Usage: "Partition ID to fetch user data from (default 0 = root partition)",
+					Value: 0,
+				},
+			},
+			Action: func(c *cli.Context) error {
+				return AdminGetTaskQueueUserData(c, clientFactory)
+			},
+		},
 	}
 }