|
| 1 | +#!/usr/bin/env node |
| 2 | + |
| 3 | +// https://stackoverflow.com/questions/54898994/bulkupdate-in-sequelize-orm/69044138#69044138 |
| 4 | + |
| 5 | +const assert = require('assert'); |
| 6 | +const path = require('path'); |
| 7 | + |
| 8 | +const { Sequelize, DataTypes, Op } = require('sequelize'); |
| 9 | + |
| 10 | +const sequelize = new Sequelize({ |
| 11 | + dialect: 'sqlite', |
| 12 | + storage: 'tmp.' + path.basename(__filename) + '.sqlite', |
| 13 | +}); |
| 14 | + |
| 15 | +(async () => { |
| 16 | +const Inverses = sequelize.define('Inverses', |
| 17 | + { |
| 18 | + value: { |
| 19 | + type: DataTypes.INTEGER, |
| 20 | + primaryKey: true, |
| 21 | + }, |
| 22 | + inverse: { |
| 23 | + type: DataTypes.INTEGER, |
| 24 | + }, |
| 25 | + name: { |
| 26 | + type: DataTypes.STRING, |
| 27 | + }, |
| 28 | + }, |
| 29 | + { timestamps: false } |
| 30 | +); |
| 31 | +await Inverses.sync({force: true}) |
| 32 | +await Inverses.create({value: 2, inverse: -2, name: 'two'}); |
| 33 | +await Inverses.create({value: 3, inverse: -3, name: 'three'}); |
| 34 | +await Inverses.create({value: 5, inverse: -5, name: 'five'}); |
| 35 | + |
| 36 | +// Initial state. |
| 37 | +assert.strictEqual((await Inverses.findOne({ where: { value: 2 } })).inverse, -2); |
| 38 | +assert.strictEqual((await Inverses.findOne({ where: { value: 3 } })).inverse, -3); |
| 39 | +assert.strictEqual((await Inverses.findOne({ where: { value: 5 } })).inverse, -5); |
| 40 | +assert.strictEqual(await Inverses.count(), 3); |
| 41 | + |
| 42 | +// Update to fixed value. |
| 43 | +await Inverses.update( |
| 44 | + { inverse: 0, }, |
| 45 | + { where: { value: { [Op.gt]: 2 } } }, |
| 46 | +); |
| 47 | +assert.strictEqual((await Inverses.findOne({ where: { value: 2 } })).inverse, -2); |
| 48 | +assert.strictEqual((await Inverses.findOne({ where: { value: 3 } })).inverse, 0); |
| 49 | +assert.strictEqual((await Inverses.findOne({ where: { value: 5 } })).inverse, 0); |
| 50 | +assert.strictEqual(await Inverses.count(), 3); |
| 51 | + |
| 52 | +// Update to match another column. |
| 53 | +await Inverses.update( |
| 54 | + { inverse: sequelize.col('value'), }, |
| 55 | + { where: { value: { [Op.gt]: 2 } } }, |
| 56 | +); |
| 57 | +assert.strictEqual((await Inverses.findOne({ where: { value: 2 } })).inverse, -2); |
| 58 | +assert.strictEqual((await Inverses.findOne({ where: { value: 3 } })).inverse, 3); |
| 59 | +assert.strictEqual((await Inverses.findOne({ where: { value: 5 } })).inverse, 5); |
| 60 | +assert.strictEqual(await Inverses.count(), 3); |
| 61 | + |
| 62 | +// Update to match another column with modification. |
| 63 | +await Inverses.update( |
| 64 | + { inverse: sequelize.fn('1 + ', sequelize.col('value')), }, |
| 65 | + { where: { value: { [Op.gt]: 2 } } }, |
| 66 | +); |
| 67 | +assert.strictEqual((await Inverses.findOne({ where: { value: 2 } })).inverse, -2); |
| 68 | +assert.strictEqual((await Inverses.findOne({ where: { value: 3 } })).inverse, 4); |
| 69 | +assert.strictEqual((await Inverses.findOne({ where: { value: 5 } })).inverse, 6); |
| 70 | +assert.strictEqual(await Inverses.count(), 3); |
| 71 | + |
| 72 | +// A string function test. |
| 73 | +await Inverses.update( |
| 74 | + { name: sequelize.fn('upper', sequelize.col('name')), }, |
| 75 | + { where: { value: { [Op.gt]: 2 } } }, |
| 76 | +); |
| 77 | +assert.strictEqual((await Inverses.findOne({ where: { value: 2 } })).name, 'two'); |
| 78 | +assert.strictEqual((await Inverses.findOne({ where: { value: 3 } })).name, 'THREE'); |
| 79 | +assert.strictEqual((await Inverses.findOne({ where: { value: 5 } })).name, 'FIVE'); |
| 80 | +assert.strictEqual(await Inverses.count(), 3); |
| 81 | + |
| 82 | +await sequelize.close(); |
| 83 | +})(); |
0 commit comments