diff options
Diffstat (limited to 'weed/worker/client.go')
| -rw-r--r-- | weed/worker/client.go | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/weed/worker/client.go b/weed/worker/client.go index ef7e431c0..b9042f18c 100644 --- a/weed/worker/client.go +++ b/weed/worker/client.go @@ -651,7 +651,7 @@ func (c *GrpcAdminClient) RequestTask(workerID string, capabilities []types.Task Type: types.TaskType(taskAssign.TaskType), Status: types.TaskStatusAssigned, VolumeID: taskAssign.Params.VolumeId, - Server: taskAssign.Params.Server, + Server: getServerFromParams(taskAssign.Params), Collection: taskAssign.Params.Collection, Priority: types.TaskPriority(taskAssign.Priority), CreatedAt: time.Unix(taskAssign.CreatedTime, 0), @@ -934,3 +934,11 @@ func (m *MockAdminClient) AddMockTask(task *types.TaskInput) { func CreateAdminClient(adminServer string, workerID string, dialOption grpc.DialOption) (AdminClient, error) { return NewGrpcAdminClient(adminServer, workerID, dialOption), nil } + +// getServerFromParams extracts server address from unified sources +func getServerFromParams(params *worker_pb.TaskParams) string { + if len(params.Sources) > 0 { + return params.Sources[0].Node + } + return "" +} |
