@keyvhq/mysql/src/index.js

MySQL/MariaDB storage adapter for Keyv

'use strict'

const KeyvSql = require('@keyvhq/sql')
const mysql = require('mysql2/promise')

class KeyvMysql extends KeyvSql {
  constructor (uri, options) {
    uri = uri || {}
    if (typeof uri === 'string') {
      uri = { uri }
    }

    if (uri.uri) {
      uri = Object.assign({ uri: uri.uri }, uri)
    }

    options = Object.assign(
      {
        dialect: 'mysql',
        uri: 'mysql://localhost'
      },
      uri,
      options
    )
    const pool = mysql.createPool(options.uri)
    options.connect = () =>
      Promise.resolve()
        .then(() => {
          return sql => pool.execute(sql).then(data => data[0])
        })

    super(options)
  }
}

module.exports = KeyvMysql