1. 程式人生 > >sequelize 應用hook 實現對分表的訪問

sequelize 應用hook 實現對分表的訪問

https://github.com/cclient/sequelize-sharding
https://www.npmjs.com/package/sequelize-sharding

實際有效的程式碼就幾行,主要時間都花在這篇部落格和readme.md上

————————————————

公司裡有node專案用sequelize外掛操作mysql

mysql部分表涉及到分表方案,而sequelizel 預設不支援分表

mysql服務端運維層面,java客戶端應用層面,對分庫分表的支援比較成熟,而nodejs客戶端就比較欠缺了

sequelize預設不支援,只能選擇手動擴充套件

思路和java方面的一致,都很簡單,就是在dao操作前後作一個切面,定製個改要操作的庫/表,不只是分庫分表,讀寫分離,異構表優化,多租戶資源隔離,都是按這種思路作的,sequelize也一致

簡單梳理了一下sequelize的執行結構

使用者操作直接面對的是Model

Model定義類似

export class User {
    constructor(id: number,name: string,created_on: Date,updated_on: Date) {
        this.id = id
        this.name = name
        this.created_on = created_on
        this.updated_on = updated_on
    }
    id: number
    name: string
    created_on: Date
    updated_on: Date
}


const USER_ORM_MAP
={ id: { type: Sequelize.INTEGER, autoIncrement: true,primaryKey: true }, name : Sequelize.STRING, created_on : Sequelize.DATE, updated_on : Sequelize.DATE } const DUser = sequelize.define("user", USER_ORM_MAP,{ freezeTableName: true,createdAt:false,updatedAt:false,deletedAt:false
});

 

所有的dao操作都過Model類的例項DUser完成

export async function  getUsersByIds(ids:Array<string>) :Promise<Array<User>>
{
    if(site_ids.length==0){
        return [];
    }
    let hasUsers= await DUser.findAll({where:{id:ids}})
    return hasUsers as Array<User>;
}

 

Model原始碼
https://github.com/sequelize/sequelize/blob/b37985d550e8aeb2bc518f3bd5dcb09c95a142d8/lib/model.js
內有兩個核心物件

static get QueryInterface() {
    return this.sequelize.getQueryInterface();
}

static get QueryGenerator() {
    return this.QueryInterface.QueryGenerator;
}

 

Model對sql的操作通過QueryGenerator實現

https://github.com/sequelize/sequelize/blob/b37985d550e8aeb2bc518f3bd5dcb09c95a142d8/test/unit/sql/insert.test.js

const Support   = require('../support'),
  DataTypes = require('../../../lib/data-types'),
  expectsql = Support.expectsql,
  current   = Support.sequelize,
  sql       = current.dialect.QueryGenerator;

expectsql(sql.insertQuery(User.tableName, {user_name: 'triggertest'}, User.rawAttributes, options),
        {
          query: {
            mssql: 'declare @tmp table ([id] INTEGER,[user_name] NVARCHAR(255));INSERT INTO [users] ([user_name]) OUTPUT INSERTED.[id],INSERTED.[user_name] into @tmp VALUES ($1);select * from @tmp;',
            postgres: 'INSERT INTO "users" ("user_name") VALUES ($1) RETURNING *;',
            default: 'INSERT INTO `users` (`user_name`) VALUES ($1);'
          },
          bind: ['triggertest']
        });

 

原始碼不用再看,到這裡就夠了

sql = current.dialect.QueryGenerator;
sql.insertQuery(User.tableName, {user_name: 'triggertest'}, User.rawAttributes, options)

 

再看Model的原始碼 tableName的部分相關資訊

* @param {string}                  [options.tableName] Defaults to pluralized model name, unless freezeTableName is true, in which case it uses model name verbatim

if (!this.options.tableName) {
      this.tableName = this.options.freezeTableName ? this.name : Utils.underscoredIf(Utils.pluralize(this.name), this.underscored);
    } else {
      this.tableName = this.options.tableName;
    }


static getTableName() { // testhint options:none
    return this.QueryGenerator.addSchema(this);
}

 

預設不公開tableName欄位,也不提供setTableName,tableName完全在定義時構造

理論上更改User.tableName,即可完成換表操作

測試

if (!module.parent) {
getUsersByIds([1,2,3,4]).then(u1=>{
        DUser.tableName="user2";
}).then(()=>{
        getUsersByIds([1,2,3,4]).then(bbb=>{
        })
})
}

 

列印日誌

sequelize deprecated String based operators are now deprecated. Please use Symbol based operators for better security, read more at http://docs.sequelizejs.com/manual/tutorial/querying.html#operators node_modules/sequelize/lib/sequelize.js:237:13

Executing (default): SELECT `id`, `name`, `created_on`, `updated_on` FROM `user` AS `user` WHERE `user`.`id` IN ('1', '2');

Executing (default): SELECT `id`, `name`, `created_on`, `updated_on` FROM `user2` AS `user` WHERE `user`.`id` IN ('1', '2');

驗證成功

也幸好tableName並未設為不可修改,不然改sequelize原始碼還得多花點工夫

剩下就是功能開發了

在User操作之前

let customTableName=getCustomTableName(user);
User.tableName=customTableName;
User.insert(user);

這樣作程式碼侵入性較高,少點改改還可以,大規模應用則太不優雅

優雅的方案
1 定製修改 sequelize 程式碼,使支援分表,對sequelize原始碼不是非常熟悉,滿足某列動態生成的的需求,最多1天搞定,如果想優雅的提到原始碼裡,要花的工夫就多了
2 定製切面 shimmer,這是nodejs aop程式設計的工具,目的是aop,實現方式和java不同,java是執行時通過反射動態構造類,nodejs是在包裹原始包的的物件,像是裝飾者模式
3 思路和2類似,其實sequelize提了原生的切面,用原生切面即可

sequelize提供幾項Hooks
https://github.com/sequelize/sequelize/blob/master/docs/hooks.md
原本是為資料操作提供入口

User.addHook('beforeValidate', (user, options) => {
  user.mood = 'happy';
});

 


主要是操作物件是user實體,hooks算是sequelize預設提供的部分切面

我們可以看看在部分hooks上能否實現對User.tableName的更改

檢視所有 hooks https://github.com/sequelize/sequelize/blob/master/lib/hooks.js

先在beforeFind上作測試

if (!module.parent) {
    User.addHook('beforeFind', (user, options) => {
        User.tableName="user2";
        console.log("beforeFind","set custom tableName",User.tableName)
      })

getUsersByIds([1,2,3,4]).then(u1=>{
}).then(()=>{
        getUsersByIds([1,2,3,4]).then(bbb=>{
        })
})
}

 

看日誌 果然生效

最終選型,先用方案3快速實現,方案3完成後,有精力用方案1實現,方案2因為方案3的存在,沒什麼應用的價值

----

原始碼寫的不是很細緻,只是滿足現階段最初級的需求,更細化和其他拓展需求,可以用同樣的思路實現

去掉異常檢測的程式碼,可以去掉所有依賴

實現用ts,看sequelize原生類結構清晰,測試則是js,js原生程式碼很久不寫了,湊合吧

最後其實分庫分表分割槽的應用,有點過氣的感覺(應用還是很廣,但因為各方面都很成熟完備,沒有再討論優化的空間,顯得很冷清),有空可以單寫一篇出來,技術性的文章太多了,主要想扯點非技術性的東西,給過氣的東西一個總結