public GraphQLFieldDefinition onField(SchemaDirectiveWiringEnvironment<GraphQLFieldDefinition> environment) {
    GraphQLFieldDefinition field = (GraphQLFieldDefinition)environment.getElement();
    DataFetcher originalDataFetcher = (DataFetcher)ClassUtil.call("getDataFetcher", field);
    DataFetcher dataFetcher = DataFetcherFactories.wrapDataFetcher(originalDataFetcher, (dataFetchingEnvironment, value) -> {
        String format = (String)dataFetchingEnvironment.getArgument("format");
        if (StringUtil.isBlank(format)) {
            if (value instanceof LocalDateTime) {
                return Date.from(((LocalDateTime)value).atZone(ZoneId.systemDefault()).toInstant()).getTime();
            } else {
                return value instanceof Date ? value : value;
            }
        } else {
            DateTimeFormatter dateTimeFormatter = this.buildFormatter(format);
            if (value instanceof LocalDateTime) {
                return dateTimeFormatter.format((LocalDateTime)value);
            } else {
                return value instanceof Date ? dateTimeFormatter.format(LocalDateTime.ofInstant(((Date)value).toInstant(), ZoneId.systemDefault())) : value;
            }
        }
    });
    return field.transform((builder) -> {
        builder.argument(GraphQLArgument.newArgument().name("format").type(Scalars.GraphQLString)).dataFetcher(dataFetcher);
    });
}

@Override
public GraphQLInputObjectField onInputObjectField(SchemaDirectiveWiringEnvironment<GraphQLInputObjectField> environment) {
    GraphQLInputObjectField field = environment.getElement();
    Object value = field.getDefaultValue();
    if (value != null) {
        Object pattern = field.getDirective("dateFormat").getArgument("format").getValue();
        String input = value.toString();
        value = LocalDateTime.parse(input, this.buildFormatter(Optional.ofNullable(pattern).map(Object::toString).orElse(null)));
    }
    Object finalValue = value;
    return field.transform(builder -> builder.defaultValue(finalValue).build());
}

private DateTimeFormatter buildFormatter(String format) {
    String dtFormat = format != null ? format : "YYYY-MM-dd'T'HH:mm:ss.SSS'Z'";
    return DateTimeFormatter.ofPattern(dtFormat);
}