Restore back modules

Author: kokosing

trino-rest-base/pom.xml

@@ -0,0 +1,59 @@
+<?xml version="1.0"?>
+<!--
+  ~ Licensed 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.
+  -->
+
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+
+    <parent>
+        <groupId>rocks.trino</groupId>
+        <artifactId>trino-rest</artifactId>
+        <version>0.1-SNAPSHOT</version>
+    </parent>
+
+    <artifactId>trino-rest-base</artifactId>
+    <description>Presto rest base</description>
+    <packaging>jar</packaging>
+    <version>0.1-SNAPSHOT</version>
+
+    <properties>
+        <air.main.basedir>${project.parent.basedir}</air.main.basedir>
+    </properties>
+
+    <dependencies>
+        <!-- Presto SPI -->
+        <dependency>
+            <groupId>io.trino</groupId>
+            <artifactId>trino-spi</artifactId>
+            <scope>provided</scope>
+        </dependency>
+
+        <dependency>
+            <groupId>io.airlift</groupId>
+            <artifactId>slice</artifactId>
+            <scope>provided</scope>
+        </dependency>
+
+        <dependency>
+            <groupId>com.fasterxml.jackson.core</groupId>
+            <artifactId>jackson-annotations</artifactId>
+        </dependency>
+
+        <dependency>
+            <groupId>com.google.guava</groupId>
+            <artifactId>guava</artifactId>
+            <scope>provided</scope>
+        </dependency>
+    </dependencies>
+</project>

trino-rest-base/src/main/java/rocks/trino/rest/Rest.java

@@ -0,0 +1,53 @@
+/*
+ * Licensed 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 rocks.trino.rest;
+
+import io.trino.spi.connector.ColumnMetadata;
+import io.trino.spi.connector.ConnectorTableMetadata;
+import io.trino.spi.connector.SchemaTableName;
+import io.trino.spi.type.Type;
+
+import java.util.Collection;
+import java.util.List;
+import java.util.function.Consumer;
+
+import static java.util.stream.Collectors.toList;
+
+public interface Rest
+{
+    ConnectorTableMetadata getTableMetadata(SchemaTableName schemaTableName);
+
+    List<String> listSchemas();
+
+    default List<SchemaTableName> listTables()
+    {
+        return listSchemas().stream()
+                .flatMap(schema -> listTables(schema).stream())
+                .collect(toList());
+    }
+
+    List<SchemaTableName> listTables(String schema);
+
+    Collection<? extends List<?>> getRows(SchemaTableName schemaTableName);
+
+    Consumer<List> createRowSink(SchemaTableName schemaTableName);
+
+    default List<Type> getTypes(SchemaTableName schemaTableName)
+    {
+        return getTableMetadata(schemaTableName).getColumns().stream()
+                .map(ColumnMetadata::getType)
+                .collect(toList());
+    }
+}

trino-rest-base/src/main/java/rocks/trino/rest/RestColumnHandle.java

@@ -0,0 +1,69 @@
+/*
+ * Licensed 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 rocks.trino.rest;
+
+import com.fasterxml.jackson.annotation.JsonCreator;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import io.trino.spi.connector.ColumnHandle;
+import io.trino.spi.type.Type;
+
+import java.util.Objects;
+
+public class RestColumnHandle
+        implements ColumnHandle
+{
+    private final String name;
+    private final Type type;
+
+    @JsonCreator
+    public RestColumnHandle(
+            @JsonProperty("name") String name,
+            @JsonProperty("type") Type type)
+    {
+        this.name = name;
+        this.type = type;
+    }
+
+    @JsonProperty("name")
+    public String getName()
+    {
+        return name;
+    }
+
+    @JsonProperty("type")
+    public Type getType()
+    {
+        return type;
+    }
+
+    @Override
+    public boolean equals(Object o)
+    {
+        if (this == o) {
+            return true;
+        }
+        if (o == null || getClass() != o.getClass()) {
+            return false;
+        }
+        RestColumnHandle that = (RestColumnHandle) o;
+        return Objects.equals(name, that.name);
+    }
+
+    @Override
+    public int hashCode()
+    {
+        return Objects.hash(name);
+    }
+}

trino-rest-base/src/main/java/rocks/trino/rest/RestConnector.java

@@ -0,0 +1,60 @@
+/*
+ * Licensed 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 rocks.trino.rest;
+
+import io.trino.spi.NodeManager;
+import io.trino.spi.connector.Connector;
+import io.trino.spi.connector.ConnectorMetadata;
+import io.trino.spi.connector.ConnectorRecordSetProvider;
+import io.trino.spi.connector.ConnectorSplitManager;
+import io.trino.spi.connector.ConnectorTransactionHandle;
+import io.trino.spi.transaction.IsolationLevel;
+
+public class RestConnector
+        implements Connector
+{
+    private final NodeManager nodeManager;
+    private final Rest rest;
+
+    public RestConnector(NodeManager nodeManager, Rest rest)
+    {
+        this.nodeManager = nodeManager;
+        this.rest = rest;
+    }
+
+    @Override
+    public ConnectorTransactionHandle beginTransaction(IsolationLevel isolationLevel, boolean readOnly)
+    {
+        return new RestTransactionHandle(0);
+    }
+
+    @Override
+    public ConnectorMetadata getMetadata(ConnectorTransactionHandle transaction)
+    {
+        return new RestMetadata(rest);
+    }
+
+    @Override
+    public ConnectorSplitManager getSplitManager()
+    {
+        return new RestSplitManager(nodeManager);
+    }
+
+    @Override
+    public ConnectorRecordSetProvider getRecordSetProvider()
+    {
+        return new RestRecordSetProvider(rest);
+    }
+}

trino-rest-base/src/main/java/rocks/trino/rest/RestConnectorFactory.java

@@ -0,0 +1,95 @@
+/*
+ * Licensed 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 rocks.trino.rest;
+
+import io.trino.spi.NodeManager;
+import io.trino.spi.connector.ColumnHandle;
+import io.trino.spi.connector.Connector;
+import io.trino.spi.connector.ConnectorContext;
+import io.trino.spi.connector.ConnectorFactory;
+import io.trino.spi.connector.ConnectorHandleResolver;
+import io.trino.spi.connector.ConnectorInsertTableHandle;
+import io.trino.spi.connector.ConnectorSplit;
+import io.trino.spi.connector.ConnectorTableHandle;
+import io.trino.spi.connector.ConnectorTableLayoutHandle;
+import io.trino.spi.connector.ConnectorTransactionHandle;
+
+import java.util.Map;
+
+public class RestConnectorFactory
+        implements ConnectorFactory
+{
+    private final RestFactory restFactory;
+    private final String name;
+
+    public RestConnectorFactory(String name, RestFactory restFactory)
+    {
+        this.restFactory = restFactory;
+        this.name = name;
+    }
+
+    @Override
+    public String getName()
+    {
+        return name;
+    }
+
+    @Override
+    public Connector create(String s, Map<String, String> config, ConnectorContext context)
+    {
+        NodeManager nodeManager = context.getNodeManager();
+
+        return new RestConnector(nodeManager, restFactory.create(config));
+    }
+
+    @Override
+    public ConnectorHandleResolver getHandleResolver()
+    {
+        return new ConnectorHandleResolver()
+        {
+            public Class<? extends ConnectorTableHandle> getTableHandleClass()
+            {
+                return RestTableHandle.class;
+            }
+
+            public Class<? extends ColumnHandle> getColumnHandleClass()
+            {
+                return RestColumnHandle.class;
+            }
+
+            public Class<? extends ConnectorSplit> getSplitClass()
+            {
+                return RestConnectorSplit.class;
+            }
+
+            public Class<? extends ConnectorTableLayoutHandle> getTableLayoutHandleClass()
+            {
+                return RestConnectorTableLayoutHandle.class;
+            }
+
+            @Override
+            public Class<? extends ConnectorTransactionHandle> getTransactionHandleClass()
+            {
+                return RestTransactionHandle.class;
+            }
+
+            @Override
+            public Class<? extends ConnectorInsertTableHandle> getInsertTableHandleClass()
+            {
+                return RestInsertTableHandle.class;
+            }
+        };
+    }
+}