[DPE-3159] Add pgvector extension/plugin (#313)

* add pgvector extension/plugin
* fix charm unit tests
* add pgvector integration test
* fix lint test
* Update snap to revision 95 to revision 96

Author: TakoB222

config.yaml

@@ -219,6 +219,10 @@ options:
     default: false
     type: boolean
     description: Enable spi extension
+  plugin_vector_enable:
+    default: false
+    type: boolean
+    description: Enable pgvector extension
   profile:
    description: |
       Profile representing the scope of deployment, and used to tune resource allocation.

src/config.py

@@ -64,6 +64,7 @@ class CharmConfig(BaseConfigModel):
     plugin_tsm_system_time_enable: bool
     plugin_uuid_ossp_enable: bool
     plugin_spi_enable: bool
+    plugin_vector_enable: bool
     request_date_style: Optional[str]
     request_standard_conforming_strings: Optional[bool]
     request_time_zone: Optional[str]

src/constants.py

@@ -32,7 +32,7 @@
 # Snap constants.
 PGBACKREST_EXECUTABLE = "charmed-postgresql.pgbackrest"
 POSTGRESQL_SNAP_NAME = "charmed-postgresql"
-SNAP_PACKAGES = [(POSTGRESQL_SNAP_NAME, {"revision": "95"})]
+SNAP_PACKAGES = [(POSTGRESQL_SNAP_NAME, {"revision": "96"})]
 
 SNAP_COMMON_PATH = "/var/snap/charmed-postgresql/common"
 SNAP_CURRENT_PATH = "/var/snap/charmed-postgresql/current"

tests/integration/test_plugins.py

@@ -57,6 +57,9 @@
 AUTOINC_EXTENSION_STATEMENT = "CREATE TABLE ids (id int4, idesc text);CREATE TRIGGER ids_nextid BEFORE INSERT OR UPDATE ON ids FOR EACH ROW EXECUTE PROCEDURE autoinc (id, next_id);"
 INSERT_USERNAME_EXTENSION_STATEMENT = "CREATE TABLE username_test (name text, username text not null);CREATE TRIGGER insert_usernames BEFORE INSERT OR UPDATE ON username_test FOR EACH ROW EXECUTE PROCEDURE insert_username (username);"
 MODDATETIME_EXTENSION_STATEMENT = "CREATE TABLE mdt (moddate timestamp DEFAULT CURRENT_TIMESTAMP NOT NULL);CREATE TRIGGER mdt_moddatetime BEFORE UPDATE ON mdt FOR EACH ROW EXECUTE PROCEDURE moddatetime (moddate);"
+VECTOR_EXTENSION_STATEMENT = (
+    "CREATE TABLE vector_test (id bigserial PRIMARY KEY, embedding vector(3));"
+)
 
 
 @pytest.mark.group(1)
@@ -110,6 +113,7 @@ async def test_plugins(ops_test: OpsTest) -> None:
             INSERT_USERNAME_EXTENSION_STATEMENT,
             MODDATETIME_EXTENSION_STATEMENT,
         ],
+        "plugin_vector_enable": VECTOR_EXTENSION_STATEMENT,
     }
 
     def enable_disable_config(enabled: False):

tests/unit/test_charm.py

@@ -396,6 +396,9 @@ def test_enable_disable_extensions(self, _):
   plugin_spi_enable:
     default: false
     type: boolean
+  plugin_vector_enable:
+    default: false
+    type: boolean
   profile:
     default: production
     type: string"""