Remove Trekktype and refactor related code

#deploy-skatte-service

Removed the Trekktype class and refactored related code to use Forskuddstrekk directly. Updated mapping strategies and utility to align with the new structure. This change simplifies the data model and improves maintainability.

apps/skattekort-service/src/main/java/no/nav/skattekortservice/dto/SkattekortResponsIntermediate.java

@@ -1,11 +1,15 @@
 package no.nav.skattekortservice.dto;
 
+import com.fasterxml.jackson.annotation.JsonSubTypes;
+import com.fasterxml.jackson.annotation.JsonTypeInfo;
 import lombok.AllArgsConstructor;
 import lombok.Builder;
 import lombok.Data;
+import lombok.EqualsAndHashCode;
 import lombok.Getter;
 import lombok.NoArgsConstructor;
 import lombok.RequiredArgsConstructor;
+import lombok.experimental.SuperBuilder;
 import no.nav.testnav.libs.dto.skattekortservice.v1.IdentifikatorForEnhetEllerPerson;
 
 import java.time.LocalDate;
@@ -88,9 +92,6 @@ public static class Skattekort {
         private LocalDate utstedtDato;
         private Long skattekortidentifikator;
         private List<Forskuddstrekk> forskuddstrekk;
-        private List<Frikort> frikort;
-        private List<Trekktabell> trekktabell;
-        private List<Trekkprosent> trekkprosent;
 
         public List<Forskuddstrekk> getForskuddstrekk() {
 
@@ -99,71 +100,53 @@ public List<Forskuddstrekk> getForskuddstrekk() {
             }
             return forskuddstrekk;
         }
-
-        public List<Frikort> getFrikort() {
-
-            if (isNull(frikort)) {
-                frikort = new ArrayList<>();
-            }
-            return frikort;
-        }
-
-        public List<Trekktabell> getTrekktabell() {
-
-            if (isNull(trekktabell)) {
-                trekktabell = new ArrayList<>();
-            }
-            return trekktabell;
-        }
-
-        public List<Trekkprosent> getTrekkprosent() {
-
-            if (isNull(trekkprosent)) {
-                trekkprosent = new ArrayList<>();
-            }
-            return trekkprosent;
-        }
     }
 
     @Data
-    @Builder
+    @SuperBuilder
     @NoArgsConstructor
     @AllArgsConstructor
-    public static class Forskuddstrekk {
+    @JsonTypeInfo(use = JsonTypeInfo.Id.NAME, property = "xsi:type")
+    @JsonSubTypes({
+            @JsonSubTypes.Type(value = Frikort.class, name = "Frikort"),
+            @JsonSubTypes.Type(value = Trekktabell.class, name = "Trekktabell"),
+            @JsonSubTypes.Type(value = Trekkprosent.class, name = "Trekkprosent")
+    })
+    public abstract static class Forskuddstrekk {
 
         private Trekkode trekkode;
+        private String type;
     }
 
+    @EqualsAndHashCode(callSuper = true)
     @Data
-    @Builder
+    @SuperBuilder
     @NoArgsConstructor
     @AllArgsConstructor
-    public static class Frikort {
-
-        private Trekkode trekkode;
+    public static class Frikort extends Forskuddstrekk {
 
         private Integer frikortbeloep;
     }
 
+    @EqualsAndHashCode(callSuper = true)
     @Data
-    @Builder
+    @SuperBuilder
     @NoArgsConstructor
     @AllArgsConstructor
-    public static class Trekktabell {
-
-        private Trekkode trekkode;
+    public static class Trekktabell extends Forskuddstrekk {
 
         private Tabelltype tabelltype;
         private String tabellnummer;
         private Integer prosentsats;
         private Integer antallMaanederForTrekk;
     }
 
+    @EqualsAndHashCode(callSuper = true)
     @Data
-    @Builder
+    @SuperBuilder
     @NoArgsConstructor
     @AllArgsConstructor
-    public static class Trekkprosent {
+    public static class Trekkprosent extends Forskuddstrekk {
 
         private Trekkode trekkode;
 

apps/skattekort-service/src/main/java/no/nav/skattekortservice/mapper/ArbeidsgiverMappingStrategy.java

@@ -5,7 +5,6 @@
 import ma.glasnost.orika.MapperFactory;
 import ma.glasnost.orika.MappingContext;
 import no.nav.skattekortservice.dto.SkattekortResponsIntermediate;
-import no.nav.testnav.libs.dto.skattekortservice.v1.Forskuddstrekk;
 import no.nav.testnav.libs.dto.skattekortservice.v1.Frikort;
 import no.nav.testnav.libs.dto.skattekortservice.v1.Resultatstatus;
 import no.nav.testnav.libs.dto.skattekortservice.v1.Skattekort;
@@ -17,6 +16,8 @@
 import no.nav.testnav.libs.dto.skattekortservice.v1.Trekktabell;
 import org.springframework.stereotype.Component;
 
+import static java.util.Objects.nonNull;
+
 @Slf4j
 @Component
 public class ArbeidsgiverMappingStrategy implements MappingStrategy {
@@ -39,22 +40,22 @@ public void mapAtoB(SkattekortResponsIntermediate.Skattekortmelding source, Skat
                 })
                 .register();
 
-        factory.classMap(SkattekortResponsIntermediate.Forskuddstrekk.class, Forskuddstrekk.class)
-                .customize(new CustomMapper<>() {
-                    @Override
-                    public void mapAtoB(SkattekortResponsIntermediate.Forskuddstrekk source, Forskuddstrekk target, MappingContext context) {
-
-                        target.setTrekkode(Trekkode.valueOf(source.getTrekkode().getValue()));
-                    }
-                })
-                .register();
+//        factory.classMap(SkattekortResponsIntermediate.Forskuddstrekk.class, Forskuddstrekk.class)
+//                .customize(new CustomMapper<>() {
+//                    @Override
+//                    public void mapAtoB(SkattekortResponsIntermediate.Forskuddstrekk source, Forskuddstrekk target, MappingContext context) {
+//
+//                        target.setTrekkode(Trekkode.valueOf(source.getTrekkode().getValue()));
+//                    }
+//                })
+//                .register();
 
         factory.classMap(SkattekortResponsIntermediate.Frikort.class, Frikort.class)
                 .customize(new CustomMapper<>() {
                     @Override
                     public void mapAtoB(SkattekortResponsIntermediate.Frikort source, Frikort target, MappingContext context) {
 
-                        target.setTrekkode(Trekkode.valueOf(source.getTrekkode().getValue()));
+                        target.setTrekkode(getTrekkodeValue(source.getTrekkode()));
                         target.setFrikortbeloep(source.getFrikortbeloep());
                     }
                 })
@@ -65,7 +66,7 @@ public void mapAtoB(SkattekortResponsIntermediate.Frikort source, Frikort target
                     @Override
                     public void mapAtoB(SkattekortResponsIntermediate.Trekktabell source, Trekktabell target, MappingContext context) {
 
-                        target.setTrekkode(Trekkode.valueOf(source.getTrekkode().getValue()));
+                        target.setTrekkode(getTrekkodeValue(source.getTrekkode()));
                         target.setTabelltype(Tabelltype.valueOf(source.getTabelltype().getValue()));
                         target.setTabellnummer(source.getTabellnummer());
                         target.setProsentsats(source.getProsentsats());
@@ -79,11 +80,16 @@ public void mapAtoB(SkattekortResponsIntermediate.Trekktabell source, Trekktabel
                     @Override
                     public void mapAtoB(SkattekortResponsIntermediate.Trekkprosent source, Trekkprosent target, MappingContext context) {
 
-                        target.setTrekkode(Trekkode.valueOf(target.getTrekkode().getValue()));
+                        target.setTrekkode(getTrekkodeValue(source.getTrekkode()));
                         target.setProsentsats(source.getProsentsats());
                         target.setAntallMaanederForTrekk(source.getAntallMaanederForTrekk());
                     }
                 })
                 .register();
     }
+
+    private static Trekkode getTrekkodeValue(SkattekortResponsIntermediate.Trekkode trekkode) {
+
+        return nonNull(trekkode.getValue()) ? Trekkode.valueOf(trekkode.getValue()) : null;
+    }
 }

apps/skattekort-service/src/main/java/no/nav/skattekortservice/mapper/TrekktypeMappingStrategy.java

@@ -9,7 +9,6 @@
 import no.nav.testnav.libs.dto.skattekortservice.v1.Skattekort;
 import no.nav.testnav.libs.dto.skattekortservice.v1.Trekkprosent;
 import no.nav.testnav.libs.dto.skattekortservice.v1.Trekktabell;
-import no.nav.testnav.libs.dto.skattekortservice.v1.Trekktype;
 import org.springframework.stereotype.Component;
 
 import java.util.Objects;
@@ -26,7 +25,7 @@ public void register(MapperFactory factory) {
                     @Override
                     public void mapAtoB(Skattekort skattekort, no.skatteetaten.fastsetting.formueinntekt.forskudd.skattekorttilarbeidsgiver.v3.Skattekort skattekort2, MappingContext context) {
 
-                        skattekort2.getForskuddstrekk().addAll(skattekort.getTrekktype().stream()
+                        skattekort2.getForskuddstrekk().addAll(skattekort.getForskuddstrekk().stream()
                                 .map(trekktype -> {
                                     if (nonNull(trekktype.getFrikort())) {
                                         return mapperFacade.map(trekktype.getFrikort(),
@@ -37,9 +36,6 @@ public void mapAtoB(Skattekort skattekort, no.skatteetaten.fastsetting.formueinn
                                     } else if (nonNull(trekktype.getTrekkprosent())) {
                                         return mapperFacade.map(trekktype.getTrekkprosent(),
                                                 no.skatteetaten.fastsetting.formueinntekt.forskudd.skattekorttilarbeidsgiver.v3.Trekkprosent.class);
-                                    } else if (nonNull(trekktype.getForskuddstrekk())) {
-                                        return mapperFacade.map(trekktype.getForskuddstrekk(),
-                                                no.skatteetaten.fastsetting.formueinntekt.forskudd.skattekorttilarbeidsgiver.v3.Forskuddstrekk.class);
                                     } else {
                                         return null;
                                     }
@@ -48,7 +44,7 @@ public void mapAtoB(Skattekort skattekort, no.skatteetaten.fastsetting.formueinn
                                 .toList());
                     }
                 })
-                .exclude("trekktype")
+                .exclude("forskuddstrekk")
                 .byDefault()
                 .register();
 
@@ -57,29 +53,21 @@ public void mapAtoB(Skattekort skattekort, no.skatteetaten.fastsetting.formueinn
                     @Override
                     public void mapAtoB(SkattekortResponsIntermediate.Skattekort skattekort, Skattekort skattekort2, MappingContext context) {
 
-                        skattekort2.getTrekktype().addAll(skattekort.getFrikort().stream()
-                                .map(frikort -> Trekktype.builder()
-                                        .frikort(mapperFacade.map(frikort, Frikort.class))
-                                        .build())
-                                .toList());
-                        skattekort2.getTrekktype().addAll(skattekort.getForskuddstrekk().stream()
-                                .map(forskuddstrekk -> Trekktype.builder()
-                                        .forskuddstrekk(mapperFacade.map(forskuddstrekk, Forskuddstrekk.class))
-                                        .build())
-                                .toList());
-                        skattekort2.getTrekktype().addAll(skattekort.getTrekkprosent().stream()
-                                .map(trekkprosent -> Trekktype.builder()
-                                        .trekkprosent(mapperFacade.map(trekkprosent, Trekkprosent.class))
-                                        .build())
-                                .toList());
-                        skattekort2.getTrekktype().addAll(skattekort.getTrekktabell().stream()
-                                .map(trekktabell -> Trekktype.builder()
-                                        .trekktabell(mapperFacade.map(trekktabell, Trekktabell.class))
-                                        .build())
-                                .toList());
+                        skattekort2.setSkattekortidentifikator(skattekort.getSkattekortidentifikator());
+                        skattekort2.setUtstedtDato(skattekort.getUtstedtDato());
+                        skattekort.getForskuddstrekk()
+                                .forEach(forskuddstrekk -> skattekort2.getForskuddstrekk().add(
+                                        Forskuddstrekk.builder()
+                                                .frikort(forskuddstrekk instanceof SkattekortResponsIntermediate.Frikort ?
+                                                        mapperFacade.map(forskuddstrekk, Frikort.class) : null)
+                                                .trekktabell(forskuddstrekk instanceof SkattekortResponsIntermediate.Trekktabell ?
+                                                        mapperFacade.map(forskuddstrekk, Trekktabell.class) : null)
+                                                .trekkprosent(forskuddstrekk instanceof SkattekortResponsIntermediate.Trekkprosent ?
+                                                        mapperFacade.map(forskuddstrekk, Trekkprosent.class) : null)
+                                                .build()
+                                ));
                     }
                 })
-                .byDefault()
                 .register();
     }
 }

apps/skattekort-service/src/main/java/no/nav/skattekortservice/utility/SkattekortValidator.java

@@ -28,7 +28,7 @@ private static void validateSkattekort(SkattekortRequestDTO skattekort) {
                 .map(ArbeidsgiverSkatt::getArbeidstaker)
                 .flatMap(Collection::stream)
                 .map(Skattekortmelding::getSkattekort)
-                .map(Skattekort::getTrekktype)
+                .map(Skattekort::getForskuddstrekk)
                 .flatMap(Collection::stream)
                 .forEach(trekktype -> {
                     if (trekktype.isAllEmpty()) {

libs/data-transfer-objects/src/main/java/no/nav/testnav/libs/dto/skattekortservice/v1/Forskuddstrekk.java

@@ -1,15 +1,46 @@
 package no.nav.testnav.libs.dto.skattekortservice.v1;
 
-import lombok.AllArgsConstructor;
-import lombok.Builder;
+import com.fasterxml.jackson.annotation.JsonIgnore;
 import lombok.Data;
 import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+
+import java.util.Arrays;
+import java.util.Objects;
 
 @Data
-@Builder
+@SuperBuilder
 @NoArgsConstructor
-@AllArgsConstructor
 public class Forskuddstrekk {
 
-    private Trekkode trekkode;
+    private Frikort frikort;
+    private Trekktabell trekktabell;
+    private Trekkprosent trekkprosent;
+
+    @JsonIgnore
+    public boolean isAllEmpty() {
+
+        return contentsCount() == 0;
+    }
+
+    @JsonIgnore
+    public boolean isAmbiguous() {
+
+        return contentsCount() > 1;
+    }
+
+    private long contentsCount() {
+
+        return Arrays.stream(getClass().getDeclaredFields())
+                .map(field -> {
+                    try {
+                        field.setAccessible(true);
+                        return field.get(this);
+                    } catch (IllegalAccessException e) {
+                        throw new RuntimeException(e);
+                    }
+                })
+                .filter(Objects::nonNull)
+                .count();
+    }
 }

libs/data-transfer-objects/src/main/java/no/nav/testnav/libs/dto/skattekortservice/v1/Skattekort.java

@@ -19,14 +19,14 @@ public class Skattekort {
 
     private LocalDate utstedtDato;
     private Long skattekortidentifikator;
-    private List<Trekktype> trekktype;
+    private List<Forskuddstrekk> forskuddstrekk;
 
-    public List<Trekktype> getTrekktype() {
+    public List<Forskuddstrekk> getForskuddstrekk() {
 
-        if (isNull(trekktype)) {
-            trekktype = new ArrayList<>();
+        if (isNull(forskuddstrekk)) {
+            forskuddstrekk = new ArrayList<>();
         }
-        return trekktype;
 
+        return forskuddstrekk;
     }
 }