srowen commented on a change in pull request #24273: [SPARK-27344][SQL][TEST]
Support the LocalDate and Instant classes in Java Bean encoders
URL: https://github.com/apache/spark/pull/24273#discussion_r271320069
##########
File path:
sql/core/src/test/java/test/org/apache/spark/sql/JavaBeanDeserializationSuite.java
##########
@@ -509,4 +514,95 @@ public void setId(Integer id) {
this.id = id;
}
}
+
+ @Test
+ public void testBeanWithLocalDateAndInstant() {
+ String originConf =
spark.conf().get(SQLConf.DATETIME_JAVA8API_ENABLED().key());
+ try {
+ spark.conf().set(SQLConf.DATETIME_JAVA8API_ENABLED().key(), "true");
+ List<Row> inputRows = new ArrayList<>();
+ List<LocalDateInstantRecord> expectedRecords = new ArrayList<>();
+
+ for (long idx = 0 ; idx < 5 ; idx++) {
+ Row row = createLocalDateInstantRow(idx);
+ inputRows.add(row);
+ expectedRecords.add(createLocalDateInstantRecord(row));
+ }
+
+ Encoder<LocalDateInstantRecord> encoder =
Encoders.bean(LocalDateInstantRecord.class);
+
+ StructType schema = new StructType()
+ .add("localDateField", DataTypes.DateType)
+ .add("instantField", DataTypes.TimestampType);
+
+ Dataset<Row> dataFrame = spark.createDataFrame(inputRows, schema);
+ Dataset<LocalDateInstantRecord> dataset = dataFrame.as(encoder);
+
+ List<LocalDateInstantRecord> records = dataset.collectAsList();
+
+ Assert.assertEquals(expectedRecords, records);
+ } finally {
+ spark.conf().set(SQLConf.DATETIME_JAVA8API_ENABLED().key(),
originConf);
+ }
+ }
+
+ public static final class LocalDateInstantRecord {
+ private String localDateField;
+ private String instantField;
+
+ public LocalDateInstantRecord() { }
+
+ public String getLocalDateField() {
+ return localDateField;
+ }
+
+ public void setLocalDateField(String localDateField) {
+ this.localDateField = localDateField;
+ }
+
+ public String getInstantField() {
+ return instantField;
+ }
+
+ public void setInstantField(String instantField) {
+ this.instantField = instantField;
+ }
+
+ @Override
+ public boolean equals(Object o) {
+ if (this == o) return true;
+ if (o == null || getClass() != o.getClass()) return false;
+ LocalDateInstantRecord that = (LocalDateInstantRecord) o;
+ return Objects.equals(localDateField, that.localDateField) &&
+ Objects.equals(instantField, that.instantField);
+ }
+
+ @Override
+ public int hashCode() {
+ return Objects.hash(localDateField, instantField);
+ }
+
+ @Override
+ public String toString() {
+ return com.google.common.base.Objects.toStringHelper(this)
+ .add("localDateField", localDateField)
+ .add("instantField", instantField)
+ .toString();
+ }
+ }
+
+ private static Row createLocalDateInstantRow(Long index) {
+ Object[] values = new Object[] { LocalDate.ofEpochDay(42),
Instant.ofEpochSecond(42) };
Review comment:
Total nit, but I think you can omit "new Object[]"
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
