From b967df69e570817a476aaaa17900e233cbcebc14 Mon Sep 17 00:00:00 2001
From: msm1992 <madusha.sachini@yahoo.com>
Date: Mon, 2 Sep 2024 12:33:35 +0530
Subject: [PATCH] Fix API policy dropping issue after migration

---
 .../java/org/wso2/carbon/apimgt/impl/APIProviderImpl.java  | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/components/apimgt/org.wso2.carbon.apimgt.impl/src/main/java/org/wso2/carbon/apimgt/impl/APIProviderImpl.java b/components/apimgt/org.wso2.carbon.apimgt.impl/src/main/java/org/wso2/carbon/apimgt/impl/APIProviderImpl.java
index 5c4ecc9dd0cc..64ff8a2fb66c 100644
--- a/components/apimgt/org.wso2.carbon.apimgt.impl/src/main/java/org/wso2/carbon/apimgt/impl/APIProviderImpl.java
+++ b/components/apimgt/org.wso2.carbon.apimgt.impl/src/main/java/org/wso2/carbon/apimgt/impl/APIProviderImpl.java
@@ -6781,6 +6781,13 @@ public Environment getEnvironment(String organization, String uuid) throws APIMa
     @Override
     public void setOperationPoliciesToURITemplates(String apiId, Set<URITemplate> uriTemplates)
             throws APIManagementException {
+        //In case the mediation sequences are not migrated yet with an API update, force an API update to  make sure
+        // the existing API sequences are migrated to API Policies
+        API api = getAPIbyUUID(apiId, organization);
+        if (APIUtil.isSequenceDefined(api.getInSequence()) || APIUtil.isSequenceDefined(api.getOutSequence())
+                || APIUtil.isSequenceDefined(api.getFaultSequence())) {
+            migrateMediationPoliciesOfAPI(api, tenantDomain, true);
+        }
 
         Set<URITemplate> uriTemplatesWithPolicies = apiMgtDAO.getURITemplatesWithOperationPolicies(apiId);