feat: add set datatype support for aurora-data-api (#7665)

* adds support for type set to data-api * adds support for type set to data-api * fix spacing
typeorm · May 20, 2021 · b6c1836 · b6c1836
1 parent b9cc9c9
commit b6c1836
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/driver/aurora-data-api/AuroraDataApiDriver.ts b/src/driver/aurora-data-api/AuroraDataApiDriver.ts
@@ -117,6 +117,7 @@ export class AuroraDataApiDriver implements Driver {
         "longblob",
         "longtext",
         "enum",
+        "set",
         "binary",
         "varbinary",
         // json data type
@@ -441,7 +442,7 @@ export class AuroraDataApiDriver implements Driver {
         } else if (columnMetadata.type === "timestamp" || columnMetadata.type === "datetime" || columnMetadata.type === Date) {
             return DateUtils.mixedDateToDate(value);
 
-        } else if (columnMetadata.type === "simple-array") {
+        } else if (columnMetadata.type === "simple-array" || columnMetadata.type === "set") {
             return DateUtils.simpleArrayToString(value);
 
         } else if (columnMetadata.type === "simple-json") {
@@ -480,7 +481,7 @@ export class AuroraDataApiDriver implements Driver {
         } else if (columnMetadata.type === "time") {
             value = DateUtils.mixedTimeToString(value);
 
-        } else if (columnMetadata.type === "simple-array") {
+        } else if (columnMetadata.type === "simple-array" || columnMetadata.type === "set") {
             value = DateUtils.stringToSimpleArray(value);
 
         } else if (columnMetadata.type === "simple-json") {
@@ -561,7 +562,10 @@ export class AuroraDataApiDriver implements Driver {
         if ((columnMetadata.type === "enum" || columnMetadata.type === "simple-enum") && defaultValue !== undefined) {
             return `'${defaultValue}'`;
         }
+        if ((columnMetadata.type === "set") && defaultValue !== undefined) {
+            return `'${DateUtils.simpleArrayToString(defaultValue)}'`;
 
+        }
         if (typeof defaultValue === "number") {
             return "" + defaultValue;
 

diff --git a/src/driver/aurora-data-api/AuroraDataApiQueryRunner.ts b/src/driver/aurora-data-api/AuroraDataApiQueryRunner.ts
@@ -1339,7 +1339,7 @@ export class AuroraDataApiQueryRunner extends BaseQueryRunner implements QueryRu
                             tableColumn.scale = parseInt(dbColumn["NUMERIC_SCALE"]);
                     }
 
-                    if (tableColumn.type === "enum" || tableColumn.type === "simple-enum") {
+                    if (tableColumn.type === "enum" || tableColumn.type === "simple-enum" || tableColumn.type === "set") {
                         const colType = dbColumn["COLUMN_TYPE"];
                         const items = colType.substring(colType.indexOf("(") + 1, colType.lastIndexOf(")")).split(",");
                         tableColumn.enum = (items as string[]).map(item => {