session/lib/redis-store.js

/**
 * redis版会话管理
 * @author yutent<yutent.io@gmail.com>
 * @date 2020/09/18 16:35:26
 */
import 'es.shim'
import Ioredis from 'ioredis'

export default class Session {

  #store = null
  #ttl = 60

  constructor(opt) {
    this.#store = new Ioredis({
      host: opt.db.host || '127.0.0.1',
      port: opt.db.port || 6379,
      db: opt.db.db || 0
    })
    this.#ttl = opt.ttl
  }

  update(ssid) {
    // 设置session有效期
    this.#store.expire(ssid, this.#ttl)
  }

  // 获取session字段值, 需要await指令
  get(ssid, key) {
    let defer = Promise.defer()

    this.#store.hgetall(ssid, (err, obj) => {
      if (err) {
        return defer.reject(err)
      }

      for (let i in obj) {
        if (obj[i]) {
          obj[i] = Number.parse(obj[i])
        }
      }
      //不传key时,直接返回全部字段
      if (key) {
        defer.resolve(obj[key])
      } else {
        defer.resolve(obj)
      }
    })
    return defer.promise
  }

  //设置session字段值
  set(ssid, key, val) {
    if (typeof key === 'object') {
      for (let i in key) {
        this.#store.hset(ssid, i, key[i])
      }
    } else {
      this.#store.hset(ssid, key, val)
    }
  }

  //删除单个字段
  unset(ssid, key) {
    this.#store.hdel(ssid, key)
  }

  //清除个人session
  clear(ssid) {
    this.#store.del(ssid)
  }
}