This is an automated email from the ASF dual-hosted git repository. lahirujayathilake pushed a commit to branch master in repository https://gitbox.apache.org/repos/asf/airavata-custos.git
commit cb7f988a63260aac293efbe50d72f2e8c2774c32 Author: lahiruj <[email protected]> AuthorDate: Wed Oct 8 16:24:54 2025 -0400 handle AMIE data project inactivation events --- .../handler/InformTransactionCompleteHandler.java | 52 ++++++++++++ .../handler/RequestProjectInactivateHandler.java | 92 ++++++++++++++++++++++ 2 files changed, 144 insertions(+) diff --git a/amie-decoder/src/main/java/org/apache/custos/amie/handler/InformTransactionCompleteHandler.java b/amie-decoder/src/main/java/org/apache/custos/amie/handler/InformTransactionCompleteHandler.java new file mode 100644 index 000000000..5fbb37cb8 --- /dev/null +++ b/amie-decoder/src/main/java/org/apache/custos/amie/handler/InformTransactionCompleteHandler.java @@ -0,0 +1,52 @@ +/* + * 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.custos.amie.handler; + +import com.fasterxml.jackson.databind.JsonNode; +import org.apache.custos.amie.model.PacketEntity; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.stereotype.Component; + +/** + * Handles the 'inform_transaction_complete' packet. + * <p> + * This is a terminal packet that declares the end of a transaction. This requires no reply. + */ +@Component +public class InformTransactionCompleteHandler implements PacketHandler { + + private static final Logger LOGGER = LoggerFactory.getLogger(InformTransactionCompleteHandler.class); + + @Override + public String supportsType() { + return "inform_transaction_complete"; + } + + @Override + public void handle(JsonNode packetJson, PacketEntity packetEntity) { + // This packet is purely informational and completes the transaction + JsonNode body = packetJson.path("body"); + String statusCode = body.path("StatusCode").asText("Unknown"); + String message = body.path("Message").asText(""); + + LOGGER.info("Received 'inform_transaction_complete' for packet amie_id [{}]. Status: [{}], Message: [{}]. Transaction is now closed.", + packetEntity.getAmieId(), statusCode, message); + } +} diff --git a/amie-decoder/src/main/java/org/apache/custos/amie/handler/RequestProjectInactivateHandler.java b/amie-decoder/src/main/java/org/apache/custos/amie/handler/RequestProjectInactivateHandler.java new file mode 100644 index 000000000..b49255210 --- /dev/null +++ b/amie-decoder/src/main/java/org/apache/custos/amie/handler/RequestProjectInactivateHandler.java @@ -0,0 +1,92 @@ +/* + * 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.custos.amie.handler; + +import com.fasterxml.jackson.databind.JsonNode; +import org.apache.custos.amie.client.AmieClient; +import org.apache.custos.amie.model.PacketEntity; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.stereotype.Component; +import org.springframework.util.Assert; + +import java.util.ArrayList; +import java.util.HashMap; +import java.util.List; +import java.util.Map; + +/** + * Handles the 'request_project_inactivate' AMIE packet + */ +@Component +public class RequestProjectInactivateHandler implements PacketHandler { + + private static final Logger LOGGER = LoggerFactory.getLogger(RequestProjectInactivateHandler.class); + + private final AmieClient amieClient; + + public RequestProjectInactivateHandler(AmieClient amieClient) { + this.amieClient = amieClient; + } + + @Override + public String supportsType() { + return "request_project_inactivate"; + } + + @Override + public void handle(JsonNode packetJson, PacketEntity packetEntity) { + LOGGER.info("Starting 'request_project_inactivate' handler for packet amie_id [{}].", packetEntity.getAmieId()); + + JsonNode body = packetJson.path("body"); + String projectId = body.path("ProjectID").asText(); + String personId = body.path("PersonID").asText(); + + Assert.hasText(projectId, "'ProjectID' must not be empty."); + Assert.hasText(personId, "'PersonID' must not be empty."); + LOGGER.info("Packet validated. ProjectID to inactivate: [{}].", projectId); + + // TODO - perform the business logic + // - find ALL user accounts associated with this project and inactivate each of them --> slurm inactivate + LOGGER.info("Simulating business logic: Marking project [{}] and all its associated user accounts as inactive.", projectId); + + + // Construct and send the 'notify_project_inactivate' reply + Map<String, Object> replyBody = new HashMap<>(); + Map<String, Object> bodyContent = new HashMap<>(); + + bodyContent.put("ProjectID", projectId); + bodyContent.put("PersonID", personId); + + List<String> resourceList = new ArrayList<>(); + JsonNode rlNode = body.path("ResourceList"); + if (rlNode.isArray()) { + rlNode.forEach(node -> resourceList.add(node.asText())); + } + bodyContent.put("ResourceList", resourceList); + + replyBody.put("type", "notify_project_inactivate"); + replyBody.put("body", bodyContent); + + amieClient.replyToPacket(packetEntity.getAmieId(), replyBody); + + LOGGER.info("Successfully completed 'request_project_inactivate' handler and sent reply for packet amie_id [{}].", packetEntity.getAmieId()); + } +} +
