parsing for CREATE PEER, PubSub and ES

PeerDB-io · Apr 12, 2024 · 6da04de · 6da04de
1 parent 773ae09
commit 6da04de
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/keywords.rs b/src/keywords.rs
@@ -236,6 +236,7 @@ define_keywords!(
     DUPLICATE,
     DYNAMIC,
     EACH,
+    ELASTICSEARCH,
     ELEMENT,
     ELEMENTS,
     ELSE,
@@ -510,6 +511,7 @@ define_keywords!(
     PRIVILEGES,
     PROCEDURE,
     PROGRAM,
+    PUBSUB,
     PURGE,
     QUALIFY,
     QUARTER,

diff --git a/src/parser/mod.rs b/src/parser/mod.rs
@@ -8406,6 +8406,8 @@ impl<'a> Parser<'a> {
             Some(Keyword::S3) => Ok(PeerType::S3),
             Some(Keyword::SQLSERVER) => Ok(PeerType::SQLServer),
             Some(Keyword::EVENTHUBS) => Ok(PeerType::Eventhubs),
+            Some(Keyword::PUBSUB) => Ok(PeerType::PubSub),
+            Some(Keyword::ELASTICSEARCH) => Ok(PeerType::Elasticsearch),
             other => {
                 let supported_peer_types = [
                     "BIGQUERY",
@@ -8416,6 +8418,8 @@ impl<'a> Parser<'a> {
                     "S3",
                     "SQLSERVER",
                     "EVENTHUBS",
+                    "PUBSUB",
+                    "ELASTICSEARCH"
                 ];
                 let err = format!(
                     "expected peertype as one of {}, got {:#?}",