Support clickhouse for sqlparser (#35)

* support clickhouse for sqlparser

* lint

Author: Amogh-Bharadwaj

src/ast/mod.rs

@@ -2876,6 +2876,7 @@ pub enum PeerType {
     Eventhubs,
     PubSub,
     Elasticsearch,
+    Clickhouse,
 }
 
 impl fmt::Display for PeerType {
@@ -2892,6 +2893,7 @@ impl fmt::Display for PeerType {
             PeerType::Eventhubs => write!(f, "EVENTHUBS"),
             PeerType::PubSub => write!(f, "PUBSUB"),
             PeerType::Elasticsearch => write!(f, "ELASTICSEARCH"),
+            PeerType::Clickhouse => write!(f, "CLICKHOUSE"),
         }
     }
 }

src/keywords.rs

@@ -147,6 +147,7 @@ define_keywords!(
     CHARSET,
     CHAR_LENGTH,
     CHECK,
+    CLICKHOUSE,
     CLOB,
     CLONE,
     CLOSE,

src/parser/mod.rs

@@ -9828,6 +9828,7 @@ impl<'a> Parser<'a> {
             Keyword::EVENTHUBS,
             Keyword::PUBSUB,
             Keyword::ELASTICSEARCH,
+            Keyword::CLICKHOUSE,
         ]) {
             Some(Keyword::BIGQUERY) => Ok(PeerType::Bigquery),
             Some(Keyword::MONGO) => Ok(PeerType::Mongo),
@@ -9840,6 +9841,7 @@ impl<'a> Parser<'a> {
             Some(Keyword::EVENTHUBS) => Ok(PeerType::Eventhubs),
             Some(Keyword::PUBSUB) => Ok(PeerType::PubSub),
             Some(Keyword::ELASTICSEARCH) => Ok(PeerType::Elasticsearch),
+            Some(Keyword::CLICKHOUSE) => Ok(PeerType::Clickhouse),
             other => {
                 let supported_peer_types = [
                     "BIGQUERY",
@@ -9853,6 +9855,7 @@ impl<'a> Parser<'a> {
                     "EVENTHUBS",
                     "PUBSUB",
                     "ELASTICSEARCH",
+                    "CLICKHOUSE",
                 ];
                 let err = format!(
                     "expected peertype as one of {}, got {:#?}",

tests/sqlparser_postgres.rs

@@ -3678,6 +3678,33 @@ fn parse_create_s3_peer() {
     }
 }
 
+#[test]
+fn parse_create_clickhouse_peer() {
+    match pg().verified_stmt(
+        "CREATE PEER clickhouse_1 FROM CLICKHOUSE WITH \
+            (host = 'http://clickhouse-server:8123')",
+    ) {
+        Statement::CreatePeer {
+            if_not_exists: _,
+            peer_name: _,
+            peer_type,
+            with_options,
+        } => {
+            assert_eq!(peer_type, PeerType::Clickhouse);
+            assert_eq!(
+                with_options,
+                vec![SqlOption {
+                    name: Ident::new("host"),
+                    value: Expr::Value(Value::SingleQuotedString(String::from(
+                        "http://clickhouse-server:8123"
+                    )))
+                },]
+            );
+        }
+        _ => unreachable!(),
+    }
+}
+
 #[test]
 fn parse_create_single_mirror() {
     match pg().verified_stmt("CREATE MIRROR IF NOT EXISTS test_mirror FROM p1 TO p2 WITH TABLE MAPPING ({from : s1.t1, to : s2.t2}) WITH (key1 = 'value1')") {