[BEAM-9641] Support ZetaSQL DATE type as a Beam LogicalType #11272
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,65 @@ | ||
| /* | ||
| * Licensed to the Apache Software Foundation (ASF) under one | ||
| * or more contributor license agreements. See the NOTICE file | ||
| * distributed with this work for additional information | ||
| * regarding copyright ownership. The ASF licenses this file | ||
| * to you under the Apache License, Version 2.0 (the | ||
| * "License"); you may not use this file except in compliance | ||
| * with the License. You may obtain a copy of the License at | ||
| * | ||
| * http://www.apache.org/licenses/LICENSE-2.0 | ||
| * | ||
| * Unless required by applicable law or agreed to in writing, software | ||
| * distributed under the License is distributed on an "AS IS" BASIS, | ||
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| * See the License for the specific language governing permissions and | ||
| * limitations under the License. | ||
| */ | ||
| package org.apache.beam.sdk.schemas.logicaltypes; | ||
|
|
||
| import java.time.LocalDate; | ||
| import org.apache.beam.sdk.schemas.Schema; | ||
|
|
||
| /** | ||
| * A date without a time-zone. | ||
| * | ||
| * <p>It cannot represent an instant on the time-line without additional information such as an | ||
| * offset or time-zone. | ||
| * | ||
| * <p>Its input type is a {@link LocalDate}, and base type is a {@link Long} that represents a | ||
| * incrementing count of days where day 0 is 1970-01-01 (ISO). | ||
| */ | ||
| public class Date implements Schema.LogicalType<LocalDate, Long> { | ||
|
||
|
|
||
| @Override | ||
| public String getIdentifier() { | ||
| return "beam:logical_type:date:v1"; | ||
| } | ||
|
|
||
| // unused | ||
| @Override | ||
| public Schema.FieldType getArgumentType() { | ||
| return Schema.FieldType.STRING; | ||
| } | ||
|
|
||
| // unused | ||
| @Override | ||
| public String getArgument() { | ||
| return ""; | ||
| } | ||
|
|
||
| @Override | ||
| public Schema.FieldType getBaseType() { | ||
| return Schema.FieldType.INT64; | ||
| } | ||
|
|
||
| @Override | ||
| public Long toBaseType(LocalDate input) { | ||
| return input == null ? null : input.toEpochDay(); | ||
| } | ||
|
|
||
| @Override | ||
| public LocalDate toInputType(Long base) { | ||
| return base == null ? null : LocalDate.ofEpochDay(base); | ||
| } | ||
| } | ||
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,30 @@ | ||
| /* | ||
| * Licensed to the Apache Software Foundation (ASF) under one | ||
| * or more contributor license agreements. See the NOTICE file | ||
| * distributed with this work for additional information | ||
| * regarding copyright ownership. The ASF licenses this file | ||
| * to you under the Apache License, Version 2.0 (the | ||
| * "License"); you may not use this file except in compliance | ||
| * with the License. You may obtain a copy of the License at | ||
| * | ||
| * http://www.apache.org/licenses/LICENSE-2.0 | ||
| * | ||
| * Unless required by applicable law or agreed to in writing, software | ||
| * distributed under the License is distributed on an "AS IS" BASIS, | ||
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| * See the License for the specific language governing permissions and | ||
| * limitations under the License. | ||
| */ | ||
| package org.apache.beam.sdk.schemas.logicaltypes; | ||
|
|
||
| import java.time.LocalDate; | ||
| import org.apache.beam.sdk.schemas.Schema.LogicalType; | ||
|
|
||
| /** Beam {@link org.apache.beam.sdk.schemas.Schema.LogicalType}s corresponding to SQL data types. */ | ||
| public class SqlTypes { | ||
|
|
||
| private SqlTypes() {} | ||
|
|
||
| /** Beam LogicalType corresponding to ZetaSQL/CalciteSQL DATE type. */ | ||
| public static final LogicalType<LocalDate, Long> DATE = new Date(); | ||
| } |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think it is worth documenting that the
Longis an offset from an epoch (and what that epoch is).There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done.