eggjs中egg-mysql不支持mysql集群，代碼修改為支持集群

說明：暫不支持egg-mysql動態數據源，用到動態數據源請自行修改。歡迎各位大佬指導。。。

集群配置:

exports.mysql = {
    // 單數據庫信息配置
    client: {
        db1: {
            // host
            host: 'ip1',
            // 端口號
            port: '3306',
            // 用戶名
            user: 'root',
            // 密碼
            password: '123456',
            // 數據庫名
            database: 'mydatebase',
        },
        db2: {
            // host
            host: 'ip2',
            // 端口號
            port: '3306',
            // 用戶名
            user: 'root',
            // 密碼
            password: '123456',
            // 數據庫名
            database: 'mydatebase',
        },
        db3: {
            // host
            host: 'ip3',
            // 端口號
            port: '3306',
            // 用戶名
            user: 'root',
            // 密碼
            password: '123456',
            // 數據庫名
            database: 'mydatebase',
        }
    }

};

單數據庫配置:

exports.mysql = {
  // 單數據庫信息配置
 client: {
    // host
    host: 'ip1',
    // 端口號
    port: '3306',
    // 用戶名
    user: 'root',
    // 密碼
    password: '123456',
    // 數據庫名
    database: 'mydatabase',
  },
  // 是否加載到 app 上，默認開啟
  app: true,
  // 是否加載到 agent 上，默認關閉
  agent: false,

};

1.找到node_modules/ali-rds/lib/client.js，

RDSClient修改如下:

function RDSClient(options) {
  if (!(this instanceof RDSClient)) {
    return new RDSClient(options);
  }
  Operator.call(this);
  let configObj = JSON.stringify(options);
  let len = configObj.match(/\{/g);
  if (len.length > 1) {
    delete options.connectionLimit;
    delete options.database;
    let poolCluster = mysql.createPoolCluster({
      removeNodeErrorCount: 1,
      defaultSelector: "RR" //RR,RANDOM,ORDER
    });
    for (let node in options) {
      poolCluster.add(`${node}`, options[`${node}`]);
    }
    this.pool = poolCluster.of('*', 'RR');
    console.log('<<<<<<----mysql createPoolCluster---->>>>>>>', options, len)
  } else {
    this.pool = mysql.createPool(options);
    console.log('<<<<<<-----mysql createPool--->>>>>>>', options, len)
  }
  [
    'query',
    'getConnection',
  ].forEach(method => {
    this.pool[method] = promisify(this.pool[method]);
  });
}

　2./node_modules/egg-mysql/lib/mysql.js，修改如下

'use strict';

const assert = require('assert');
const rds = require('ali-rds');

let count = 0;

module.exports = app => {
  app.addSingleton('mysql', createOneClient);
};

function createOneClient(config, app) {
  const client = rds(config);
  app.beforeStart(function* () {
    const rows = yield client.query('select now() as currentTime;');
    const index = count++;
    app.coreLogger.info(`[egg-mysql] instance[${index}] status OK, rds currentTime: ${rows[0].currentTime}`);
  });
  return client;
}

posted @ 2020-05-26 17:31 江山一族閱讀(542) 評論(0) 收藏舉報

刷新頁面返回頂部

江山一族

eggjs中egg-mysql不支持mysql集群，代碼修改為支持集群

公告