From 4e06e95039b362e9b8093472c24341997043a8c0 Mon Sep 17 00:00:00 2001
From: Saad Sahibjan <saadsahibjan@gmail.com>
Date: Thu, 17 Oct 2019 12:59:24 +0530
Subject: [PATCH] Add correctiveAction to policy bean class

---
 .../device/mgt/common/policy/mgt/Policy.java  | 34 +++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/components/device-mgt/org.wso2.carbon.device.mgt.common/src/main/java/org/wso2/carbon/device/mgt/common/policy/mgt/Policy.java b/components/device-mgt/org.wso2.carbon.device.mgt.common/src/main/java/org/wso2/carbon/device/mgt/common/policy/mgt/Policy.java
index 977fd60d38..452521286b 100644
--- a/components/device-mgt/org.wso2.carbon.device.mgt.common/src/main/java/org/wso2/carbon/device/mgt/common/policy/mgt/Policy.java
+++ b/components/device-mgt/org.wso2.carbon.device.mgt.common/src/main/java/org/wso2/carbon/device/mgt/common/policy/mgt/Policy.java
@@ -14,6 +14,23 @@
  * KIND, either express or implied.  See the License for the
  * specific language governing permissions and limitations
  * under the License.
+ *
+ *
+ * Copyright (c) 2019, Entgra (Pvt) Ltd. (http://entgra.io) All Rights Reserved.
+ *
+ * Entgra (Pvt) Ltd. 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.wso2.carbon.device.mgt.common.policy.mgt;
 
@@ -184,6 +201,13 @@ public class Policy implements Comparable<Policy>, Serializable {
             example = "GENERAL")
     private String policyType;
 
+    @ApiModelProperty(
+            name = "correctiveActions",
+            value = "List of corrective actions to be applied when the policy is violated",
+            example = "[{'actionType': 'POLICY', 'policyId': 1}]"
+    )
+    private List<CorrectiveAction> correctiveActions;
+
     @XmlElement
     public int getId() {
         return id;
@@ -355,6 +379,16 @@ public class Policy implements Comparable<Policy>, Serializable {
         this.policyType = policyType;
     }
 
+    @XmlElement
+    public List<CorrectiveAction> getCorrectiveActions() {
+        return correctiveActions;
+    }
+
+    public void setCorrectiveActions(
+            List<CorrectiveAction> correctiveActions) {
+        this.correctiveActions = correctiveActions;
+    }
+
     @Override
     public int compareTo(Policy o) {
         if (this.priorityId == o.priorityId)