apache · njnu-seafish · May 25, 2026 · May 25, 2026 · May 26, 2026 · May 27, 2026
@@ -105,6 +105,10 @@ public void testStartWorkflowInstance() {
             projectCode = (long) ((LinkedHashMap<String, Object>) ((List<LinkedHashMap>) queryAllProjectListResponse
                     .getBody().getData()).get(0)).get("code");
 
+            HttpResponse assignWorkerGroupsResponse =
+                    projectPage.assignWorkerGroups(loginUser, projectCode, new String[]{"default"});
+            Assertions.assertTrue(assignWorkerGroupsResponse.getBody().getSuccess());
+
             // upload test workflow definition json
             ClassLoader classLoader = getClass().getClassLoader();
             File file = new File(classLoader.getResource("workflow-json/test.json").getFile());

@@ -98,6 +98,11 @@ public void testCreateSchedule() {
 
         projectCode = (long) ((LinkedHashMap<String, Object>) ((List<LinkedHashMap>) queryAllProjectListResponse
                 .getBody().getData()).get(0)).get("code");
+
+        String[] workerGroups = {"default"};
+        HttpResponse assignWorkerGroupsResponse = projectPage.assignWorkerGroups(loginUser, projectCode, workerGroups);
+        Assertions.assertTrue(assignWorkerGroupsResponse.getBody().getSuccess());
+
         ClassLoader classLoader = getClass().getClassLoader();
         File file = new File(classLoader.getResource("workflow-json/test.json").getFile());
         workflowDefinitionPage.createWorkflowDefinition(loginUser, projectCode, file, workflowDefinitionName);

@@ -94,6 +94,11 @@ public void testCreateWorkflowDefinition() {
 
             projectCode = (long) ((LinkedHashMap<String, Object>) ((List<LinkedHashMap>) queryAllProjectListResponse
                     .getBody().getData()).get(0)).get("code");
+
+            HttpResponse assignWorkerGroupsResponse =
+                    projectPage.assignWorkerGroups(loginUser, projectCode, new String[]{"default"});
+            Assertions.assertTrue(assignWorkerGroupsResponse.getBody().getSuccess());
+
             ClassLoader classLoader = getClass().getClassLoader();
             File file = new File(classLoader.getResource("workflow-json/test.json").getFile());
             HttpResponse createWorkflowDefinitionResponse = workflowDefinitionPage

@@ -116,6 +116,10 @@ public void testQueryWorkflowInstancesByWorkflowInstanceId() {
             projectCode = (long) ((LinkedHashMap<String, Object>) ((List<LinkedHashMap>) queryAllProjectListResponse
                     .getBody().getData()).get(0)).get("code");
 
+            HttpResponse assignWorkerGroupsResponse =
+                    projectPage.assignWorkerGroups(loginUser, projectCode, new String[]{"default"});
+            assertTrue(assignWorkerGroupsResponse.getBody().getSuccess());
+
             // upload test workflow definition json
             ClassLoader classLoader = getClass().getClassLoader();
             File file = new File(classLoader.getResource("workflow-json/test.json").getFile());

@@ -122,6 +122,10 @@ public void testCreateUpstreamWorkflow() throws Exception {
         Assertions.assertNotEquals(0, projectCode, "project should be found by name");
         log.info("project code: {}", projectCode);
 
+        HttpResponse assignWorkerGroupsResponse =
+                projectPage.assignWorkerGroups(loginUser, projectCode, new String[]{"default"});
+        Assertions.assertTrue(assignWorkerGroupsResponse.getBody().getSuccess());
+
         // create upstream workflow definition (shell task: echo hello)
         ClassLoader classLoader = getClass().getClassLoader();
         File file = new File(classLoader.getResource("workflow-json/test.json").getFile());

@@ -106,6 +106,10 @@ public void testEmrServerlessSuccessWorkflowInstance() throws Exception {
         projectCode = (long) ((LinkedHashMap<String, Object>) ((List<LinkedHashMap>) queryAllProjectListResponse
                 .getBody().getData()).get(0)).get("code");
 
+        HttpResponse assignWorkerGroupsResponse =
+                projectPage.assignWorkerGroups(loginUser, projectCode, new String[]{"default"});
+        Assertions.assertTrue(assignWorkerGroupsResponse.getBody().getSuccess());
+
         // upload test workflow definition json
         ClassLoader classLoader = getClass().getClassLoader();
         File file = new File(classLoader

@@ -103,6 +103,10 @@ public void testGrpcFailedWorkflowInstance() {
             projectCode = (long) ((LinkedHashMap<String, Object>) ((List<LinkedHashMap>) queryAllProjectListResponse
                     .getBody().getData()).get(0)).get("code");
 
+            HttpResponse assignWorkerGroupsResponse =
+                    projectPage.assignWorkerGroups(loginUser, projectCode, new String[]{"default"});
+            Assertions.assertTrue(assignWorkerGroupsResponse.getBody().getSuccess());
+
             // upload test workflow definition json
             ClassLoader classLoader = getClass().getClassLoader();
             File file = new File(classLoader.getResource("workflow-json/task-grpc/grpcFailedWorkflow.json").getFile());
@@ -153,6 +157,10 @@ public void testGrpcSuccessWorkflowInstance() {
             projectCode = (long) ((LinkedHashMap<String, Object>) ((List<LinkedHashMap>) queryAllProjectListResponse
                     .getBody().getData()).get(0)).get("code");
 
+            HttpResponse assignWorkerGroupsResponse =
+                    projectPage.assignWorkerGroups(loginUser, projectCode, new String[]{"default"});
+            Assertions.assertTrue(assignWorkerGroupsResponse.getBody().getSuccess());
+
             // upload test workflow definition json
             ClassLoader classLoader = getClass().getClassLoader();
             File file = new File(classLoader.getResource("workflow-json/task-grpc/grpcSuccessWorkflow.json").getFile());

@@ -177,4 +177,17 @@ public HttpResponse deleteProject(User loginUser, Long code) {
         String url = String.format("/projects/%d", code);
         return requestClient.delete(url, headers, params);
     }
+
+    public HttpResponse assignWorkerGroups(User loginUser, long projectCode, String[] workerGroups) {
+        Map<String, Object> params = new HashMap<>();
+        params.put("loginUser", loginUser);
+        params.put("workerGroups", workerGroups);
+        Map<String, String> headers = new HashMap<>();
+        headers.put(Constants.SESSION_ID_KEY, sessionId);
+
+        RequestClient requestClient = new RequestClient();
+        String url = String.format("/projects/%d/worker-group", projectCode);
+        return requestClient.post(url, headers, params);
+    }
+
 }
@@ -21,20 +21,12 @@
 import org.apache.dolphinscheduler.api.test.entity.HttpResponse;
 import org.apache.dolphinscheduler.api.test.entity.HttpResponseBody;
 
-import org.apache.http.HttpEntity;
-import org.apache.http.client.methods.CloseableHttpResponse;
-import org.apache.http.client.methods.HttpPost;
-import org.apache.http.entity.ContentType;
-import org.apache.http.entity.mime.MultipartEntityBuilder;
-import org.apache.http.impl.client.CloseableHttpClient;
-import org.apache.http.impl.client.HttpClients;
-import org.apache.http.message.BasicHeader;
-
-import java.io.File;
-import java.nio.file.Files;
+import java.net.URLEncoder;
+import java.nio.charset.StandardCharsets;
 import java.util.HashMap;
 import java.util.Map;
-import java.util.Objects;
+import java.util.SortedMap;
+import java.util.TreeMap;
 
 import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
@@ -95,20 +87,62 @@ public HttpResponse get(String url, Map<String, String> headers, Map<String, Obj
     }
 
     public static String getParams(Map<String, Object> params) {
-        StringBuilder sb = new StringBuilder(Constants.QUESTION_MARK);
-        if (!params.isEmpty()) {
-            for (Map.Entry<String, Object> item : params.entrySet()) {
-                Object value = item.getValue();
-                if (Objects.nonNull(value)) {
-                    sb.append(Constants.AND_MARK);
-                    sb.append(item.getKey());
-                    sb.append(Constants.EQUAL_MARK);
-                    sb.append(value);
+        return getParams(params, true);
+    }
+
+    public static String getParams(Map<String, Object> params, boolean includeQuestionMark) {
+        if (params == null || params.isEmpty()) {
+            return "";
+        }
+
+        if (!(params instanceof SortedMap)) {
+            params = new TreeMap<>(params);
+        }
+
+        StringBuilder sb = new StringBuilder(params.size() * 16);
+        boolean isFirst = true;
+
+        for (Map.Entry<String, Object> entry : params.entrySet()) {
+            Object value = entry.getValue();
+            if (value == null) {
+                continue;
+            }
+
+            String key = entry.getKey();
+
+            if (value.getClass().isArray()) {
+                int length = java.lang.reflect.Array.getLength(value);
+                for (int i = 0; i < length; i++) {
+                    Object item = java.lang.reflect.Array.get(value, i);
+                    if (item != null) {
+                        appendParam(sb, isFirst, key, item.toString(), includeQuestionMark);
+                        isFirst = false;
+                    }
                 }
+            } else {
+                appendParam(sb, isFirst, key, value.toString(), includeQuestionMark);
+                isFirst = false;
+            }
+        }
+
+        return sb.toString();
+    }
+
+    private static void appendParam(StringBuilder sb, boolean isFirst, String key, String value,
+                                    boolean includeQuestionMark) {
+        if (isFirst) {
+            if (includeQuestionMark) {
+                sb.append(Constants.QUESTION_MARK);
             }
-            return sb.toString();
         } else {
-            return "";
+            sb.append(Constants.AND_MARK);
+        }
+        try {
+            sb.append(URLEncoder.encode(key, StandardCharsets.UTF_8.name()))
+                    .append(Constants.EQUAL_MARK)
+                    .append(URLEncoder.encode(value, StandardCharsets.UTF_8.name()));
+        } catch (Exception e) {
+            sb.append(key).append(Constants.EQUAL_MARK).append(value);
         }
     }
 
@@ -121,7 +155,8 @@ public HttpResponse post(String url, Map<String, String> headers, Map<String, Ob
         String requestUrl = String.format("%s%s", Constants.DOLPHINSCHEDULER_API_URL, url);
         headers.put("Content-Type", Constants.REQUEST_CONTENT_TYPE);
         Headers headersBuilder = Headers.of(headers);
-        RequestBody requestBody = FormBody.create(getParams(params), MediaType.parse(Constants.REQUEST_CONTENT_TYPE));
+        RequestBody requestBody =
+                FormBody.create(getParams(params, false), MediaType.parse(Constants.REQUEST_CONTENT_TYPE));
         log.info("POST request to {}, Headers: {}, Params: {}", requestUrl, headersBuilder, params);
         Request request = new Request.Builder()
                 .headers(headersBuilder)
@@ -159,7 +194,8 @@ public HttpResponse put(String url, Map<String, String> headers, Map<String, Obj
         String requestUrl = String.format("%s%s", Constants.DOLPHINSCHEDULER_API_URL, url);
         headers.put("Content-Type", Constants.REQUEST_CONTENT_TYPE);
         Headers headersBuilder = Headers.of(headers);
-        RequestBody requestBody = FormBody.create(getParams(params), MediaType.parse(Constants.REQUEST_CONTENT_TYPE));
+        RequestBody requestBody =
+                FormBody.create(getParams(params, false), MediaType.parse(Constants.REQUEST_CONTENT_TYPE));
         log.info("PUT request to {}, Headers: {}, Params: {}", requestUrl, headersBuilder, params);
         Request request = new Request.Builder()
                 .headers(headersBuilder)
@@ -188,33 +224,6 @@ public HttpResponse put(String url, Map<String, String> headers, Map<String, Obj
         return httpResponse;
     }
 
-    public CloseableHttpResponse postWithFile(String url, Map<String, String> headers, Map<String, Object> params,
-                                              File file) {
-        try {
-            Headers headersBuilder = Headers.of(headers);
-            MultipartEntityBuilder builder = MultipartEntityBuilder.create();
-            builder.addTextBody("json", getParams(params), ContentType.MULTIPART_FORM_DATA);
-            builder.addBinaryBody(
-                    "file",
-                    Files.newInputStream(file.toPath()),
-                    ContentType.APPLICATION_OCTET_STREAM,
-                    file.getName());
-            HttpEntity multipart = builder.build();
-            String requestUrl = String.format("%s%s", Constants.DOLPHINSCHEDULER_API_URL, url);
-            log.info("POST request to {}, Headers: {}, Params: {}", requestUrl, headersBuilder, params);
-            HttpPost httpPost = new HttpPost(requestUrl);
-            for (Map.Entry<String, String> header : headers.entrySet()) {
-                httpPost.setHeader(new BasicHeader(header.getKey(), header.getValue()));
-            }
-            httpPost.setEntity(multipart);
-            CloseableHttpClient client = HttpClients.createDefault();
-            return client.execute(httpPost);
-        } catch (Exception e) {
-            log.error("error", e);
-        }
-        return null;
-    }
-
     @SneakyThrows
     public HttpResponse delete(String url, Map<String, String> headers, Map<String, Object> params) {
         if (headers == null) {

@@ -0,0 +1,50 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.dolphinscheduler.api.dto;
+
+import lombok.Data;
+
+@Data
+public class ProjectPreferencesDTO {
+
+    private String taskPriority;
+
+    private String workerGroup;
+
+    private Long environmentCode;
+
+    private Integer failRetryTimes;
+
+    private Integer failRetryInterval;
+
+    private Integer cpuQuota;
+
+    private Integer memoryMax;
+
+    private Boolean timeoutFlag;
+
+    private String[] timeoutNotifyStrategy;
+
+    private Integer timeout;
+
+    private String warningType;
+
+    private String tenantCode;
+
+    private Integer warningGroupId;
+}
@@ -583,6 +583,9 @@ public enum Status {
     USED_WORKER_GROUP_EXISTS(1402004,
             "You can not reassign worker groups to the project, cause these worker groups {0} are already used.",
             "Worker组{0}被项目中任务或定时引用，无法重新分配"),
+    WORKER_GROUP_NOT_ASSIGNED_TO_PROJECT(1402005,
+            "Worker group [{0}] is not assigned to the project.",
+            "Worker组[{0}]未分配给项目"),
     CREATE_WORKFLOW_LINEAGE_ERROR(1403001, "create workflow lineage error", "创建工作流血缘错误"),
     UPDATE_WORKFLOW_LINEAGE_ERROR(1403002, "update workflow lineage error", "更新工作流血缘错误"),
     DELETE_WORKFLOW_LINEAGE_ERROR(1403003, "delete workflow lineage error", "删除工作流血缘错误"),

@@ -22,6 +22,7 @@
 import org.apache.dolphinscheduler.dao.entity.User;
 
 import java.util.List;
+import java.util.Set;
 
 public interface ProjectWorkerGroupRelationService {
 
@@ -43,4 +44,22 @@ public interface ProjectWorkerGroupRelationService {
      */
     List<ProjectWorkerGroup> queryAssignedWorkerGroupsByProject(User loginUser, Long projectCode);
 
+    /**
+     * check if worker group is assigned to project
+     *
+     * @param projectCode project code
+     * @param workerGroup worker group name
+     * @return true if worker group is assigned to project
+     */
+    boolean isWorkerGroupAssignedToProject(Long projectCode, String workerGroup);
+
+    /**
+     * get all assigned worker group names for a project
+     * This includes both directly assigned worker groups and worker groups used by tasks/schedules
+     *
+     * @param projectCode project code
+     * @return set of all assigned worker group names
+     */
+    Set<String> getAllAssignedWorkerGroupNames(Long projectCode);
+
 }