Merge pull request #29 from Samrose-Ahmed/master

Make Client::get and some functions public
nooberfsh · Aug 17, 2023 · adf326f · adf326f
2 parents 7ead4d6 + 3d0f011
commit adf326f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/client.rs b/src/client.rs
@@ -98,7 +98,7 @@ impl ClientBuilder {
     }
 
     pub fn schema(mut self, s: impl ToString) -> Self {
-        self.session.catalog = Some(s.to_string());
+        self.session.schema = Some(s.to_string());
         self
     }
 
@@ -423,7 +423,7 @@ impl Client {
         retry!(self, get_next, url, self.max_attempt)
     }
 
-    async fn get<T: Presto + 'static>(&self, sql: String) -> Result<QueryResult<T>> {
+    pub async fn get<T: Presto + 'static>(&self, sql: String) -> Result<QueryResult<T>> {
         let req = self.client.post(self.url.clone()).body(sql);
         let req = {
             let session = self.session.read().await;
@@ -441,7 +441,7 @@ impl Client {
         self.send(req).await
     }
 
-    async fn get_next<T: Presto + 'static>(&self, url: &str) -> Result<QueryResult<T>> {
+    pub async fn get_next<T: Presto + 'static>(&self, url: &str) -> Result<QueryResult<T>> {
         let req = self.client.get(url);
         let req = {
             let session = self.session.read().await;

diff --git a/src/types/data_set.rs b/src/types/data_set.rs
@@ -53,7 +53,7 @@ impl<T: Presto> DataSet<T> {
         self.data.as_slice()
     }
 
-    pub(crate) fn merge(&mut self, other: DataSet<T>) {
+    pub fn merge(&mut self, other: DataSet<T>) {
         self.data.extend(other.data)
     }
 }