diff --git a/processor/src/main/java/org/mapstruct/ap/internal/conversion/DateToSqlTimeConversion.java b/processor/src/main/java/org/mapstruct/ap/internal/conversion/DateToSqlTimeConversion.java index 767012741..f4470ecd0 100644 --- a/processor/src/main/java/org/mapstruct/ap/internal/conversion/DateToSqlTimeConversion.java +++ b/processor/src/main/java/org/mapstruct/ap/internal/conversion/DateToSqlTimeConversion.java @@ -18,7 +18,12 @@ */ package org.mapstruct.ap.internal.conversion; +import java.sql.Time; +import java.util.Set; + import org.mapstruct.ap.internal.model.common.ConversionContext; +import org.mapstruct.ap.internal.model.common.Type; +import org.mapstruct.ap.internal.util.Collections; /** * Conversion between {@link java.util.Date} and {@link java.sql.Time}. @@ -29,7 +34,12 @@ public class DateToSqlTimeConversion extends SimpleConversion { @Override protected String getToExpression(ConversionContext conversionContext) { - return "new java.sql.Time( .getTime() )"; + return "new Time( .getTime() )"; + } + + @Override + protected Set getToConversionImportTypes(ConversionContext conversionContext) { + return Collections.asSet( conversionContext.getTypeFactory().getType( Time.class ) ); } @Override diff --git a/processor/src/main/java/org/mapstruct/ap/internal/conversion/DateToSqlTimestampConversion.java b/processor/src/main/java/org/mapstruct/ap/internal/conversion/DateToSqlTimestampConversion.java index 291bb279d..a84945d93 100644 --- a/processor/src/main/java/org/mapstruct/ap/internal/conversion/DateToSqlTimestampConversion.java +++ b/processor/src/main/java/org/mapstruct/ap/internal/conversion/DateToSqlTimestampConversion.java @@ -18,7 +18,12 @@ */ package org.mapstruct.ap.internal.conversion; +import java.sql.Timestamp; +import java.util.Set; + import org.mapstruct.ap.internal.model.common.ConversionContext; +import org.mapstruct.ap.internal.model.common.Type; +import org.mapstruct.ap.internal.util.Collections; /** * Conversion between {@link java.util.Date} and {@link java.sql.Timestamp}. @@ -29,7 +34,12 @@ public class DateToSqlTimestampConversion extends SimpleConversion { @Override protected String getToExpression(ConversionContext conversionContext) { - return "new java.sql.Timestamp( .getTime() )"; + return "new Timestamp( .getTime() )"; + } + + @Override + protected Set getToConversionImportTypes(ConversionContext conversionContext) { + return Collections.asSet( conversionContext.getTypeFactory().getType( Timestamp.class ) ); } @Override