Version 1.2

- Bug fixes
- Add support for array types

Author: za-arthur

.gitignore

@@ -41,3 +41,4 @@ lib*.pc
 /tmp_install/
 Dockerfile
 pg_variables--1.1.sql
+pg_variables--1.2.sql

Makefile

@@ -4,8 +4,8 @@ MODULE_big = pg_variables
 OBJS = pg_variables.o pg_variables_record.o $(WIN32RES)
 
 EXTENSION = pg_variables
-EXTVERSION = 1.1
-DATA = pg_variables--1.0.sql pg_variables--1.0--1.1.sql
+EXTVERSION = 1.2
+DATA = pg_variables--1.0.sql pg_variables--1.0--1.1.sql pg_variables--1.1--1.2.sql
 DATA_built = $(EXTENSION)--$(EXTVERSION).sql
 
 PGFILEDESC = "pg_variables - sessional variables"

README.md

@@ -91,6 +91,13 @@ Function | Returns
 `pgv_set(package text, name text, value anynonarray, is_transactional bool default false)` | `void`
 `pgv_get(package text, name text, var_type anynonarray, strict bool default true)` | `anynonarray`
 
+## Array variables functions
+
+Function | Returns
+-------- | -------
+`pgv_set(package text, name text, value anyarray, is_transactional bool default false)` | `void`
+`pgv_get(package text, name text, var_type anyarray, strict bool default true)` | `anyarray`
+
 ## **Deprecated** scalar variables functions
 
 ### Integer variables
@@ -181,7 +188,7 @@ Note that **pgv_stats()** works only with the PostgreSQL 9.6 and newer.
 
 ## Examples
 
-It is easy to use functions to work with scalar variables:
+It is easy to use functions to work with scalar and array variables:
 
 ```sql
 SELECT pgv_set('vars', 'int1', 101);
@@ -196,6 +203,13 @@ SELECT SELECT pgv_get('vars', 'text1', NULL::text);
     pgv_get
 ---------------
  text variable
+
+SELECT pgv_set('vars', 'arr1', '{101,102}'::int[]);
+
+SELECT pgv_get('vars', 'arr1', NULL::int[]);
+  pgv_get
+-----------
+ {101,102}
 ```
 
 Let's assume we have a **tab** table:
@@ -246,6 +260,7 @@ You can list packages and variables:
 SELECT * FROM pgv_list() order by package, name;
  package | name  | is_transactional
 ---------+-------+------------------
+ vars    | arr1  | f
  vars    | int1  | f
  vars    | r1    | f
  vars    | text1 | f
@@ -257,7 +272,7 @@ And get used memory in bytes:
 SELECT * FROM pgv_stats() order by package;
  package | allocated_memory
 ---------+------------------
- vars    |            32768
+ vars    |            49152
 ```
 
 You can delete variables or whole packages:

pg_variables--1.1--1.2.sql

@@ -0,0 +1,16 @@
+/* contrib/pg_variables/pg_variables--1.1--1.2.sql */
+
+-- complain if script is sourced in psql, rather than via ALTER EXTENSION
+\echo Use "ALTER EXTENSION pg_variables UPDATE TO '1.2'" to load this file. \quit
+
+-- Functions to work with arrays
+
+CREATE FUNCTION pgv_set(package text, name text, value anyarray, is_transactional bool default false)
+RETURNS void
+AS 'MODULE_PATHNAME', 'variable_set_array'
+LANGUAGE C VOLATILE;
+
+CREATE FUNCTION pgv_get(package text, name text, var_type anyarray, strict bool default true)
+RETURNS anyarray
+AS 'MODULE_PATHNAME', 'variable_get_array'
+LANGUAGE C VOLATILE;

pg_variables.c

@@ -237,6 +237,7 @@ VARIABLE_GET_TEMPLATE(0, 1, 2, jsonb, JSONBOID)
 
 /* current API */
 VARIABLE_GET_TEMPLATE(0, 1, 3, any, get_fn_expr_argtype(fcinfo->flinfo, 2))
+VARIABLE_GET_TEMPLATE(0, 1, 3, array, get_fn_expr_argtype(fcinfo->flinfo, 2))
 
 
 #define VARIABLE_SET_TEMPLATE(type, typid) \
@@ -275,6 +276,7 @@ VARIABLE_SET_TEMPLATE(jsonb, JSONBOID)
 
 /* current API */
 VARIABLE_SET_TEMPLATE(any, get_fn_expr_argtype(fcinfo->flinfo, 2))
+VARIABLE_SET_TEMPLATE(array, get_fn_expr_argtype(fcinfo->flinfo, 2))
 
 
 Datum

pg_variables.control

@@ -1,5 +1,5 @@
 # pg_variables extension
 comment = 'session variables with various types'
-default_version = '1.1'
+default_version = '1.2'
 module_pathname = '$libdir/pg_variables'
 relocatable = true

sql/pg_variables_any.sql

@@ -139,6 +139,26 @@ SELECT pgv_get('vars', 'd1', NULL::jsonb);
 SELECT pgv_set('vars', 'jNULL', NULL::jsonb);
 SELECT pgv_get('vars', 'jNULL', NULL::jsonb);
 
+-- Array variables
+SELECT pgv_set('vars', 'arr1', '{1, 2, null}'::int[]);
+SELECT pgv_set('vars', 'arr2', '{"bar", "balance", "active"}'::text[]);
+SELECT pgv_set('vars2', 'j1', '{1, 2, null}'::int[]);
+
+SELECT pgv_get('vars', 'arr1', NULL::int[]);
+SELECT pgv_get('vars', 'arr2', NULL::int[]);
+SELECT pgv_set('vars', 'arr1', '{"bar", "balance", "active"}'::text[]);
+SELECT pgv_set('vars', 'arr1', '{3, 4, 5}'::int[]);
+SELECT pgv_get('vars', 'arr1', NULL::int[]);
+
+SELECT pgv_get('vars', 'arr3', NULL::int[]);
+SELECT pgv_get('vars', 'arr3', NULL::int[], false);
+SELECT pgv_exists('vars', 'arr3');
+SELECT pgv_exists('vars', 'arr1');
+SELECT pgv_get('vars2', 'j1', NULL::int[]);
+
+SELECT pgv_set('vars', 'arrNULL', NULL::int[]);
+SELECT pgv_get('vars', 'arrNULL', NULL::int[]);
+
 -- Manipulate variables
 SELECT * FROM pgv_list() order by package, name;