const mysql = require('mysql');
const _ = require('lodash');
const md5 = require('yoho-md5');
const qs = require('querystring');
const MemoryCache = require('./memory-cache');

class SqlHelper {
    constructor(database) {
        this.config = global.yoho.config.mysql;
        this.logger = global.yoho.logger;
        this.memoryCache = new MemoryCache();
        database = database || 'mysql';
        this.createPool(database);
    }
    createPool(database) {
        this.pool = mysql.createPool(Object.assign(this.config.connect, {
            database,
            queryFormat: function(query, values) {
                if (!values) {
                    return query;
                }
                return query.replace(/\:(\w+)/g, function(txt, key) {
                    if (values.hasOwnProperty(key)) {
                        return this.escape(values[key]);
                    }
                    return txt;
                }.bind(this));
            },
            connectTimeout: 2000
        }));
    }
    getConnection() {
        return new Promise((resolve, reject) => {
            this.pool.getConnection((connErr, connection) => {
                if (connErr) {
                    this.logger.error(connErr);
                    reject(connErr);
                } else {
                    resolve(connection);
                }
            });
        });
    }
    query(sql, params, options) {
        return this.execute(sql, params, options);
    }
    delete(sql, params) {
        return this.execute(sql, params).then(result => {
            return result.affectedRows;
        });
    }
    update(sql, params) {
        return this.execute(sql, params).then(result => {
            return result.changedRows;
        });
    }
    insert(sql, params) {
        return this.execute(sql, params).then(result => {
            return result.insertId;
        });
    }
    execute(sql, params, options = {}) {
        let cache = parseInt(options.cache, 10) || 0;
        let cacheKey;

        // 读取缓存
        if (cache) {
            cacheKey = md5(`${sql}${qs.stringify(params)}${cache}`);

            let cacheResult = this.memoryCache.get(cacheKey);

            if (!_.isUndefined(cacheResult)) {
                return Promise.resolve(cacheResult);
            }
        }

        return new Promise((resolve, reject) => {
            this.getConnection().then(connection => {
                connection.query(sql, params, (queryErr, result) => {
                    connection.release();
                    if (queryErr) {
                        this.logger.error(queryErr);
                        reject(queryErr);
                    } else {
                        // 写入缓存
                        cache && this.memoryCache.set(cacheKey, result, cache);
                        resolve(result);
                    }
                });
            });
        });
    }
    transaction(sqls, cb) {
        return new Promise((resolve, reject) => {
            this.getConnection().then(connection => {
                let promises = _.map(sqls, sql => {
                    return new Promise((res, rej) => {
                        connection.query(sql, (queryErr, result) => {
                            if (queryErr) {
                                connection.rollback();
                                this.logger.error(queryErr);
                                rej(queryErr);
                            } else {
                                cb && cb(sql); // eslint-disable-line
                                res(result);
                            }
                        });
                    });
                });

                Promise.all(promises).then(results => {
                    connection.commit(err => {
                        if (err) {
                            connection.rollback(() => {
                                connection.release();
                            });
                            reject();
                        } else {
                            connection.release();
                            resolve(results);
                        }
                    });
                }, () => {
                    reject();
                });
            });
        });
    }
    changeDatabase(database) {
        return new Promise(resolve => {
            this.pool.end(() => {
                this.createPool(database);
                resolve();
            });
        });
    }
    close() {
        this.pool.end(() => {
            this.logger.log('end');
        });
    }
}

module.exports = SqlHelper;