prostgles-server 4.2.159 → 4.2.161
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/Auth/setEmailProvider.js +2 -2
- package/dist/Auth/setEmailProvider.js.map +1 -1
- package/lib/Auth/AuthHandler.ts +436 -0
- package/lib/Auth/AuthTypes.ts +280 -0
- package/lib/Auth/getSafeReturnURL.ts +35 -0
- package/lib/Auth/sendEmail.ts +83 -0
- package/lib/Auth/setAuthProviders.ts +128 -0
- package/lib/Auth/setEmailProvider.ts +85 -0
- package/lib/Auth/setupAuthRoutes.ts +161 -0
- package/lib/DBEventsManager.ts +178 -0
- package/lib/DBSchemaBuilder.ts +225 -0
- package/lib/DboBuilder/DboBuilder.ts +319 -0
- package/lib/DboBuilder/DboBuilderTypes.ts +361 -0
- package/lib/DboBuilder/QueryBuilder/Functions.ts +1153 -0
- package/lib/DboBuilder/QueryBuilder/QueryBuilder.ts +288 -0
- package/lib/DboBuilder/QueryBuilder/getJoinQuery.ts +263 -0
- package/lib/DboBuilder/QueryBuilder/getNewQuery.ts +271 -0
- package/lib/DboBuilder/QueryBuilder/getSelectQuery.ts +136 -0
- package/lib/DboBuilder/QueryBuilder/prepareHaving.ts +22 -0
- package/lib/DboBuilder/QueryStreamer.ts +250 -0
- package/lib/DboBuilder/TableHandler/DataValidator.ts +428 -0
- package/lib/DboBuilder/TableHandler/TableHandler.ts +205 -0
- package/lib/DboBuilder/TableHandler/delete.ts +115 -0
- package/lib/DboBuilder/TableHandler/insert.ts +183 -0
- package/lib/DboBuilder/TableHandler/insertTest.ts +78 -0
- package/lib/DboBuilder/TableHandler/onDeleteFromFileTable.ts +62 -0
- package/lib/DboBuilder/TableHandler/runInsertUpdateQuery.ts +134 -0
- package/lib/DboBuilder/TableHandler/update.ts +126 -0
- package/lib/DboBuilder/TableHandler/updateBatch.ts +49 -0
- package/lib/DboBuilder/TableHandler/updateFile.ts +48 -0
- package/lib/DboBuilder/TableHandler/upsert.ts +34 -0
- package/lib/DboBuilder/ViewHandler/ViewHandler.ts +393 -0
- package/lib/DboBuilder/ViewHandler/count.ts +38 -0
- package/lib/DboBuilder/ViewHandler/find.ts +153 -0
- package/lib/DboBuilder/ViewHandler/getExistsCondition.ts +73 -0
- package/lib/DboBuilder/ViewHandler/getExistsFilters.ts +74 -0
- package/lib/DboBuilder/ViewHandler/getInfo.ts +32 -0
- package/lib/DboBuilder/ViewHandler/getTableJoinQuery.ts +84 -0
- package/lib/DboBuilder/ViewHandler/parseComplexFilter.ts +96 -0
- package/lib/DboBuilder/ViewHandler/parseFieldFilter.ts +105 -0
- package/lib/DboBuilder/ViewHandler/parseJoinPath.ts +208 -0
- package/lib/DboBuilder/ViewHandler/prepareSortItems.ts +163 -0
- package/lib/DboBuilder/ViewHandler/prepareWhere.ts +90 -0
- package/lib/DboBuilder/ViewHandler/size.ts +37 -0
- package/lib/DboBuilder/ViewHandler/subscribe.ts +118 -0
- package/lib/DboBuilder/ViewHandler/validateViewRules.ts +70 -0
- package/lib/DboBuilder/dboBuilderUtils.ts +222 -0
- package/lib/DboBuilder/getColumns.ts +114 -0
- package/lib/DboBuilder/getCondition.ts +201 -0
- package/lib/DboBuilder/getSubscribeRelatedTables.ts +190 -0
- package/lib/DboBuilder/getTablesForSchemaPostgresSQL.ts +426 -0
- package/lib/DboBuilder/insertNestedRecords.ts +355 -0
- package/lib/DboBuilder/parseUpdateRules.ts +187 -0
- package/lib/DboBuilder/prepareShortestJoinPaths.ts +186 -0
- package/lib/DboBuilder/runSQL.ts +182 -0
- package/lib/DboBuilder/runTransaction.ts +50 -0
- package/lib/DboBuilder/sqlErrCodeToMsg.ts +254 -0
- package/lib/DboBuilder/uploadFile.ts +69 -0
- package/lib/Event_Trigger_Tags.ts +118 -0
- package/lib/FileManager/FileManager.ts +358 -0
- package/lib/FileManager/getValidatedFileType.ts +69 -0
- package/lib/FileManager/initFileManager.ts +187 -0
- package/lib/FileManager/upload.ts +62 -0
- package/lib/FileManager/uploadStream.ts +79 -0
- package/lib/Filtering.ts +463 -0
- package/lib/JSONBValidation/validate_jsonb_schema_sql.ts +502 -0
- package/lib/JSONBValidation/validation.ts +143 -0
- package/lib/Logging.ts +127 -0
- package/lib/PostgresNotifListenManager.ts +143 -0
- package/lib/Prostgles.ts +485 -0
- package/lib/ProstglesTypes.ts +196 -0
- package/lib/PubSubManager/PubSubManager.ts +609 -0
- package/lib/PubSubManager/addSub.ts +138 -0
- package/lib/PubSubManager/addSync.ts +141 -0
- package/lib/PubSubManager/getCreatePubSubManagerError.ts +72 -0
- package/lib/PubSubManager/getPubSubManagerInitQuery.ts +662 -0
- package/lib/PubSubManager/initPubSubManager.ts +79 -0
- package/lib/PubSubManager/notifListener.ts +173 -0
- package/lib/PubSubManager/orphanTriggerCheck.ts +70 -0
- package/lib/PubSubManager/pushSubData.ts +55 -0
- package/lib/PublishParser/PublishParser.ts +162 -0
- package/lib/PublishParser/getFileTableRules.ts +124 -0
- package/lib/PublishParser/getSchemaFromPublish.ts +141 -0
- package/lib/PublishParser/getTableRulesWithoutFileTable.ts +177 -0
- package/lib/PublishParser/publishTypesAndUtils.ts +399 -0
- package/lib/RestApi.ts +127 -0
- package/lib/SchemaWatch/SchemaWatch.ts +90 -0
- package/lib/SchemaWatch/createSchemaWatchEventTrigger.ts +3 -0
- package/lib/SchemaWatch/getValidatedWatchSchemaType.ts +45 -0
- package/lib/SchemaWatch/getWatchSchemaTagList.ts +27 -0
- package/lib/SyncReplication.ts +557 -0
- package/lib/TableConfig/TableConfig.ts +468 -0
- package/lib/TableConfig/getColumnDefinitionQuery.ts +111 -0
- package/lib/TableConfig/getConstraintDefinitionQueries.ts +95 -0
- package/lib/TableConfig/getFutureTableSchema.ts +64 -0
- package/lib/TableConfig/getPGIndexes.ts +53 -0
- package/lib/TableConfig/getTableColumnQueries.ts +129 -0
- package/lib/TableConfig/initTableConfig.ts +326 -0
- package/lib/index.ts +13 -0
- package/lib/initProstgles.ts +319 -0
- package/lib/onSocketConnected.ts +102 -0
- package/lib/runClientRequest.ts +129 -0
- package/lib/shortestPath.ts +122 -0
- package/lib/typeTests/DBoGenerated.d.ts +320 -0
- package/lib/typeTests/dboTypeCheck.ts +81 -0
- package/lib/utils.ts +15 -0
- package/package.json +1 -1
|
@@ -0,0 +1,662 @@
|
|
|
1
|
+
|
|
2
|
+
import { tryCatch } from "prostgles-types";
|
|
3
|
+
import { pgp } from "../DboBuilder/DboBuilderTypes";
|
|
4
|
+
import { asValue, NOTIF_CHANNEL, NOTIF_TYPE, PubSubManager } from "./PubSubManager";
|
|
5
|
+
const { version } = require("../../package.json");
|
|
6
|
+
import { getAppCheckQuery } from "./orphanTriggerCheck";
|
|
7
|
+
import { DboBuilder } from "../DboBuilder/DboBuilder";
|
|
8
|
+
|
|
9
|
+
export const DB_OBJ_NAMES = {
|
|
10
|
+
trigger_add_remove_func: "prostgles.trigger_add_remove_func",
|
|
11
|
+
data_watch_func: "prostgles.prostgles_trigger_function",
|
|
12
|
+
schema_watch_func: "prostgles.schema_watch_func",
|
|
13
|
+
schema_watch_trigger: "prostgles_schema_watch_trigger_new",
|
|
14
|
+
schema_watch_trigger_drop: "prostgles_schema_watch_trigger_new_drop"
|
|
15
|
+
} as const;
|
|
16
|
+
|
|
17
|
+
const PROSTGLES_SCHEMA_EXISTS_QUERY = `
|
|
18
|
+
SELECT 1
|
|
19
|
+
FROM information_schema.columns
|
|
20
|
+
WHERE table_schema = 'prostgles'
|
|
21
|
+
AND table_name = 'versions'
|
|
22
|
+
AND column_name = 'schema_md5'
|
|
23
|
+
`;
|
|
24
|
+
const PROSTGLES_SCHEMA_VERSION_OK_QUERY = `
|
|
25
|
+
SELECT 1
|
|
26
|
+
FROM prostgles.versions
|
|
27
|
+
WHERE (string_to_array(version, '.')::int[] > string_to_array(\${version}, '.')::int[])
|
|
28
|
+
OR (string_to_array(version, '.')::int[] = string_to_array(\${version}, '.')::int[])
|
|
29
|
+
AND schema_md5 = \${schema_md5}
|
|
30
|
+
`;
|
|
31
|
+
|
|
32
|
+
const getInitQuery = (debugMode: boolean | undefined, pgVersion: number) => {
|
|
33
|
+
|
|
34
|
+
const canReplaceTriggers = pgVersion >= 140006;
|
|
35
|
+
const createTriggerQuery = canReplaceTriggers ?
|
|
36
|
+
`CREATE OR REPLACE TRIGGER %1$I`:
|
|
37
|
+
`
|
|
38
|
+
DROP TRIGGER IF EXISTS %1$I ON %2$s;
|
|
39
|
+
CREATE TRIGGER %1$I
|
|
40
|
+
`;
|
|
41
|
+
|
|
42
|
+
return `
|
|
43
|
+
BEGIN; -- TRANSACTION ISOLATION LEVEL SERIALIZABLE;
|
|
44
|
+
|
|
45
|
+
--SET TRANSACTION ISOLATION LEVEL SERIALIZABLE;
|
|
46
|
+
|
|
47
|
+
/*
|
|
48
|
+
* ${PubSubManager.EXCLUDE_QUERY_FROM_SCHEMA_WATCH_ID}
|
|
49
|
+
*/
|
|
50
|
+
|
|
51
|
+
DO
|
|
52
|
+
$do$
|
|
53
|
+
BEGIN
|
|
54
|
+
|
|
55
|
+
/* Drop older version. */
|
|
56
|
+
IF EXISTS (SELECT 1 FROM information_schema.schemata WHERE schema_name = 'prostgles') THEN
|
|
57
|
+
|
|
58
|
+
/* The seemingly useless IF nesting is done to prevent pg evaluating the entire condition and throw a 'schema_md5 column does not exist' */
|
|
59
|
+
IF
|
|
60
|
+
/* Backwards compatibility. Cannot check schema version */
|
|
61
|
+
NOT EXISTS(
|
|
62
|
+
${PROSTGLES_SCHEMA_EXISTS_QUERY}
|
|
63
|
+
)
|
|
64
|
+
THEN
|
|
65
|
+
DROP SCHEMA IF EXISTS prostgles CASCADE;
|
|
66
|
+
ELSIF
|
|
67
|
+
/* There is no newer version or same same version but different schema */
|
|
68
|
+
NOT EXISTS (
|
|
69
|
+
${PROSTGLES_SCHEMA_VERSION_OK_QUERY}
|
|
70
|
+
)
|
|
71
|
+
THEN
|
|
72
|
+
DROP SCHEMA IF EXISTS prostgles CASCADE;
|
|
73
|
+
|
|
74
|
+
END IF;
|
|
75
|
+
|
|
76
|
+
END IF;
|
|
77
|
+
|
|
78
|
+
|
|
79
|
+
IF NOT EXISTS (
|
|
80
|
+
SELECT 1
|
|
81
|
+
FROM information_schema.schemata
|
|
82
|
+
WHERE schema_name = 'prostgles'
|
|
83
|
+
)
|
|
84
|
+
THEN
|
|
85
|
+
|
|
86
|
+
CREATE SCHEMA IF NOT EXISTS prostgles;
|
|
87
|
+
COMMENT ON SCHEMA prostgles IS 'Used by prostgles-server to enable data/schema change tracking through subscribe/sync/watchSchema';
|
|
88
|
+
|
|
89
|
+
CREATE TABLE IF NOT EXISTS prostgles.versions(
|
|
90
|
+
version TEXT PRIMARY KEY,
|
|
91
|
+
schema_md5 TEXT NOT NULL,
|
|
92
|
+
added_at TIMESTAMP NOT NULL DEFAULT now()
|
|
93
|
+
);
|
|
94
|
+
COMMENT ON TABLE prostgles.versions IS 'Stores the prostgles schema creation query hash and package version number to identify when a newer schema needs to be re-created';
|
|
95
|
+
|
|
96
|
+
INSERT INTO prostgles.versions(version, schema_md5)
|
|
97
|
+
VALUES(${asValue(version)}, \${schema_md5})
|
|
98
|
+
ON CONFLICT DO NOTHING;
|
|
99
|
+
|
|
100
|
+
|
|
101
|
+
CREATE OR REPLACE FUNCTION prostgles.random_string(length INTEGER DEFAULT 33) RETURNS TEXT AS $$
|
|
102
|
+
DECLARE
|
|
103
|
+
chars TEXT[] := '{0,1,2,3,4,5,6,7,8,9,A,B,C,D,E,F,G,H,I,J,K,L,M,N,O,P,Q,R,S,T,U,V,W,X,Y,Z,a,b,c,d,e,f,g,h,i,j,k,l,m,n,o,p,q,r,s,t,u,v,w,x,y,z}';
|
|
104
|
+
result TEXT := '';
|
|
105
|
+
i INTEGER := 0;
|
|
106
|
+
BEGIN
|
|
107
|
+
|
|
108
|
+
IF length < 0 THEN
|
|
109
|
+
RAISE exception 'Given length cannot be less than 0';
|
|
110
|
+
END IF;
|
|
111
|
+
|
|
112
|
+
FOR i IN 1..length LOOP
|
|
113
|
+
result := result || chars[1+random()*(array_length(chars, 1)-1)];
|
|
114
|
+
END LOOP;
|
|
115
|
+
|
|
116
|
+
RETURN result;
|
|
117
|
+
|
|
118
|
+
END;
|
|
119
|
+
$$ language plpgsql;
|
|
120
|
+
COMMENT ON FUNCTION prostgles.random_string IS 'UUIDs without installing pgcrypto';
|
|
121
|
+
|
|
122
|
+
|
|
123
|
+
CREATE OR REPLACE FUNCTION prostgles.debug(VARIADIC args TEXT[]) RETURNS VOID AS $$
|
|
124
|
+
BEGIN
|
|
125
|
+
|
|
126
|
+
--PERFORM pg_notify('debug', concat_ws(' ', args));
|
|
127
|
+
IF
|
|
128
|
+
NOT EXISTS (
|
|
129
|
+
SELECT 1
|
|
130
|
+
FROM information_schema.tables
|
|
131
|
+
WHERE table_schema = 'prostgles'
|
|
132
|
+
AND table_name = 'debug'
|
|
133
|
+
)
|
|
134
|
+
THEN
|
|
135
|
+
CREATE TABLE IF NOT EXISTS prostgles.debug(m TEXT);
|
|
136
|
+
END IF;
|
|
137
|
+
|
|
138
|
+
INSERT INTO prostgles.debug(m) VALUES(concat_ws(' ', args));
|
|
139
|
+
|
|
140
|
+
END;
|
|
141
|
+
$$ LANGUAGE plpgsql;
|
|
142
|
+
COMMENT ON FUNCTION prostgles.debug IS 'Used for internal debugging';
|
|
143
|
+
|
|
144
|
+
|
|
145
|
+
CREATE OR REPLACE FUNCTION prostgles.user(key text default null)
|
|
146
|
+
RETURNS TEXT AS $$
|
|
147
|
+
DECLARE user_text text;
|
|
148
|
+
DECLARE user_jsonb JSONB = '{}'::JSONB;
|
|
149
|
+
BEGIN
|
|
150
|
+
user_text := current_setting('prostgles.user', true);
|
|
151
|
+
IF length(user_text) > 0 THEN
|
|
152
|
+
user_jsonb := user_text::JSONB;
|
|
153
|
+
END IF;
|
|
154
|
+
|
|
155
|
+
IF length(key) > 0 THEN
|
|
156
|
+
RETURN jsonb_extract_path(user_jsonb, key);
|
|
157
|
+
END IF;
|
|
158
|
+
RETURN user_jsonb;
|
|
159
|
+
END;
|
|
160
|
+
$$ LANGUAGE plpgsql;
|
|
161
|
+
COMMENT ON FUNCTION prostgles."user" IS 'Used for row level security';
|
|
162
|
+
|
|
163
|
+
CREATE OR REPLACE FUNCTION prostgles.user_id()
|
|
164
|
+
RETURNS UUID AS $$
|
|
165
|
+
BEGIN
|
|
166
|
+
RETURN prostgles.user('id')::UUID;
|
|
167
|
+
END;
|
|
168
|
+
$$ LANGUAGE plpgsql;
|
|
169
|
+
COMMENT ON FUNCTION prostgles.user_id IS 'Session user id';
|
|
170
|
+
|
|
171
|
+
CREATE OR REPLACE FUNCTION prostgles.user_type()
|
|
172
|
+
RETURNS TEXT AS $$
|
|
173
|
+
BEGIN
|
|
174
|
+
RETURN prostgles.user('type')::TEXT;
|
|
175
|
+
END;
|
|
176
|
+
$$ LANGUAGE plpgsql;
|
|
177
|
+
COMMENT ON FUNCTION prostgles.user_type IS 'Session user type';
|
|
178
|
+
|
|
179
|
+
CREATE TABLE IF NOT EXISTS prostgles.apps (
|
|
180
|
+
id TEXT PRIMARY KEY DEFAULT prostgles.random_string(),
|
|
181
|
+
added TIMESTAMP DEFAULT NOW(),
|
|
182
|
+
application_name TEXT,
|
|
183
|
+
watching_schema_tag_names _TEXT,
|
|
184
|
+
check_frequency_ms INTEGER NOT NULL
|
|
185
|
+
);
|
|
186
|
+
COMMENT ON TABLE prostgles.apps IS 'Keep track of prostgles server apps connected to db to combine common triggers. Heartbeat used due to no logout triggers in postgres';
|
|
187
|
+
|
|
188
|
+
CREATE TABLE IF NOT EXISTS prostgles.app_triggers (
|
|
189
|
+
app_id TEXT NOT NULL,
|
|
190
|
+
table_name TEXT NOT NULL,
|
|
191
|
+
condition TEXT NOT NULL,
|
|
192
|
+
condition_hash TEXT NOT NULL,
|
|
193
|
+
|
|
194
|
+
/* The view from the root subscription, found in the condition.
|
|
195
|
+
We need this because old_table/new_table data is not reflected in the view inside the AFTER trigger
|
|
196
|
+
*/
|
|
197
|
+
related_view_name TEXT,
|
|
198
|
+
related_view_def TEXT, /* view definition */
|
|
199
|
+
|
|
200
|
+
inserted TIMESTAMP NOT NULL DEFAULT NOW(),
|
|
201
|
+
last_used TIMESTAMP NOT NULL DEFAULT NOW(),
|
|
202
|
+
PRIMARY KEY (app_id, table_name, condition_hash)
|
|
203
|
+
);
|
|
204
|
+
COMMENT ON TABLE prostgles.app_triggers IS 'Tables and conditions that are currently subscribed/synced';
|
|
205
|
+
|
|
206
|
+
|
|
207
|
+
CREATE OR REPLACE VIEW prostgles.v_triggers AS
|
|
208
|
+
SELECT *
|
|
209
|
+
, (ROW_NUMBER() OVER( ORDER BY table_name, condition ))::text AS id
|
|
210
|
+
, ROW_NUMBER() OVER(PARTITION BY app_id, table_name ORDER BY table_name, condition ) - 1 AS c_id
|
|
211
|
+
FROM prostgles.app_triggers;
|
|
212
|
+
COMMENT ON VIEW prostgles.v_triggers IS 'Augment trigger table with natural IDs and per app IDs';
|
|
213
|
+
|
|
214
|
+
|
|
215
|
+
CREATE OR REPLACE FUNCTION ${DB_OBJ_NAMES.data_watch_func}() RETURNS TRIGGER
|
|
216
|
+
AS $$
|
|
217
|
+
|
|
218
|
+
DECLARE t_ids TEXT[];
|
|
219
|
+
DECLARE c_ids INTEGER[];
|
|
220
|
+
DECLARE err_c_ids INTEGER[];
|
|
221
|
+
DECLARE unions TEXT := '';
|
|
222
|
+
DECLARE query TEXT := '';
|
|
223
|
+
DECLARE v_trigger RECORD;
|
|
224
|
+
DECLARE has_errors BOOLEAN := FALSE;
|
|
225
|
+
|
|
226
|
+
DECLARE err_text TEXT;
|
|
227
|
+
DECLARE err_detail TEXT;
|
|
228
|
+
DECLARE err_hint TEXT;
|
|
229
|
+
|
|
230
|
+
DECLARE view_def_query TEXT := '';
|
|
231
|
+
|
|
232
|
+
DECLARE escaped_table TEXT;
|
|
233
|
+
|
|
234
|
+
BEGIN
|
|
235
|
+
|
|
236
|
+
--PERFORM pg_notify('debug', concat_ws(' ', 'TABLE', TG_TABLE_NAME, TG_OP));
|
|
237
|
+
|
|
238
|
+
escaped_table := concat_ws('.', CASE WHEN TG_TABLE_SCHEMA <> CURRENT_SCHEMA THEN format('%I', TG_TABLE_SCHEMA) END, format('%I', TG_TABLE_NAME));
|
|
239
|
+
|
|
240
|
+
SELECT string_agg(
|
|
241
|
+
format(
|
|
242
|
+
$c$
|
|
243
|
+
SELECT CASE WHEN EXISTS(
|
|
244
|
+
SELECT 1
|
|
245
|
+
FROM %s
|
|
246
|
+
WHERE %s
|
|
247
|
+
) THEN %s::text END AS t_ids
|
|
248
|
+
$c$,
|
|
249
|
+
table_name,
|
|
250
|
+
condition,
|
|
251
|
+
id
|
|
252
|
+
),
|
|
253
|
+
E' UNION \n '
|
|
254
|
+
)
|
|
255
|
+
INTO unions
|
|
256
|
+
FROM prostgles.v_triggers
|
|
257
|
+
WHERE table_name = escaped_table;
|
|
258
|
+
|
|
259
|
+
|
|
260
|
+
/* unions = 'old_table union new_table' or any one of the tables */
|
|
261
|
+
IF unions IS NOT NULL THEN
|
|
262
|
+
|
|
263
|
+
SELECT
|
|
264
|
+
format(
|
|
265
|
+
E'WITH %I AS (\n %s \n) ',
|
|
266
|
+
TG_TABLE_NAME,
|
|
267
|
+
concat_ws(
|
|
268
|
+
E' UNION ALL \n ',
|
|
269
|
+
CASE WHEN (TG_OP = 'DELETE' OR TG_OP = 'UPDATE') THEN ' SELECT * FROM old_table ' END,
|
|
270
|
+
CASE WHEN (TG_OP = 'INSERT' OR TG_OP = 'UPDATE') THEN ' SELECT * FROM new_table ' END
|
|
271
|
+
)
|
|
272
|
+
)
|
|
273
|
+
||
|
|
274
|
+
COALESCE((
|
|
275
|
+
SELECT ', ' || string_agg(format(E' %s AS ( \n %s \n ) ', related_view_name, related_view_def), ', ')
|
|
276
|
+
FROM (
|
|
277
|
+
SELECT DISTINCT related_view_name, related_view_def
|
|
278
|
+
FROM prostgles.v_triggers
|
|
279
|
+
WHERE table_name = escaped_table
|
|
280
|
+
AND related_view_name IS NOT NULL
|
|
281
|
+
AND related_view_def IS NOT NULL
|
|
282
|
+
) t
|
|
283
|
+
), '')
|
|
284
|
+
||
|
|
285
|
+
format(
|
|
286
|
+
$c$
|
|
287
|
+
SELECT ARRAY_AGG(DISTINCT t.t_ids)
|
|
288
|
+
FROM (
|
|
289
|
+
%s
|
|
290
|
+
) t
|
|
291
|
+
$c$,
|
|
292
|
+
unions
|
|
293
|
+
)
|
|
294
|
+
INTO query;
|
|
295
|
+
|
|
296
|
+
BEGIN
|
|
297
|
+
EXECUTE query INTO t_ids;
|
|
298
|
+
|
|
299
|
+
--RAISE NOTICE 'trigger fired ok';
|
|
300
|
+
|
|
301
|
+
EXCEPTION WHEN OTHERS THEN
|
|
302
|
+
|
|
303
|
+
has_errors := TRUE;
|
|
304
|
+
|
|
305
|
+
GET STACKED DIAGNOSTICS
|
|
306
|
+
err_text = MESSAGE_TEXT,
|
|
307
|
+
err_detail = PG_EXCEPTION_DETAIL,
|
|
308
|
+
err_hint = PG_EXCEPTION_HINT;
|
|
309
|
+
|
|
310
|
+
END;
|
|
311
|
+
|
|
312
|
+
--RAISE NOTICE 'has_errors: % ', has_errors;
|
|
313
|
+
--RAISE NOTICE 'unions: % , cids: %', unions, c_ids;
|
|
314
|
+
|
|
315
|
+
IF (t_ids IS NOT NULL OR has_errors) THEN
|
|
316
|
+
|
|
317
|
+
FOR v_trigger IN
|
|
318
|
+
SELECT app_id, string_agg(c_id::text, ',') as cids
|
|
319
|
+
FROM prostgles.v_triggers
|
|
320
|
+
WHERE id = ANY(t_ids)
|
|
321
|
+
OR has_errors
|
|
322
|
+
GROUP BY app_id
|
|
323
|
+
LOOP
|
|
324
|
+
|
|
325
|
+
PERFORM pg_notify(
|
|
326
|
+
${asValue(NOTIF_CHANNEL.preffix)} || v_trigger.app_id ,
|
|
327
|
+
LEFT(concat_ws(
|
|
328
|
+
${asValue(PubSubManager.DELIMITER)},
|
|
329
|
+
|
|
330
|
+
${asValue(NOTIF_TYPE.data)},
|
|
331
|
+
COALESCE(escaped_table, 'MISSING'),
|
|
332
|
+
COALESCE(TG_OP, 'MISSING'),
|
|
333
|
+
CASE WHEN has_errors
|
|
334
|
+
THEN concat_ws('; ', 'error', err_text, err_detail, err_hint, 'query: ' || query )
|
|
335
|
+
ELSE COALESCE(v_trigger.cids, '')
|
|
336
|
+
END
|
|
337
|
+
${debugMode? (", COALESCE(current_query(), 'current_query ??'), ' ', query") : ""}
|
|
338
|
+
), 7999/4) -- Some chars are 2bytes -> 'Ω'
|
|
339
|
+
);
|
|
340
|
+
END LOOP;
|
|
341
|
+
|
|
342
|
+
|
|
343
|
+
IF has_errors THEN
|
|
344
|
+
|
|
345
|
+
DELETE FROM prostgles.app_triggers;
|
|
346
|
+
RAISE NOTICE 'trigger dropped due to exception: % % %', err_text, err_detail, err_hint;
|
|
347
|
+
|
|
348
|
+
END IF;
|
|
349
|
+
|
|
350
|
+
END IF;
|
|
351
|
+
END IF;
|
|
352
|
+
|
|
353
|
+
${getAppCheckQuery()}
|
|
354
|
+
|
|
355
|
+
RETURN NULL;
|
|
356
|
+
|
|
357
|
+
END;
|
|
358
|
+
|
|
359
|
+
$$ LANGUAGE plpgsql;
|
|
360
|
+
COMMENT ON FUNCTION ${DB_OBJ_NAMES.data_watch_func} IS 'Prostgles internal function used to notify when data in the table changed';
|
|
361
|
+
|
|
362
|
+
CREATE OR REPLACE FUNCTION ${DB_OBJ_NAMES.trigger_add_remove_func}() RETURNS TRIGGER
|
|
363
|
+
AS $$
|
|
364
|
+
|
|
365
|
+
DECLARE operations TEXT[] := ARRAY['insert', 'update', 'delete'];
|
|
366
|
+
DECLARE op TEXT;
|
|
367
|
+
DECLARE query TEXT;
|
|
368
|
+
DECLARE trg_name TEXT;
|
|
369
|
+
DECLARE trw RECORD;
|
|
370
|
+
DECLARE app RECORD;
|
|
371
|
+
DECLARE start_time BIGINT;
|
|
372
|
+
DECLARE changed_triggers_count integer;
|
|
373
|
+
|
|
374
|
+
BEGIN
|
|
375
|
+
|
|
376
|
+
start_time := EXTRACT(EPOCH FROM now()) * 1000;
|
|
377
|
+
|
|
378
|
+
--RAISE NOTICE 'prostgles.app_triggers % ', TG_OP;
|
|
379
|
+
|
|
380
|
+
/* If no other listeners (app_triggers) left on table then DISABLE actual table data watch triggers */
|
|
381
|
+
IF TG_OP = 'DELETE' THEN
|
|
382
|
+
|
|
383
|
+
--RAISE NOTICE 'DELETE trigger_add_remove_func table: % ', ' ' || COALESCE((SELECT concat_ws(' ', string_agg(table_name, ' & '), count(*), min(inserted) ) FROM prostgles.app_triggers) , ' 0 ');
|
|
384
|
+
--RAISE NOTICE 'DELETE trigger_add_remove_func old_table: % ', '' || COALESCE((SELECT concat_ws(' ', string_agg(table_name, ' & '), count(*), min(inserted) ) FROM old_table), ' 0 ');
|
|
385
|
+
|
|
386
|
+
SELECT count(*)
|
|
387
|
+
FROM old_table
|
|
388
|
+
INTO changed_triggers_count;
|
|
389
|
+
|
|
390
|
+
/* Disable actual triggers if needed */
|
|
391
|
+
FOR trw IN
|
|
392
|
+
SELECT DISTINCT table_name
|
|
393
|
+
FROM old_table ot
|
|
394
|
+
WHERE NOT EXISTS (
|
|
395
|
+
SELECT 1
|
|
396
|
+
FROM prostgles.app_triggers t
|
|
397
|
+
WHERE t.table_name = ot.table_name
|
|
398
|
+
)
|
|
399
|
+
AND EXISTS (
|
|
400
|
+
SELECT trigger_name
|
|
401
|
+
FROM information_schema.triggers
|
|
402
|
+
WHERE trigger_name IN (
|
|
403
|
+
concat_ws('_', 'prostgles_triggers', table_name, 'insert'),
|
|
404
|
+
concat_ws('_', 'prostgles_triggers', table_name, 'update'),
|
|
405
|
+
concat_ws('_', 'prostgles_triggers', table_name, 'delete')
|
|
406
|
+
)
|
|
407
|
+
)
|
|
408
|
+
LOOP
|
|
409
|
+
|
|
410
|
+
FOREACH op IN ARRAY operations
|
|
411
|
+
LOOP
|
|
412
|
+
trg_name := concat_ws('_', 'prostgles_triggers', trw.table_name, op);
|
|
413
|
+
|
|
414
|
+
EXECUTE format(' ALTER TABLE %s DISABLE TRIGGER %I ;', trw.table_name, trg_name);
|
|
415
|
+
END LOOP;
|
|
416
|
+
|
|
417
|
+
END LOOP;
|
|
418
|
+
|
|
419
|
+
/* If newly added listeners on table then CREATE table data watch triggers */
|
|
420
|
+
ELSIF TG_OP = 'INSERT' THEN
|
|
421
|
+
|
|
422
|
+
SELECT count(*)
|
|
423
|
+
FROM new_table
|
|
424
|
+
INTO changed_triggers_count;
|
|
425
|
+
|
|
426
|
+
/* Loop through newly added tables to add data watch triggers */
|
|
427
|
+
FOR trw IN
|
|
428
|
+
|
|
429
|
+
SELECT DISTINCT table_name
|
|
430
|
+
FROM new_table nt
|
|
431
|
+
|
|
432
|
+
/* Table did not exist prior to this insert */
|
|
433
|
+
WHERE NOT EXISTS (
|
|
434
|
+
SELECT 1
|
|
435
|
+
FROM prostgles.app_triggers t
|
|
436
|
+
WHERE t.table_name = nt.table_name
|
|
437
|
+
AND t.inserted < nt.inserted -- exclude current record (this is an after trigger). Turn into before trigger?
|
|
438
|
+
)
|
|
439
|
+
|
|
440
|
+
/* Table is valid
|
|
441
|
+
AND EXISTS (
|
|
442
|
+
SELECT 1
|
|
443
|
+
FROM information_schema.tables
|
|
444
|
+
WHERE table_schema = 'public'
|
|
445
|
+
AND table_name = nt.table_name
|
|
446
|
+
)
|
|
447
|
+
*/
|
|
448
|
+
LOOP
|
|
449
|
+
|
|
450
|
+
IF (
|
|
451
|
+
SELECT COUNT(*)
|
|
452
|
+
FROM information_schema.triggers
|
|
453
|
+
WHERE trigger_name IN (
|
|
454
|
+
'prostgles_triggers_' || trw.table_name || '_insert',
|
|
455
|
+
'prostgles_triggers_' || trw.table_name || '_update',
|
|
456
|
+
'prostgles_triggers_' || trw.table_name || '_delete'
|
|
457
|
+
)
|
|
458
|
+
) = 3
|
|
459
|
+
THEN
|
|
460
|
+
query := concat_ws(E'\n',
|
|
461
|
+
format(' ALTER TABLE %s ENABLE TRIGGER %I ;', trw.table_name, 'prostgles_triggers_' || trw.table_name || '_insert'),
|
|
462
|
+
format(' ALTER TABLE %s ENABLE TRIGGER %I ;', trw.table_name, 'prostgles_triggers_' || trw.table_name || '_update'),
|
|
463
|
+
format(' ALTER TABLE %s ENABLE TRIGGER %I ;', trw.table_name, 'prostgles_triggers_' || trw.table_name || '_delete')
|
|
464
|
+
);
|
|
465
|
+
ELSE
|
|
466
|
+
|
|
467
|
+
query := format(
|
|
468
|
+
$q$
|
|
469
|
+
${createTriggerQuery}
|
|
470
|
+
AFTER INSERT ON %2$s
|
|
471
|
+
REFERENCING NEW TABLE AS new_table
|
|
472
|
+
FOR EACH STATEMENT EXECUTE PROCEDURE ${DB_OBJ_NAMES.data_watch_func}();
|
|
473
|
+
/* removed to allow less privileges for a user to create subscriptions
|
|
474
|
+
COMMENT ON TRIGGER %1$I ON %2$s IS 'Prostgles internal trigger used to notify when data in the table changed';
|
|
475
|
+
*/
|
|
476
|
+
$q$,
|
|
477
|
+
'prostgles_triggers_' || trw.table_name || '_insert', trw.table_name
|
|
478
|
+
) || format(
|
|
479
|
+
$q$
|
|
480
|
+
${createTriggerQuery}
|
|
481
|
+
AFTER UPDATE ON %2$s
|
|
482
|
+
REFERENCING OLD TABLE AS old_table NEW TABLE AS new_table
|
|
483
|
+
FOR EACH STATEMENT EXECUTE PROCEDURE ${DB_OBJ_NAMES.data_watch_func}();
|
|
484
|
+
--COMMENT ON TRIGGER %1$I ON %2$s IS 'Prostgles internal trigger used to notify when data in the table changed';
|
|
485
|
+
$q$,
|
|
486
|
+
'prostgles_triggers_' || trw.table_name || '_update', trw.table_name
|
|
487
|
+
) || format(
|
|
488
|
+
$q$
|
|
489
|
+
${createTriggerQuery}
|
|
490
|
+
AFTER DELETE ON %2$s
|
|
491
|
+
REFERENCING OLD TABLE AS old_table
|
|
492
|
+
FOR EACH STATEMENT EXECUTE PROCEDURE ${DB_OBJ_NAMES.data_watch_func}();
|
|
493
|
+
--COMMENT ON TRIGGER %1$I ON %2$s IS 'Prostgles internal trigger used to notify when data in the table changed';
|
|
494
|
+
$q$,
|
|
495
|
+
'prostgles_triggers_' || trw.table_name || '_delete', trw.table_name
|
|
496
|
+
);
|
|
497
|
+
END IF;
|
|
498
|
+
|
|
499
|
+
|
|
500
|
+
--RAISE NOTICE ' % ', query;
|
|
501
|
+
|
|
502
|
+
|
|
503
|
+
query := format(
|
|
504
|
+
$q$
|
|
505
|
+
DO $e$
|
|
506
|
+
BEGIN
|
|
507
|
+
/* ${ PubSubManager.EXCLUDE_QUERY_FROM_SCHEMA_WATCH_ID} */
|
|
508
|
+
%s
|
|
509
|
+
|
|
510
|
+
END $e$;
|
|
511
|
+
$q$,
|
|
512
|
+
query
|
|
513
|
+
) ;
|
|
514
|
+
|
|
515
|
+
|
|
516
|
+
EXECUTE query;
|
|
517
|
+
|
|
518
|
+
END LOOP;
|
|
519
|
+
|
|
520
|
+
END IF;
|
|
521
|
+
|
|
522
|
+
/** Notify all apps about trigger table change */
|
|
523
|
+
IF changed_triggers_count > 0 THEN
|
|
524
|
+
FOR app IN
|
|
525
|
+
SELECT * FROM prostgles.apps
|
|
526
|
+
LOOP
|
|
527
|
+
PERFORM pg_notify(
|
|
528
|
+
${asValue(NOTIF_CHANNEL.preffix)} || app.id,
|
|
529
|
+
LEFT(concat_ws(
|
|
530
|
+
${asValue(PubSubManager.DELIMITER)},
|
|
531
|
+
${asValue(NOTIF_TYPE.data_trigger_change)},
|
|
532
|
+
json_build_object(
|
|
533
|
+
'TG_OP', TG_OP,
|
|
534
|
+
'duration', (EXTRACT(EPOCH FROM now()) * 1000) - start_time,
|
|
535
|
+
'query', ${debugMode? 'LEFT(current_query(), 400)' : "'Only shown in debug mode'"}
|
|
536
|
+
)
|
|
537
|
+
)::TEXT, 7999/4)
|
|
538
|
+
);
|
|
539
|
+
END LOOP;
|
|
540
|
+
END IF;
|
|
541
|
+
|
|
542
|
+
RETURN NULL;
|
|
543
|
+
END;
|
|
544
|
+
|
|
545
|
+
$$ LANGUAGE plpgsql;
|
|
546
|
+
COMMENT ON FUNCTION ${DB_OBJ_NAMES.trigger_add_remove_func} IS 'Used to add/remove table watch triggers concurrently ';
|
|
547
|
+
|
|
548
|
+
DROP TRIGGER IF EXISTS prostgles_triggers_insert ON prostgles.app_triggers;
|
|
549
|
+
CREATE TRIGGER prostgles_triggers_insert
|
|
550
|
+
AFTER INSERT ON prostgles.app_triggers
|
|
551
|
+
REFERENCING NEW TABLE AS new_table
|
|
552
|
+
FOR EACH STATEMENT EXECUTE PROCEDURE ${DB_OBJ_NAMES.trigger_add_remove_func}();
|
|
553
|
+
|
|
554
|
+
DROP TRIGGER IF EXISTS prostgles_triggers_delete ON prostgles.app_triggers;
|
|
555
|
+
CREATE TRIGGER prostgles_triggers_delete
|
|
556
|
+
AFTER DELETE ON prostgles.app_triggers
|
|
557
|
+
REFERENCING OLD TABLE AS old_table
|
|
558
|
+
FOR EACH STATEMENT EXECUTE PROCEDURE ${DB_OBJ_NAMES.trigger_add_remove_func}();
|
|
559
|
+
|
|
560
|
+
|
|
561
|
+
CREATE OR REPLACE FUNCTION ${DB_OBJ_NAMES.schema_watch_func}() RETURNS event_trigger AS $$
|
|
562
|
+
|
|
563
|
+
DECLARE curr_query TEXT := '';
|
|
564
|
+
DECLARE app RECORD;
|
|
565
|
+
DECLARE objects_changed BOOLEAN := false;
|
|
566
|
+
|
|
567
|
+
BEGIN
|
|
568
|
+
|
|
569
|
+
IF TG_event = 'ddl_command_end' THEN
|
|
570
|
+
objects_changed := EXISTS (
|
|
571
|
+
SELECT *
|
|
572
|
+
FROM pg_event_trigger_ddl_commands()
|
|
573
|
+
);
|
|
574
|
+
END IF;
|
|
575
|
+
IF TG_event = 'sql_drop' THEN
|
|
576
|
+
objects_changed := EXISTS (
|
|
577
|
+
SELECT *
|
|
578
|
+
FROM pg_event_trigger_dropped_objects()
|
|
579
|
+
);
|
|
580
|
+
END IF;
|
|
581
|
+
|
|
582
|
+
/*
|
|
583
|
+
This event trigger will outlive a prostgles app instance.
|
|
584
|
+
Must ensure it only fires if an app instance is running
|
|
585
|
+
*/
|
|
586
|
+
IF
|
|
587
|
+
objects_changed
|
|
588
|
+
AND EXISTS (
|
|
589
|
+
SELECT 1
|
|
590
|
+
FROM information_schema.tables
|
|
591
|
+
WHERE table_schema = 'prostgles'
|
|
592
|
+
AND table_name = 'apps'
|
|
593
|
+
)
|
|
594
|
+
THEN
|
|
595
|
+
|
|
596
|
+
SELECT LEFT(COALESCE(current_query(), ''), 5000)
|
|
597
|
+
INTO curr_query;
|
|
598
|
+
|
|
599
|
+
FOR app IN
|
|
600
|
+
SELECT *
|
|
601
|
+
FROM prostgles.apps
|
|
602
|
+
WHERE tg_tag = ANY(watching_schema_tag_names)
|
|
603
|
+
AND curr_query NOT ILIKE '%${PubSubManager.EXCLUDE_QUERY_FROM_SCHEMA_WATCH_ID}%'
|
|
604
|
+
LOOP
|
|
605
|
+
PERFORM pg_notify(
|
|
606
|
+
${asValue(NOTIF_CHANNEL.preffix)} || app.id,
|
|
607
|
+
LEFT(concat_ws(
|
|
608
|
+
${asValue(PubSubManager.DELIMITER)},
|
|
609
|
+
${asValue(NOTIF_TYPE.schema)},
|
|
610
|
+
tg_tag ,
|
|
611
|
+
TG_event,
|
|
612
|
+
${debugMode? 'curr_query' : "'Only shown in debug mode'"}
|
|
613
|
+
), 7999/4)
|
|
614
|
+
);
|
|
615
|
+
END LOOP;
|
|
616
|
+
|
|
617
|
+
${getAppCheckQuery()}
|
|
618
|
+
|
|
619
|
+
END IF;
|
|
620
|
+
|
|
621
|
+
END;
|
|
622
|
+
$$ LANGUAGE plpgsql;
|
|
623
|
+
COMMENT ON FUNCTION ${DB_OBJ_NAMES.schema_watch_func} IS 'Prostgles internal function used to notify when schema has changed';
|
|
624
|
+
|
|
625
|
+
END IF;
|
|
626
|
+
|
|
627
|
+
END
|
|
628
|
+
$do$;
|
|
629
|
+
|
|
630
|
+
COMMIT;
|
|
631
|
+
`;
|
|
632
|
+
|
|
633
|
+
}
|
|
634
|
+
|
|
635
|
+
/**
|
|
636
|
+
* Initialize the prostgles schema and functions needed for realtime data and schema changes
|
|
637
|
+
* undefined returned if the database contains the apropriate prostgles schema
|
|
638
|
+
*/
|
|
639
|
+
export const getPubSubManagerInitQuery = async function(this: DboBuilder): Promise<string | undefined> {
|
|
640
|
+
|
|
641
|
+
const versionNum = await this.db.one("SELECT current_setting('server_version_num')::int as val");
|
|
642
|
+
const initQuery = getInitQuery(this.prostgles.opts.DEBUG_MODE, versionNum.val);
|
|
643
|
+
const { schema_md5 = "none" } = await this.db.oneOrNone("SELECT md5($1) as schema_md5", [initQuery.trim()]);
|
|
644
|
+
const query = pgp.as.format(initQuery, { schema_md5, version });
|
|
645
|
+
const existingSchema = await this.db.any(PROSTGLES_SCHEMA_EXISTS_QUERY);
|
|
646
|
+
if(!existingSchema.length){
|
|
647
|
+
console.log("getPubSubManagerInitQuery: No prostgles.versions table found. Creating...");
|
|
648
|
+
return query;
|
|
649
|
+
}
|
|
650
|
+
const { existingSchemaVersions } = await tryCatch(async () => {
|
|
651
|
+
const existingSchemaVersions = await this.db.any(PROSTGLES_SCHEMA_VERSION_OK_QUERY, { schema_md5, version });
|
|
652
|
+
return {
|
|
653
|
+
existingSchemaVersions
|
|
654
|
+
}
|
|
655
|
+
});
|
|
656
|
+
if(!existingSchemaVersions?.length){
|
|
657
|
+
console.log("getPubSubManagerInitQuery: Outdated prostgles schema. Re-creating...");
|
|
658
|
+
return query;
|
|
659
|
+
}
|
|
660
|
+
|
|
661
|
+
return undefined;
|
|
662
|
+
}
|