rename to pg_vectorize

.gitignore

@@ -4,3 +4,6 @@
 *.iml
 **/*.rs.bk
 Cargo.lock
+.env
+pg_cron
+test.sql

Cargo.toml

@@ -1,7 +1,8 @@
 [package]
-name = "tembo"
+name = "vectorize"
 version = "0.0.0"
 edition = "2021"
+publish = false
 
 [lib]
 crate-type = ["cdylib"]

README.md

@@ -5,7 +5,7 @@ Dev pgrx settings
 Required postgresql.conf settings
 
 ```
-cron.database_name = 'tembo'
+cron.database_name = 'vectorize'
 ```
 
 

sql/meta.sql

@@ -1,4 +1,4 @@
-CREATE TABLE tembo_meta (
+CREATE TABLE vectorize_meta (
     job_id bigserial,
     name TEXT NOT NULL UNIQUE,
     job_type TEXT NOT NULL,

src/executor.rs

@@ -15,9 +15,9 @@ use sqlx::types::chrono::Utc;
 use sqlx::{FromRow, PgPool, Pool, Postgres, Row};
 
 // schema for every job
-// also schema for the tembo.tembo_meta table
+// also schema for the vectorize.vectorize_meta table
 #[derive(Clone, Debug, Deserialize, FromRow, Serialize)]
-pub struct TemboMeta {
+pub struct VectorizeMeta {
     pub job_id: i64,
     pub name: String,
     pub job_type: types::JobType,
@@ -42,7 +42,7 @@ pub struct ColumnJobParams {
 #[derive(Clone, Deserialize, Debug, Serialize)]
 pub struct JobMessage {
     pub job_name: String,
-    pub job_meta: TemboMeta,
+    pub job_meta: VectorizeMeta,
     pub inputs: Vec<Inputs>,
 }
 
@@ -66,7 +66,7 @@ fn job_execute(job_name: String) -> pgrx::JsonB {
         let queue = pgmq::PGMQueueExt::new(db_url, 2)
             .await
             .expect("failed to init db connection");
-        let meta = get_tembo_meta(&job_name, conn)
+        let meta = get_vectorize_meta(&job_name, conn)
             .await
             .expect("failed to get job meta");
         let job_params = serde_json::from_value::<ColumnJobParams>(meta.params.clone())
@@ -101,15 +101,15 @@ fn job_execute(job_name: String) -> pgrx::JsonB {
 }
 
 // get job meta
-pub async fn get_tembo_meta(
+pub async fn get_vectorize_meta(
     job_name: &str,
     conn: Pool<Postgres>,
-) -> Result<TemboMeta, DatabaseError> {
+) -> Result<VectorizeMeta, DatabaseError> {
     let row = sqlx::query_as!(
-        TemboMeta,
+        VectorizeMeta,
         "
         SELECT *
-        FROM tembo.tembo_meta
+        FROM vectorize.vectorize_meta
         WHERE name = $1
         ",
         job_name.to_string(),
@@ -188,7 +188,7 @@ fn get_inputs_query(
     WHERE {last_updated_col} > 
     (
         SELECT last_completion
-        FROM tembo_meta
+        FROM vectorize_meta
         WHERE name = '{job_name}'
     )::timestamp
     "

src/init.rs

@@ -5,7 +5,7 @@ use crate::{
     types::{self, Transformer},
 };
 
-pub const PGMQ_QUEUE_NAME: &str = "tembo_vectorize_queue";
+pub const PGMQ_QUEUE_NAME: &str = "vectorize_queue";
 
 #[pg_extern]
 fn init_table(
@@ -92,10 +92,10 @@ fn init_pgmq() -> Result<(), spi::Error> {
 fn init_job_query() -> String {
     format!(
         "
-        INSERT INTO {schema}.tembo_meta (name, job_type, transformer, search_alg, params)
+        INSERT INTO {schema}.vectorize_meta (name, job_type, transformer, search_alg, params)
         VALUES ($1, $2, $3, $4, $5);
         ",
-        schema = types::TEMBO_SCHEMA
+        schema = types::VECTORIZE_SCHEMA
     )
 }
 
@@ -109,6 +109,6 @@ fn init_embedding_table(job_name: &str) -> String {
             updated_at TIMESTAMP WITH TIME ZONE DEFAULT (now() at time zone 'utc') not null
         );
         ",
-        schema = types::TEMBO_SCHEMA
+        schema = types::VECTORIZE_SCHEMA
     )
 }

src/types.rs

@@ -3,7 +3,7 @@ use serde::{Deserialize, Serialize};
 use std::fmt::{Display, Formatter};
 use std::str::FromStr;
 
-pub const TEMBO_SCHEMA: &str = "tembo";
+pub const VECTORIZE_SCHEMA: &str = "vectorize";
 
 #[allow(non_camel_case_types)]
 #[derive(Clone, Debug, Serialize, Deserialize, PostgresEnum)]

src/worker.rs

@@ -25,9 +25,9 @@ use crate::executor::JobMessage;
 
 #[pg_guard]
 pub extern "C" fn _PG_init() {
-    BackgroundWorkerBuilder::new("PG Tembo Background Worker")
+    BackgroundWorkerBuilder::new("PG Vectorize Background Worker")
         .set_function("background_worker_main")
-        .set_library("tembo")
+        .set_library("vectorize")
         .enable_spi_access()
         .load();
 }

vectorize.control

@@ -0,0 +1,7 @@
+comment = 'Vector search in two functions'
+default_version = '@CARGO_VERSION@'
+module_pathname = '$libdir/vectorize'
+relocatable = false
+superuser = true
+schema = 'vectorize'
+requires = 'pg_cron,pgmq'