typeorm 0.3.17 → 0.3.18-dev.122c897

package/browser/driver/sap/SapQueryRunner.js

@@ -194,7 +194,19 @@ export class SapQueryRunner extends BaseQueryRunner {
      * Returns raw data stream.
      */
     async stream(query, parameters, onEnd, onError) {
-        throw new TypeORMError(`Stream is not supported by SAP driver.`);
+        if (this.isReleased)
+            throw new QueryRunnerAlreadyReleasedError();
+        const databaseConnection = await this.connect();
+        this.driver.connection.logger.logQuery(query, parameters, this);
+        const prepareAsync = promisify(databaseConnection.prepare).bind(databaseConnection);
+        const statement = await prepareAsync(query);
+        const resultSet = statement.executeQuery(parameters);
+        const stream = this.driver.streamClient.createObjectStream(resultSet);
+        if (onEnd)
+            stream.on("end", onEnd);
+        if (onError)
+            stream.on("error", onError);
+        return stream;
     }
     /**
      * Returns all available database names including system databases.
@@ -262,7 +274,7 @@ export class SapQueryRunner extends BaseQueryRunner {
         if (!parsedTableName.schema) {
             parsedTableName.schema = await this.getCurrentSchema();
         }
-        const sql = `SELECT * FROM "SYS"."TABLE_COLUMNS" WHERE "SCHEMA_NAME" = ${parsedTableName.schema} AND "TABLE_NAME" = ${parsedTableName.tableName} AND "COLUMN_NAME" = '${columnName}'`;
+        const sql = `SELECT * FROM "SYS"."TABLE_COLUMNS" WHERE "SCHEMA_NAME" = '${parsedTableName.schema}' AND "TABLE_NAME" = '${parsedTableName.tableName}' AND "COLUMN_NAME" = '${columnName}'`;
         const result = await this.query(sql);
         return result.length ? true : false;
     }
@@ -744,9 +756,15 @@ export class SapQueryRunner extends BaseQueryRunner {
                 clonedTable.columns[clonedTable.columns.indexOf(oldTableColumn)].name = newColumn.name;
                 oldColumn.name = newColumn.name;
             }
-            if (this.isColumnChanged(oldColumn, newColumn)) {
-                upQueries.push(new Query(`ALTER TABLE ${this.escapePath(table)} ALTER (${this.buildCreateColumnSql(newColumn)})`));
-                downQueries.push(new Query(`ALTER TABLE ${this.escapePath(table)} ALTER (${this.buildCreateColumnSql(oldColumn)})`));
+            if (this.isColumnChanged(oldColumn, newColumn, true)) {
+                upQueries.push(new Query(`ALTER TABLE ${this.escapePath(table)} ALTER (${this.buildCreateColumnSql(newColumn, !(oldColumn.default === null ||
+                    oldColumn.default === undefined), !oldColumn.isNullable)})`));
+                downQueries.push(new Query(`ALTER TABLE ${this.escapePath(table)} ALTER (${this.buildCreateColumnSql(oldColumn, !(newColumn.default === null ||
+                    newColumn.default === undefined), !newColumn.isNullable)})`));
+            }
+            else if (oldColumn.comment !== newColumn.comment) {
+                upQueries.push(new Query(`COMMENT ON COLUMN ${this.escapePath(table)}."${oldColumn.name}" IS ${this.escapeComment(newColumn.comment)}`));
+                downQueries.push(new Query(`COMMENT ON COLUMN ${this.escapePath(table)}."${newColumn.name}" IS ${this.escapeComment(oldColumn.comment)}`));
             }
             if (newColumn.isPrimary !== oldColumn.isPrimary) {
                 const primaryColumns = clonedTable.primaryColumns;
@@ -818,24 +836,6 @@ export class SapQueryRunner extends BaseQueryRunner {
                     downQueries.push(this.createIndexSql(table, uniqueIndex));
                 }
             }
-            if (newColumn.default !== oldColumn.default) {
-                if (newColumn.default !== null &&
-                    newColumn.default !== undefined) {
-                    upQueries.push(new Query(`ALTER TABLE ${this.escapePath(table)} ALTER ("${newColumn.name}" ${this.connection.driver.createFullType(newColumn)} DEFAULT ${newColumn.default})`));
-                    if (oldColumn.default !== null &&
-                        oldColumn.default !== undefined) {
-                        downQueries.push(new Query(`ALTER TABLE ${this.escapePath(table)} ALTER ("${oldColumn.name}" ${this.connection.driver.createFullType(oldColumn)} DEFAULT ${oldColumn.default})`));
-                    }
-                    else {
-                        downQueries.push(new Query(`ALTER TABLE ${this.escapePath(table)} ALTER ("${oldColumn.name}" ${this.connection.driver.createFullType(oldColumn)} DEFAULT NULL)`));
-                    }
-                }
-                else if (oldColumn.default !== null &&
-                    oldColumn.default !== undefined) {
-                    upQueries.push(new Query(`ALTER TABLE ${this.escapePath(table)} ALTER ("${newColumn.name}" ${this.connection.driver.createFullType(newColumn)} DEFAULT NULL)`));
-                    downQueries.push(new Query(`ALTER TABLE ${this.escapePath(table)} ALTER ("${oldColumn.name}" ${this.connection.driver.createFullType(oldColumn)} DEFAULT ${oldColumn.default})`));
-                }
-            }
             await this.executeQueries(upQueries, downQueries);
             this.replaceCachedTable(table, clonedTable);
         }
@@ -1620,7 +1620,9 @@ export class SapQueryRunner extends BaseQueryRunner {
                             dbColumn["DEFAULT_VALUE"];
                     }
                 }
-                tableColumn.comment = ""; // dbColumn["COLUMN_COMMENT"];
+                if (dbColumn["COMMENTS"]) {
+                    tableColumn.comment = dbColumn["COMMENTS"];
+                }
                 if (dbColumn["character_set_name"])
                     tableColumn.charset =
                         dbColumn["character_set_name"];
@@ -1954,6 +1956,16 @@ export class SapQueryRunner extends BaseQueryRunner {
             : foreignKeyOrName;
         return new Query(`ALTER TABLE ${this.escapePath(tableOrName)} DROP CONSTRAINT "${foreignKeyName}"`);
     }
+    /**
+     * Escapes a given comment so it's safe to include in a query.
+     */
+    escapeComment(comment) {
+        if (!comment) {
+            return "NULL";
+        }
+        comment = comment.replace(/'/g, "''").replace(/\u0000/g, ""); // Null bytes aren't allowed in comments
+        return `'${comment}'`;
+    }
     /**
      * Escapes given table or view path.
      */
@@ -1964,49 +1976,35 @@ export class SapQueryRunner extends BaseQueryRunner {
         }
         return `"${tableName}"`;
     }
-    /**
-     * Concat database name and schema name to the foreign key name.
-     * Needs because FK name is relevant to the schema and database.
-     */
-    buildForeignKeyName(fkName, schemaName, dbName) {
-        let joinedFkName = fkName;
-        if (schemaName)
-            joinedFkName = schemaName + "." + joinedFkName;
-        if (dbName)
-            joinedFkName = dbName + "." + joinedFkName;
-        return joinedFkName;
-    }
-    /**
-     * Removes parenthesis around default value.
-     * Sql server returns default value with parenthesis around, e.g.
-     *  ('My text') - for string
-     *  ((1)) - for number
-     *  (newsequentialId()) - for function
-     */
-    removeParenthesisFromDefault(defaultValue) {
-        if (defaultValue.substr(0, 1) !== "(")
-            return defaultValue;
-        const normalizedDefault = defaultValue.substr(1, defaultValue.lastIndexOf(")") - 1);
-        return this.removeParenthesisFromDefault(normalizedDefault);
-    }
     /**
      * Builds a query for create column.
      */
-    buildCreateColumnSql(column) {
+    buildCreateColumnSql(column, explicitDefault, explicitNullable) {
         let c = `"${column.name}" ` + this.connection.driver.createFullType(column);
         if (column.charset)
             c += " CHARACTER SET " + column.charset;
         if (column.collation)
             c += " COLLATE " + column.collation;
-        if (column.default !== undefined && column.default !== null)
-            // DEFAULT must be placed before NOT NULL
+        if (column.default !== undefined && column.default !== null) {
             c += " DEFAULT " + column.default;
-        if (column.isNullable !== true && !column.isGenerated)
+        }
+        else if (explicitDefault) {
+            c += " DEFAULT NULL";
+        }
+        if (!column.isGenerated) {
             // NOT NULL is not supported with GENERATED
-            c += " NOT NULL";
+            if (column.isNullable !== true)
+                c += " NOT NULL";
+            else if (explicitNullable)
+                c += " NULL";
+        }
         if (column.isGenerated === true &&
-            column.generationStrategy === "increment")
+            column.generationStrategy === "increment") {
             c += " GENERATED ALWAYS AS IDENTITY";
+        }
+        if (column.comment) {
+            c += ` COMMENT ${this.escapeComment(column.comment)}`;
+        }
         return c;
     }
 }