KV 存储与 Golang 框架的关系

package main

import (
    "github.com/coreos/bbolt"
)

// SessionStore 使用 BoltDB 进行会话管理
type SessionStore struct {
    db *bbolt.DB
}

// NewSessionStore 返回新的 SessionStore
func NewSessionStore() *SessionStore {
    db, err := bbolt.Open("sessions.db", 0600, nil)
    if err != nil {
        panic(err)
    }
    return &SessionStore{db}
}

// Set 设置给定会话 ID 的会话数据
func (s *SessionStore) Set(id string, data []byte) error {
    return s.db.Update(func(tx *bbolt.Tx) error {
        b, err := tx.CreateBucketIfNotExists([]byte("sessions"))
        if err != nil {
            return err
        }
        return b.Put([]byte(id), data)
    })
}

// Get 获取给定会话 ID 的会话数据
func (s *SessionStore) Get(id string) ([]byte, error) {
    var data []byte
    err := s.db.View(func(tx *bbolt.Tx) error {
        b := tx.Bucket([]byte("sessions"))
        if b == nil {
            return nil
        }
        data = b.Get([]byte(id))
        return nil
    })
    return data, err
}