IKC-410 Policy create, view, edit

kouncil-backend/src/main/java/com/consdata/kouncil/clusters/converter/ClusterConfigConverter.java

@@ -43,6 +43,7 @@ private static void setClusterConfigBrokers(ClusterConfig clusterConfig, Cluster
         clusterDto.getBrokers().forEach(brokerDto -> {
             BrokerConfig brokerConfig = new BrokerConfig();
             String[] hostPort = brokerDto.getBootstrapServer().split(":");
+            //todo walidacja na FE na poprawność
             brokerConfig.setHost(hostPort[0]);
             brokerConfig.setPort(Integer.parseInt(hostPort[1]));
             brokerConfig.setJmxUser(brokerDto.getJmxUser());

kouncil-backend/src/main/java/com/consdata/kouncil/datamasking/PolicyController.java

@@ -1,10 +1,15 @@
 package com.consdata.kouncil.datamasking;
 
+import com.consdata.kouncil.datamasking.dto.PolicyDto;
 import com.consdata.kouncil.model.admin.SystemFunctionNameConstants;
 import javax.annotation.security.RolesAllowed;
 import lombok.AllArgsConstructor;
 import org.springframework.web.bind.annotation.DeleteMapping;
+import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.PutMapping;
+import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
@@ -16,6 +21,24 @@ public class PolicyController {
 
     private final PolicyService policyService;
 
+    @RolesAllowed({SystemFunctionNameConstants.POLICY_DETAILS, SystemFunctionNameConstants.POLICY_UPDATE})
+    @GetMapping(path = "/{policyId}")
+    public PolicyDto getPolicyById(@PathVariable("policyId") Long id) {
+        return policyService.getPolicyById(id);
+    }
+
+    @RolesAllowed(SystemFunctionNameConstants.POLICY_CREATE)
+    @PostMapping()
+    public void addNewPolicy(@RequestBody PolicyDto policyDto) {
+        policyService.savePolicy(policyDto);
+    }
+
+    @RolesAllowed(SystemFunctionNameConstants.POLICY_UPDATE)
+    @PutMapping()
+    public void updatePolicy(@RequestBody PolicyDto policyDto) {
+        policyService.savePolicy(policyDto);
+    }
+
     @RolesAllowed(SystemFunctionNameConstants.POLICY_DELETE)
     @DeleteMapping(path = "/{id}")
     public void deletePolicy(@PathVariable("id") Long id) {

kouncil-backend/src/main/java/com/consdata/kouncil/datamasking/PolicyService.java

@@ -1,5 +1,8 @@
 package com.consdata.kouncil.datamasking;
 
+import com.consdata.kouncil.datamasking.converter.PolicyConverter;
+import com.consdata.kouncil.datamasking.converter.PolicyDtoConverter;
+import com.consdata.kouncil.datamasking.dto.PolicyDto;
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Service;
 
@@ -9,7 +12,16 @@ public class PolicyService {
 
     private final PolicyRepository policyRepository;
 
+    public void savePolicy(PolicyDto policyDto) {
+        policyRepository.save(PolicyConverter.convert(policyDto));
+    }
+
     public void deletePolicy(Long id) {
         policyRepository.deleteById(id);
     }
+
+    public PolicyDto getPolicyById(Long id) {
+        return policyRepository.findById(id).map(PolicyDtoConverter::convertToDto)
+                .orElseThrow(() -> new IllegalArgumentException(String.format("Policy with id=%s not found", id)));
+    }
 }

kouncil-backend/src/main/java/com/consdata/kouncil/datamasking/converter/PolicyConverter.java

@@ -0,0 +1,36 @@
+package com.consdata.kouncil.datamasking.converter;
+
+import com.consdata.kouncil.datamasking.dto.PolicyDto;
+import com.consdata.kouncil.model.datamasking.Policy;
+import com.consdata.kouncil.model.datamasking.PolicyField;
+import com.consdata.kouncil.model.datamasking.PolicyResource;
+import java.util.HashSet;
+import lombok.AccessLevel;
+import lombok.NoArgsConstructor;
+import org.springframework.beans.BeanUtils;
+
+@NoArgsConstructor(access = AccessLevel.PRIVATE)
+public final class PolicyConverter {
+
+    public static Policy convert(PolicyDto policyDto) {
+        Policy policy = new Policy();
+        BeanUtils.copyProperties(policyDto, policy);
+
+        policy.setFields(new HashSet<>());
+
+        policyDto.getFields().forEach(field -> {
+            PolicyField policyField = new PolicyField();
+            BeanUtils.copyProperties(field, policyField);
+            policy.getFields().add(policyField);
+        });
+
+        policy.setResources(new HashSet<>());
+        policyDto.getResources().forEach(resource -> {
+            PolicyResource policyResource = new PolicyResource();
+            BeanUtils.copyProperties(resource, policyResource);
+            policy.getResources().add(policyResource);
+        });
+
+        return policy;
+    }
+}

kouncil-backend/src/main/java/com/consdata/kouncil/datamasking/converter/PolicyDtoConverter.java

@@ -1,7 +1,10 @@
 package com.consdata.kouncil.datamasking.converter;
 
 import com.consdata.kouncil.datamasking.dto.PolicyDto;
+import com.consdata.kouncil.datamasking.dto.PolicyFieldDto;
+import com.consdata.kouncil.datamasking.dto.PolicyResourceDto;
 import com.consdata.kouncil.model.datamasking.Policy;
+import java.util.HashSet;
 import lombok.AccessLevel;
 import lombok.NoArgsConstructor;
 import org.springframework.beans.BeanUtils;
@@ -12,6 +15,22 @@ public final class PolicyDtoConverter {
     public static PolicyDto convertToDto(Policy policy) {
         PolicyDto policyDto = new PolicyDto();
         BeanUtils.copyProperties(policy, policyDto);
+
+        policyDto.setFields(new HashSet<>());
+        policy.getFields().forEach(field -> {
+            PolicyFieldDto policyFieldDto = new PolicyFieldDto();
+            BeanUtils.copyProperties(field, policyFieldDto);
+            policyDto.getFields().add(policyFieldDto);
+        });
+
+
+        policyDto.setResources(new HashSet<>());
+        policy.getResources().forEach(resource -> {
+            PolicyResourceDto policyResource = new PolicyResourceDto();
+            BeanUtils.copyProperties(resource, policyResource);
+            policyDto.getResources().add(policyResource);
+        });
+
         return policyDto;
     }
 }

kouncil-backend/src/main/java/com/consdata/kouncil/datamasking/dto/PolicyDto.java

@@ -9,7 +9,8 @@ public class PolicyDto {
 
     private Long id;
     private String name;
-    private MaskingType type;
-    private Set<String> fields;
+    private MaskingType maskingType;
+    private Boolean applyToAllResources;
+    private Set<PolicyFieldDto> fields;
     private Set<PolicyResourceDto> resources;
 }

kouncil-backend/src/main/java/com/consdata/kouncil/datamasking/dto/PolicyFieldDto.java

@@ -0,0 +1,12 @@
+package com.consdata.kouncil.datamasking.dto;
+
+import com.consdata.kouncil.model.datamasking.FieldFindRule;
+import lombok.Data;
+
+@Data
+public class PolicyFieldDto {
+
+    private Long id;
+    private FieldFindRule findRule;
+    private String field;
+}

kouncil-backend/src/main/java/com/consdata/kouncil/datamasking/dto/PolicyResourceDto.java

@@ -1,12 +1,11 @@
 package com.consdata.kouncil.datamasking.dto;
 
-import com.consdata.kouncil.clusters.dto.ClusterDto;
 import lombok.Data;
 
 @Data
 public class PolicyResourceDto {
 
     private Long id;
-    private ClusterDto cluster;
+    private Long cluster;
     private String topic;
 }

kouncil-backend/src/main/java/com/consdata/kouncil/model/datamasking/FieldFindRule.java

@@ -0,0 +1,7 @@
+package com.consdata.kouncil.model.datamasking;
+
+public enum FieldFindRule {
+
+    ANY_LEVEL,
+    EXACT_PATH
+}

kouncil-backend/src/main/java/com/consdata/kouncil/model/datamasking/Policy.java

@@ -2,9 +2,7 @@
 
 import java.util.Set;
 import javax.persistence.CascadeType;
-import javax.persistence.CollectionTable;
 import javax.persistence.Column;
-import javax.persistence.ElementCollection;
 import javax.persistence.Entity;
 import javax.persistence.EnumType;
 import javax.persistence.Enumerated;
@@ -36,12 +34,11 @@ public class Policy {
 
     @Column(name = "MASKING_TYPE")
     @Enumerated(EnumType.STRING)
-    private MaskingType type;
+    private MaskingType maskingType;
 
-    @ElementCollection
-    @CollectionTable(name = "POLICY_FIELDS", joinColumns = @JoinColumn(name = "POLICY_ID"))
-    @Column(name = "FIELD")
-    private Set<String> fields;
+    @OneToMany(cascade = {CascadeType.PERSIST, CascadeType.MERGE}, fetch = FetchType.EAGER, orphanRemoval = true)
+    @JoinColumn(name = "POLICY_ID")
+    private Set<PolicyField> fields;
 
     @Column(name = "APPLY_TO_ALL_RESOURCES")
     private Boolean applyToAllResources;

kouncil-backend/src/main/java/com/consdata/kouncil/model/datamasking/PolicyField.java

@@ -0,0 +1,40 @@
+package com.consdata.kouncil.model.datamasking;
+
+import javax.persistence.CascadeType;
+import javax.persistence.Column;
+import javax.persistence.Entity;
+import javax.persistence.EnumType;
+import javax.persistence.Enumerated;
+import javax.persistence.GeneratedValue;
+import javax.persistence.GenerationType;
+import javax.persistence.Id;
+import javax.persistence.JoinColumn;
+import javax.persistence.ManyToOne;
+import javax.persistence.SequenceGenerator;
+import javax.persistence.Table;
+import lombok.Getter;
+import lombok.Setter;
+
+@Entity
+@Table(name = "POLICY_FIELD")
+@Getter
+@Setter
+public class PolicyField {
+
+    @Id
+    @Column(name = "ID")
+    @GeneratedValue(strategy = GenerationType.SEQUENCE, generator = "SEQ_POLICY_FIELD_GEN")
+    @SequenceGenerator(name = "SEQ_POLICY_FIELD_GEN", sequenceName = "SEQ_POLICY_FIELD", initialValue = 1, allocationSize = 1)
+    private Long id;
+
+    @Column(name = "FIELD")
+    private String field;
+
+    @Column(name = "FIND_RULE")
+    @Enumerated(EnumType.STRING)
+    private FieldFindRule findRule;
+
+    @ManyToOne(cascade = {CascadeType.PERSIST, CascadeType.MERGE})
+    @JoinColumn(name = "POLICY_ID", insertable = false, updatable = false)
+    private Policy policy;
+}

kouncil-backend/src/main/java/com/consdata/kouncil/model/datamasking/PolicyResource.java

@@ -1,6 +1,5 @@
 package com.consdata.kouncil.model.datamasking;
 
-import com.consdata.kouncil.model.cluster.Cluster;
 import javax.persistence.CascadeType;
 import javax.persistence.Column;
 import javax.persistence.Entity;
@@ -26,9 +25,8 @@ public class PolicyResource {
     @SequenceGenerator(name = "SEQ_POLICY_RESOURCE_GEN", sequenceName = "SEQ_POLICY_RESOURCE", initialValue = 1, allocationSize = 1)
     private Long id;
 
-    @ManyToOne(cascade = {CascadeType.PERSIST, CascadeType.MERGE})
-    @JoinColumn(name = "CLUSTER_ID", insertable = false, updatable = false)
-    private Cluster cluster;
+    @Column(name = "CLUSTER_ID")
+    private Long cluster;
 
     @Column(name = "TOPIC")
     private String topic;

kouncil-backend/src/main/resources/db/migration/V8__data_masking.sql

@@ -1,15 +1,18 @@
 create table policy
 (
-    id           bigint not null primary key,
-    name         varchar(255),
-    masking_type varchar(255),
-    apply_to_all_resources boolean
+    id                     bigint not null primary key,
+    apply_to_all_resources boolean,
+    name                   varchar(255),
+    masking_type           varchar(255)
 );
-create table policy_fields
+
+create table policy_field
 (
-    policy_id bigint not null
-        constraint fkq4ltt72s1qqgpolry1fnmy67b references policy,
-    field     varchar(255)
+    id        bigint not null primary key,
+    field     varchar(255),
+    find_rule varchar(255),
+    policy_id bigint
+        constraint fkppv2kuelp29jag142kqgvober references policy
 );
 
 create table policy_resource
@@ -24,6 +27,7 @@ create table policy_resource
 
 CREATE SEQUENCE SEQ_POLICY MINVALUE 1 START WITH 1 INCREMENT BY 1 CACHE 10;
 CREATE SEQUENCE SEQ_POLICY_RESOURCE MINVALUE 1 START WITH 1 INCREMENT BY 1 CACHE 10;
+CREATE SEQUENCE SEQ_POLICY_FIELD MINVALUE 1 START WITH 1 INCREMENT BY 1 CACHE 10;
 
 insert into system_function(id, name, label, function_group)
 VALUES (nextval('SEQ_SYSTEM_FUNCTION'), 'POLICY_LIST', 'Policies list', 'DATA_MASKING'),

kouncil-backend/src/test/java/com/consdata/kouncil/datamasking/converter/PolicyConverterTest.java

@@ -0,0 +1,55 @@
+package com.consdata.kouncil.datamasking.converter;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.junit.jupiter.api.Assertions.assertAll;
+
+import com.consdata.kouncil.datamasking.dto.PolicyDto;
+import com.consdata.kouncil.datamasking.dto.PolicyFieldDto;
+import com.consdata.kouncil.datamasking.dto.PolicyResourceDto;
+import com.consdata.kouncil.model.datamasking.MaskingType;
+import com.consdata.kouncil.model.datamasking.Policy;
+import java.util.HashSet;
+import org.junit.jupiter.api.Test;
+
+class PolicyConverterTest {
+
+    @Test
+    void should_convert_to_entity() {
+        //given
+        PolicyDto policyDto = new PolicyDto();
+        policyDto.setId(1L);
+        policyDto.setName("test");
+        policyDto.setApplyToAllResources(false);
+        policyDto.setMaskingType(MaskingType.ALL);
+        policyDto.setFields(new HashSet<>());
+        policyDto.getFields().add(createField(1L));
+        policyDto.getFields().add(createField(2L));
+        policyDto.getFields().add(createField(3L));
+        policyDto.setResources(new HashSet<>());
+        policyDto.getResources().add(createResource(1L));
+        policyDto.getResources().add(createResource(2L));
+        //when
+        Policy policy = PolicyConverter.convert(policyDto);
+        //then
+        assertAll(
+                () -> assertThat(policy.getId()).isEqualTo(policyDto.getId()),
+                () -> assertThat(policy.getName()).isEqualTo(policyDto.getName()),
+                () -> assertThat(policy.getApplyToAllResources()).isEqualTo(policyDto.getApplyToAllResources()),
+                () -> assertThat(policy.getMaskingType()).isEqualTo(policyDto.getMaskingType()),
+                () -> assertThat(policy.getFields()).hasSize(policyDto.getFields().size()),
+                () -> assertThat(policy.getResources()).hasSize(policyDto.getResources().size())
+        );
+    }
+
+    private PolicyResourceDto createResource(long id) {
+        PolicyResourceDto policyResource = new PolicyResourceDto();
+        policyResource.setId(id);
+        return policyResource;
+    }
+
+    private PolicyFieldDto createField(long id) {
+        PolicyFieldDto policyField = new PolicyFieldDto();
+        policyField.setId(id);
+        return policyField;
+    }
+}