fix: added stored procedures

AbelHristodor · Feb 5, 2024 · afebf51 · afebf51
1 parent 7393c3b
commit afebf51
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 13 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "pgcronner"
-version = "0.1.9"
+version = "0.1.13"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

diff --git a/pgcronner.pyi b/pgcronner.pyi
@@ -39,7 +39,7 @@ class PgCronner(object):
         Get all jobs in the table
 
         :return: List of jobs as dicts
-        :throws: OSError
+        :throws: OSError, ValueError
         """
 
     def one(self, jobname: str) -> Optional[Dict[str, str]]:
@@ -49,7 +49,7 @@ class PgCronner(object):
         :param jobname: job name
         :return: Job instance or None, as dict
 
-        :throws: OsError
+        :throws: OsError, ValueError
         """
 
     def add(self, job: Job) -> bool:
@@ -59,7 +59,7 @@ class PgCronner(object):
         :param job: Job instance
         :return: True if successful
 
-        :throws: OsError
+        :throws: OsError, ValueError
         """
 
     def remove(self, jobname: str) -> bool:
@@ -69,7 +69,7 @@ class PgCronner(object):
         :param jobname: job name
         :return: True if successful
 
-        :throws: OsError
+        :throws: OsError, ValueError
         """
 
     def refresh(self) -> bool:
@@ -78,7 +78,7 @@ class PgCronner(object):
         
         :return: True if successful
 
-        :throws: OsError
+        :throws: OsError, ValueError
         """
 
     def clear(self) -> bool:
@@ -87,7 +87,7 @@ class PgCronner(object):
         
         :return: True if successful
         
-        :throws: OsError
+        :throws: OsError, ValueError
         """
 
 
@@ -97,6 +97,6 @@ class PgCronner(object):
         
         :return: True if successful
 
-        :throws: OsError
+        :throws: OsError, ValueError
         """
 
diff --git a/src/utils.rs b/src/utils.rs
@@ -31,11 +31,10 @@ pub fn create_stored_procedure(
 ) -> Result<(), DbError> {
     match client.query(
         &format!(
-            "CREATE OR REPLACE FUNCTION {}() RETURNS void AS $$
-            BEGIN
+            "CREATE OR REPLACE PROCEDURE {}() LANGUAGE SQL
+            BEGIN ATOMIC
                 {}
-            END;
-            $$ LANGUAGE plpgsql;",
+            END;",
             name, source
         ),
         &[],