Skip to content

Commit f0f1898

Browse files
ebyhrwendigo
authored andcommitted
Convert internal classes to record in Elasticsearch
1 parent a587f84 commit f0f1898

File tree

1 file changed

+10
-60
lines changed

1 file changed

+10
-60
lines changed

plugin/trino-elasticsearch/src/main/java/io/trino/plugin/elasticsearch/ElasticsearchMetadata.java

Lines changed: 10 additions & 60 deletions
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,7 @@ public ConnectorTableMetadata getTableMetadata(ConnectorSession session, Connect
187187
private ConnectorTableMetadata getTableMetadata(String schemaName, String tableName)
188188
{
189189
InternalTableMetadata internalTableMetadata = makeInternalTableMetadata(schemaName, tableName);
190-
return new ConnectorTableMetadata(new SchemaTableName(schemaName, tableName), internalTableMetadata.getColumnMetadata());
190+
return new ConnectorTableMetadata(new SchemaTableName(schemaName, tableName), internalTableMetadata.columnMetadata());
191191
}
192192

193193
private InternalTableMetadata makeInternalTableMetadata(ConnectorTableHandle table)
@@ -225,7 +225,7 @@ private List<ColumnMetadata> makeColumnMetadata(List<IndexMetadata.Field> fields
225225
for (IndexMetadata.Field field : fields) {
226226
result.add(ColumnMetadata.builder()
227227
.setName(field.getName())
228-
.setType(toTrino(field).getType())
228+
.setType(toTrino(field).type())
229229
.build());
230230
}
231231
return result.build();
@@ -243,8 +243,8 @@ private Map<String, ColumnHandle> makeColumnHandles(List<IndexMetadata.Field> fi
243243
TypeAndDecoder converted = toTrino(field);
244244
result.put(field.getName(), new ElasticsearchColumnHandle(
245245
field.getName(),
246-
converted.getType(),
247-
converted.getDecoderDescriptor(),
246+
converted.type(),
247+
converted.decoderDescriptor(),
248248
supportsPredicates(field.getType())));
249249
}
250250

@@ -291,7 +291,7 @@ private TypeAndDecoder toTrino(String prefix, IndexMetadata.Field field)
291291

292292
if (field.isArray()) {
293293
TypeAndDecoder element = toTrino(path, elementField(field));
294-
return new TypeAndDecoder(new ArrayType(element.getType()), new ArrayDecoder.Descriptor(element.getDecoderDescriptor()));
294+
return new TypeAndDecoder(new ArrayType(element.type()), new ArrayDecoder.Descriptor(element.decoderDescriptor()));
295295
}
296296

297297
IndexMetadata.Type type = field.getType();
@@ -337,8 +337,8 @@ else if (type instanceof ObjectType objectType) {
337337
TypeAndDecoder child = toTrino(path, rowField);
338338

339339
if (child != null) {
340-
decoderFields.add(new RowDecoder.NameAndDescriptor(name, child.getDecoderDescriptor()));
341-
rowFieldsBuilder.add(RowType.field(name, child.getType()));
340+
decoderFields.add(new RowDecoder.NameAndDescriptor(name, child.decoderDescriptor()));
341+
rowFieldsBuilder.add(RowType.field(name, child.type()));
342342
}
343343
}
344344

@@ -402,7 +402,7 @@ public Map<String, ColumnHandle> getColumnHandles(ConnectorSession session, Conn
402402
}
403403

404404
InternalTableMetadata tableMetadata = makeInternalTableMetadata(tableHandle);
405-
return tableMetadata.getColumnHandles();
405+
return tableMetadata.columnHandles();
406406
}
407407

408408
@Override
@@ -673,57 +673,7 @@ public Optional<TableFunctionApplicationResult<ConnectorTableHandle>> applyTable
673673
return Optional.of(new TableFunctionApplicationResult<>(tableHandle, columnHandles));
674674
}
675675

676-
private static class InternalTableMetadata
677-
{
678-
private final SchemaTableName tableName;
679-
private final List<ColumnMetadata> columnMetadata;
680-
private final Map<String, ColumnHandle> columnHandles;
681-
682-
public InternalTableMetadata(
683-
SchemaTableName tableName,
684-
List<ColumnMetadata> columnMetadata,
685-
Map<String, ColumnHandle> columnHandles)
686-
{
687-
this.tableName = tableName;
688-
this.columnMetadata = columnMetadata;
689-
this.columnHandles = columnHandles;
690-
}
691-
692-
public SchemaTableName getTableName()
693-
{
694-
return tableName;
695-
}
696-
697-
public List<ColumnMetadata> getColumnMetadata()
698-
{
699-
return columnMetadata;
700-
}
701-
702-
public Map<String, ColumnHandle> getColumnHandles()
703-
{
704-
return columnHandles;
705-
}
706-
}
707-
708-
private static class TypeAndDecoder
709-
{
710-
private final Type type;
711-
private final DecoderDescriptor decoderDescriptor;
712-
713-
public TypeAndDecoder(Type type, DecoderDescriptor decoderDescriptor)
714-
{
715-
this.type = type;
716-
this.decoderDescriptor = decoderDescriptor;
717-
}
676+
private record InternalTableMetadata(SchemaTableName tableName, List<ColumnMetadata> columnMetadata, Map<String, ColumnHandle> columnHandles) {}
718677

719-
public Type getType()
720-
{
721-
return type;
722-
}
723-
724-
public DecoderDescriptor getDecoderDescriptor()
725-
{
726-
return decoderDescriptor;
727-
}
728-
}
678+
private record TypeAndDecoder(Type type, DecoderDescriptor decoderDescriptor) {}
729679
}

0 commit comments

Comments
 (0)