BAEL-7032: Improvement to conditional mapping by using @Condition and added tests

mapstruct/src/main/java/com/baeldung/expression/dto/LicenseDto.java

@@ -11,6 +11,8 @@ public class LicenseDto {
 
     private LocalDateTime endDate;
 
+    private String licenseType;
+
     public UUID getId() {
         return id;
     }
@@ -35,4 +37,12 @@ public class LicenseDto {
         this.endDate = endDate;
     }
 
+    public String getLicenseType() {
+        return licenseType;
+    }
+
+    public void setLicenseType(String licenseType) {
+        this.licenseType = licenseType;
+    }
+
 }

mapstruct/src/main/java/com/baeldung/expression/mapper/LicenseMapper.java

@@ -6,6 +6,7 @@ import java.time.OffsetDateTime;
 import java.time.ZoneOffset;
 
 import org.mapstruct.AfterMapping;
+import org.mapstruct.Condition;
 import org.mapstruct.Mapper;
 import org.mapstruct.Mapping;
 import org.mapstruct.MappingTarget;
@@ -40,4 +41,13 @@ public interface LicenseMapper {
             .toDays() <= 14;
     }
 
-}
+    @Condition
+    default boolean mapsToExpectedLicenseType(String licenseType) {
+        try {
+            return licenseType != null && License.LicenseType.valueOf(licenseType) != null;
+        } catch (IllegalArgumentException e) {
+            return false;
+        }
+    }
+
+}

mapstruct/src/main/java/com/baeldung/expression/model/License.java

@@ -16,6 +16,8 @@ public class License {
 
     private boolean renewalRequired;
 
+    private LicenseType licenseType;
+
     public UUID getId() {
         return id;
     }
@@ -55,4 +57,16 @@ public class License {
     public void setRenewalRequired(boolean renewalRequired) {
         this.renewalRequired = renewalRequired;
     }
-}
+
+    public enum LicenseType {
+        INDIVIDUAL, FAMILY
+    }
+
+    public LicenseType getLicenseType() {
+        return licenseType;
+    }
+
+    public void setLicenseType(LicenseType licenseType) {
+        this.licenseType = licenseType;
+    }
+}