sql: initial sql support

Signed-off-by: Jeronimo Irazabal <jeronimo.irazabal@gmail.com>

src/main/java/io/codenotary/immudb4j/ImmuClient.java

@@ -20,12 +20,18 @@
 import io.codenotary.immudb.ImmuServiceGrpc;
 import io.codenotary.immudb.ImmudbProto;
 import io.codenotary.immudb.ImmudbProto.Chunk;
+import io.codenotary.immudb.ImmudbProto.NamedParam;
+import io.codenotary.immudb.ImmudbProto.NewTxResponse;
 import io.codenotary.immudb.ImmudbProto.ScanRequest;
+import io.codenotary.immudb.ImmudbProto.TxMode;
 import io.codenotary.immudb4j.basics.LatchHolder;
 import io.codenotary.immudb4j.crypto.CryptoUtils;
 import io.codenotary.immudb4j.crypto.DualProof;
 import io.codenotary.immudb4j.crypto.InclusionProof;
 import io.codenotary.immudb4j.exceptions.*;
+import io.codenotary.immudb4j.sql.SQLException;
+import io.codenotary.immudb4j.sql.SQLQueryResult;
+import io.codenotary.immudb4j.sql.SQLValue;
 import io.codenotary.immudb4j.user.Permission;
 import io.codenotary.immudb4j.user.User;
 import io.grpc.ManagedChannel;
@@ -41,9 +47,10 @@
 import java.security.PublicKey;
 import java.util.ArrayList;
 import java.util.Arrays;
+import java.util.HashMap;
 import java.util.Iterator;
 import java.util.List;
-import java.util.NoSuchElementException;
+import java.util.Map;
 import java.util.Timer;
 import java.util.TimerTask;
 import java.util.concurrent.TimeUnit;
@@ -204,6 +211,122 @@ public synchronized ImmuState currentState() throws VerificationException {
         return immuState;
     }
 
+    //
+    // ========== SQL ==========
+    //
+
+    public synchronized void beginTransaction() throws SQLException {
+        if (session == null) {
+            throw new IllegalStateException("no open session");
+        }
+
+        if (session.getTransactionID() != null) {
+            throw new IllegalStateException("transaction already initiated");
+        }
+
+        final ImmudbProto.NewTxRequest req = ImmudbProto.NewTxRequest.newBuilder()
+                .setMode(TxMode.ReadWrite)
+                .build();
+
+        final NewTxResponse res = blockingStub.newTx(req);
+
+        session.setTransactionID(res.getTransactionID());
+    }
+
+    public synchronized void commitTransaction() throws SQLException {
+        if (session == null) {
+            throw new IllegalStateException("no open session");
+        }
+
+        if (session.getTransactionID() == null) {
+            throw new IllegalStateException("no transaction initiated");
+        }
+
+        blockingStub.commit(Empty.getDefaultInstance());
+
+        session.setTransactionID( null);
+    }
+
+    public synchronized void rollbackTransaction() throws SQLException {
+        if (session == null) {
+            throw new IllegalStateException("no open session");
+        }
+
+        if (session.getTransactionID() == null) {
+            throw new IllegalStateException("no transaction initiated");
+        }
+
+        blockingStub.rollback(Empty.getDefaultInstance());
+
+        session.setTransactionID(null);
+    }
+
+    public void sqlExec(String stmt, SQLValue... params) throws SQLException {
+        sqlExec(stmt, sqlNameParams(params));
+    }
+
+    public synchronized void sqlExec(String stmt, Map<String, SQLValue> params) throws SQLException {
+        if (session == null) {
+            throw new IllegalStateException("no open session");
+        }
+
+        if (session.getTransactionID() == null) {
+            throw new IllegalStateException("no transaction initiated");
+        }
+
+        final ImmudbProto.SQLExecRequest req = ImmudbProto.SQLExecRequest.newBuilder()
+                .setSql(stmt)
+                .addAllParams(sqlEncodeParams(params))
+                .build();
+
+        blockingStub.txSQLExec(req);
+    }
+
+    public SQLQueryResult sqlQuery(String stmt, SQLValue... params) throws SQLException {
+        return sqlQuery(stmt, sqlNameParams(params));
+    }
+
+    public synchronized SQLQueryResult sqlQuery(String stmt, Map<String, SQLValue> params) throws SQLException {
+        if (session == null) {
+            throw new IllegalStateException("no open session");
+        }
+
+        if (session.getTransactionID() == null) {
+            throw new IllegalStateException("no transaction initiated");
+        }
+
+        final ImmudbProto.SQLQueryRequest req = ImmudbProto.SQLQueryRequest.newBuilder()
+                .setSql(stmt)
+                .addAllParams(sqlEncodeParams(params))
+                .build();
+
+        return new SQLQueryResult(blockingStub.txSQLQuery(req));
+    }
+
+    private Map<String, SQLValue> sqlNameParams(SQLValue... params) {
+        final Map<String, SQLValue> nparams = new HashMap<>(params.length);
+
+        for (int i = 1; i <= params.length; i++) {
+            nparams.put("param" + i, params[i-1]);
+        }
+
+        return nparams;
+    }
+
+    private Iterable<NamedParam> sqlEncodeParams(Map<String, SQLValue> params) {
+        List<ImmudbProto.NamedParam> nparams = new ArrayList<>();
+
+        for (Map.Entry<String, SQLValue> p : params.entrySet()) {
+            nparams.add(NamedParam.newBuilder()
+                .setName(p.getKey())
+                .setValue(p.getValue().asProtoSQLValue())
+                .build()
+            );
+        }
+
+        return nparams;
+    }
+
     //
     // ========== DATABASE ==========
     //

src/main/java/io/codenotary/immudb4j/ImmudbAuthRequestInterceptor.java

@@ -1,11 +1,32 @@
-package io.codenotary.immudb4j;
+/*
+Copyright 2022 CodeNotary, Inc. All rights reserved.
+
+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
 
-import io.grpc.*;
+	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 io.codenotary.immudb4j;
+
+import io.grpc.CallOptions;
+import io.grpc.Channel;
+import io.grpc.ClientCall;
+import io.grpc.ClientInterceptor;
+import io.grpc.ForwardingClientCall;
+import io.grpc.Metadata;
+import io.grpc.MethodDescriptor;
 
 public class ImmudbAuthRequestInterceptor implements ClientInterceptor {
 
     private static final String SESSION_ID = "sessionid";
+    private static final String TRANSACTION_ID = "transactionid";
 
     private final ImmuClient client;
 
@@ -26,6 +47,10 @@ public void start(Listener<RespT> responseListener, Metadata headers) {
 
                 if (session != null) {
                     headers.put(Metadata.Key.of(SESSION_ID, Metadata.ASCII_STRING_MARSHALLER), session.getSessionID());
+
+                    if (session.getTransactionID() != null) {
+                        headers.put(Metadata.Key.of(TRANSACTION_ID, Metadata.ASCII_STRING_MARSHALLER), session.getTransactionID());
+                    }
                 }
 
                 super.start(responseListener, headers);

src/main/java/io/codenotary/immudb4j/KV.java

@@ -1,3 +1,18 @@
+/*
+Copyright 2022 CodeNotary, Inc. All rights reserved.
+
+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 io.codenotary.immudb4j;
 
 import io.codenotary.immudb4j.crypto.CryptoUtils;

src/main/java/io/codenotary/immudb4j/KVPair.java

@@ -1,3 +1,18 @@
+/*
+Copyright 2022 CodeNotary, Inc. All rights reserved.
+
+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 io.codenotary.immudb4j;
 
 /**

src/main/java/io/codenotary/immudb4j/Session.java

@@ -20,6 +20,8 @@ public class Session {
     private String sessionID;
     private String database;
 
+    private String transactionID;
+
     public Session(String sessionID, String database) {
         this.sessionID = sessionID;
         this.database = database;
@@ -33,4 +35,11 @@ public String getDatabase() {
         return database;
     }
 
+    public String getTransactionID() {
+        return transactionID;
+    }
+
+    public void setTransactionID(String transactionID) {
+        this.transactionID = transactionID;
+    }
 }

src/main/java/io/codenotary/immudb4j/crypto/InclusionProof.java

@@ -16,7 +16,6 @@
 package io.codenotary.immudb4j.crypto;
 
 import io.codenotary.immudb.ImmudbProto;
-import io.codenotary.immudb4j.Utils;
 
 public class InclusionProof {
 

src/main/java/io/codenotary/immudb4j/sql/SQLException.java

@@ -0,0 +1,25 @@
+
+/*
+Copyright 2022 CodeNotary, Inc. All rights reserved.
+
+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 io.codenotary.immudb4j.sql;
+
+public class SQLException extends Exception {
+
+    public SQLException(String message) {
+        super(message);
+    }
+
+}