feat(manager): support PerformClassAction

yunionio · Oct 26, 2023 · 8a1f956 · 8a1f956
1 parent d7f2cda
commit 8a1f956
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/pom.xml b/pom.xml
@@ -4,7 +4,7 @@
 
   <groupId>com.yunionyun.mcp</groupId>
   <artifactId>mcclient</artifactId>
-  <version>3.2.4</version>
+  <version>3.2.5</version>
   <packaging>jar</packaging>
 
   <name>${project.groupId}:${project.artifactId}</name>

diff --git a/src/main/java/com/yunionyun/mcp/mcclient/managers/ResourceManager.java b/src/main/java/com/yunionyun/mcp/mcclient/managers/ResourceManager.java
@@ -56,15 +56,23 @@ protected StringBuilder getContextPath(ManagerContext[] ctx) {
 	}
 
 	private JSONObject genBody(JSONObject params) {
+		return this.genBody(params, this.keyword);
+	}
+
+	private JSONObject genBody(JSONObject params, String key) {
 		if (params != null) {
 			JSONObject body = new JSONObject();
-			body.put(this.keyword, params);
+			body.put(key, params);
 			return body;
 		} else {
 			return null;
 		}
 	}
 
+	private JSONObject genPluralBody(JSONObject params) {
+		return this.genBody(params, this.keywordPlural);
+	}
+
 	public JSONObject GetByName(Session s, String name, JSONObject query, ManagerContext[] ctx)
 			throws McClientJavaBizException, IOException, JSONClientException {
 		JSONObject query2 = new JSONObject();
@@ -255,6 +263,18 @@ public JSONObject PerformAction(Session s, String id, String action, JSONObject
 		return this.PerformAction(s, id, action, params, new ManagerContext[]{});
 	}
 
+	public JSONObject PerformClassAction(Session s, String action, JSONObject params, ManagerContext[] ctx) throws McClientJavaBizException, IOException, JSONClientException {
+		StringBuilder url = this.getContextPath(ctx);
+		url.append(this.urlKey());
+		url.append("/");
+		url.append(action);
+		return this._post(s, url.toString(), genPluralBody(params), this.keyword);
+	}
+
+	public JSONObject PerformClassAction(Session s, String action, JSONObject params) throws McClientJavaBizException, IOException, JSONClientException {
+		return this.PerformClassAction(s, action, params, new ManagerContext[]{});
+	}
+
 	public JSONObject Update(Session s, String id, JSONObject params, ManagerContext[] ctx)
 			throws McClientJavaBizException, IOException, JSONClientException {
 		StringBuilder url = this.getContextPath(ctx);