From 2531bc77cedd6b2bd2867b030b42bbe68bfebb65 Mon Sep 17 00:00:00 2001
From: viragtripathi <virag.tripathi@gmail.com>
Date: Tue, 20 Jan 2026 21:22:29 -0500
Subject: [PATCH] Add CockroachDB PeerType support

Added CockroachDB as a supported peer type following the same pattern as other database types (ClickHouse, Elasticsearch, etc.). This enables PeerDB integration with CockroachDB.
---
 src/ast/mod.rs    | 2 ++
 src/keywords.rs   | 1 +
 src/parser/mod.rs | 3 +++
 3 files changed, 6 insertions(+)

diff --git a/src/ast/mod.rs b/src/ast/mod.rs
index e47e429..83a99c2 100644
--- a/src/ast/mod.rs
+++ b/src/ast/mod.rs
@@ -2877,6 +2877,7 @@ pub enum PeerType {
     PubSub,
     Elasticsearch,
     Clickhouse,
+    CockroachDB,
 }
 
 impl fmt::Display for PeerType {
@@ -2894,6 +2895,7 @@ impl fmt::Display for PeerType {
             PeerType::PubSub => write!(f, "PUBSUB"),
             PeerType::Elasticsearch => write!(f, "ELASTICSEARCH"),
             PeerType::Clickhouse => write!(f, "CLICKHOUSE"),
+            PeerType::CockroachDB => write!(f, "COCKROACHDB"),
         }
     }
 }
diff --git a/src/keywords.rs b/src/keywords.rs
index 0b9cc03..e0399fa 100644
--- a/src/keywords.rs
+++ b/src/keywords.rs
@@ -153,6 +153,7 @@ define_keywords!(
     CLOSE,
     CLUSTER,
     COALESCE,
+    COCKROACHDB,
     COLLATE,
     COLLATION,
     COLLECT,
diff --git a/src/parser/mod.rs b/src/parser/mod.rs
index 087aee3..81825d4 100644
--- a/src/parser/mod.rs
+++ b/src/parser/mod.rs
@@ -9829,6 +9829,7 @@ impl<'a> Parser<'a> {
             Keyword::PUBSUB,
             Keyword::ELASTICSEARCH,
             Keyword::CLICKHOUSE,
+            Keyword::COCKROACHDB,
         ]) {
             Some(Keyword::BIGQUERY) => Ok(PeerType::Bigquery),
             Some(Keyword::MONGO) => Ok(PeerType::Mongo),
@@ -9842,6 +9843,7 @@ impl<'a> Parser<'a> {
             Some(Keyword::PUBSUB) => Ok(PeerType::PubSub),
             Some(Keyword::ELASTICSEARCH) => Ok(PeerType::Elasticsearch),
             Some(Keyword::CLICKHOUSE) => Ok(PeerType::Clickhouse),
+            Some(Keyword::COCKROACHDB) => Ok(PeerType::CockroachDB),
             other => {
                 let supported_peer_types = [
                     "BIGQUERY",
@@ -9856,6 +9858,7 @@ impl<'a> Parser<'a> {
                     "PUBSUB",
                     "ELASTICSEARCH",
                     "CLICKHOUSE",
+                    "COCKROACHDB",
                 ];
                 let err = format!(
                     "expected peertype as one of {}, got {:#?}",