From f5dfa6f140bbe3f712da76b0d078c3e0fbba1447 Mon Sep 17 00:00:00 2001
From: Anastasia Lubennikova <anastasia@neon.tech>
Date: Thu, 23 Nov 2023 09:36:51 +0000
Subject: [PATCH] Create extension neon in existing databases too

---
 compute_tools/src/spec.rs | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/compute_tools/src/spec.rs b/compute_tools/src/spec.rs
index 8a94950829..ae705702c5 100644
--- a/compute_tools/src/spec.rs
+++ b/compute_tools/src/spec.rs
@@ -733,6 +733,13 @@ pub fn handle_alter_extension_neon(
         // which may happen in two cases:
         // - extension was just installed
         // - extension was already installed and is up to date
+        let create_extension_neon_query = "CREATE EXTENSION IF NOT EXISTS neon";
+        info!(
+            "create extension neon query for db {} : {}",
+            &db.name, &create_extension_neon_query
+        );
+        db_client.simple_query(create_extension_neon_query)?;
+
         let alter_extension_neon_query = "ALTER EXTENSION neon UPDATE";
         info!(
             "alter extension neon query for db {} : {}",