Merge pull request #61 from hackolade/HCK-4036--v6-11-3-alpha-1-2-add…

…ed-support-for-if-not-exi FE: added support of IF NOT EXISTS for CREATE USER statements on 23c
hackolade · Sep 29, 2023 · a2310a6 · a2310a6
2 parents dae6dd4 + e75fbe9
commit a2310a6
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/forward_engineering/configs/templates.js b/forward_engineering/configs/templates.js
@@ -1,5 +1,5 @@
 module.exports = {
-	createSchema: 'CREATE USER ${schemaName} NO AUTHENTICATION',
+	createSchema: 'CREATE USER${ifNotExists} ${schemaName} NO AUTHENTICATION',
 
     comment: '\nCOMMENT ON ${object} ${objectName} IS ${comment};\n',
 

diff --git a/forward_engineering/ddlProvider.js b/forward_engineering/ddlProvider.js
@@ -134,15 +134,22 @@ module.exports = (baseProvider, options, app) => {
             };
         },
 
-        createSchema({schemaName, ifNotExist}) {
-            const schemaStatement = wrapIfNotExists(
-                assignTemplates(templates.createSchema, {
-                    schemaName: wrapInQuotes(schemaName),
-                }),
+        createSchema({schemaName, ifNotExist, dbVersion}) {
+            const usingTryCatchWrapper = shouldUseTryCatchIfNotExistsWrapper(dbVersion);
+            const schemaStatement = assignTemplates(templates.createSchema, {
+                schemaName: wrapInQuotes(schemaName),
+                ifNotExists: !usingTryCatchWrapper && ifNotExist ? ' IF NOT EXISTS' : '',
+            });
+
+            if (!usingTryCatchWrapper) {
+                return schemaStatement + ';';
+            }
+
+            return wrapIfNotExists(
+                schemaStatement,
                 ifNotExist,
                 1920,
             );
-            return schemaStatement;
         },
 
         hydrateColumn({columnDefinition, jsonSchema, schemaData, definitionJsonSchema = {}}) {