feat: support basic auth (#41)

* feat: support basic auth

* fix deps

* update examples

Author: jiacai2050

Cargo.lock

@@ -5,7 +5,7 @@ authors = ["HoraeDB Authors"]
 edition = "2021"
 
 [dependencies]
-horaedb-client = "1.0"
+horaedb-client = { git = "https://github.com/apache/incubator-horaedb-client-rs.git", rev = "cc7a2fd07a8dbaad2d405f310f4239b889709afa" }
 pyo3           = { version = "0.16", features = ["extension-module", "abi3-py37"] }
 pyo3-asyncio   = { version = "0.16", features = ["attributes", "tokio-runtime"] }
 tokio          = { version = "1", features = ["sync"] }

Cargo.toml

@@ -129,9 +129,14 @@ class RpcContext:
     timeout_ms: int
     database: str
 
+class Authorization:
+    def __init__(self): ...
+    username: str
+    password: str
 
 class Builder:
     def __init__(self, endpoint: str): ...
     def set_rpc_config(self, conf: RpcConfig): ...
     def set_default_database(self, db: str): ...
+    def set_authorization(self, auth: Authorization): ...
     def build(self) -> Client: ...

ceresdb_client.pyi

@@ -2,7 +2,7 @@
 
 import asyncio
 import datetime
-from horaedb_client import Builder, RpcContext, PointBuilder, ValueBuilder, WriteRequest, SqlQueryRequest, Mode, RpcConfig
+from horaedb_client import Builder, RpcContext, PointBuilder, ValueBuilder, WriteRequest, SqlQueryRequest, Mode, RpcConfig, Authorization
 
 
 def create_table(ctx):
@@ -75,6 +75,8 @@ def process_write_resp(resp):
     builder = Builder("127.0.0.1:8831", Mode.Direct)
     builder.set_rpc_config(rpc_config)
     builder.set_default_database("public")
+    # Required when server enable auth
+    builder.set_authorization(Authorization("test", "test"))
     client = builder.build()
 
     ctx = RpcContext()

examples/read_write.py

@@ -20,6 +20,7 @@ pub fn register_py_module(m: &PyModule) -> PyResult<()> {
     m.add_class::<Builder>()?;
     m.add_class::<RpcConfig>()?;
     m.add_class::<Mode>()?;
+    m.add_class::<Authorization>()?;
 
     Ok(())
 }
@@ -212,6 +213,30 @@ pub enum Mode {
     Proxy,
 }
 
+#[pyclass]
+#[derive(Debug, Clone)]
+pub struct Authorization {
+    username: String,
+    password: String,
+}
+
+#[pymethods]
+impl Authorization {
+    #[new]
+    pub fn new(username: String, password: String) -> Self {
+        Self { username, password }
+    }
+}
+
+impl From<Authorization> for horaedb_client::Authorization {
+    fn from(auth: Authorization) -> Self {
+        Self {
+            username: auth.username,
+            password: auth.password,
+        }
+    }
+}
+
 #[pymethods]
 impl Builder {
     #[new]
@@ -238,6 +263,11 @@ impl Builder {
         self.rust_builder = Some(builder);
     }
 
+    pub fn set_authorization(&mut self, auth: Authorization) {
+        let builder = self.rust_builder.take().unwrap().authorization(auth.into());
+        self.rust_builder = Some(builder);
+    }
+
     pub fn build(&mut self) -> Client {
         let client = self.rust_builder.take().unwrap().build();
         Client {