101 lines
2.8 KiB
Markdown
101 lines
2.8 KiB
Markdown
|
# chaistore
|
||
|
|
|
||
|
|
A [Chai](https://github.com/chaisql/chai)-based session store for [SCS](https://github.com/alexedwards/scs).
|
||
|
|
|
||
|
|
## Setup
|
||
|
|
|
||
|
|
You should have a working Chai database file containing a `sessions` table with the definition:
|
||
|
|
|
||
|
|
```sql
|
||
|
|
CREATE TABLE sessions (
|
||
|
|
token TEXT PRIMARY KEY,
|
||
|
|
data BLOB NOT NULL,
|
||
|
|
expiry TIMESTAMP NOT NULL
|
||
|
|
);
|
||
|
|
|
||
|
|
CREATE INDEX idx_sessions_expiry ON sessions(expiry);
|
||
|
|
```
|
||
|
|
|
||
|
|
## Example
|
||
|
|
|
||
|
|
```go
|
||
|
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"io"
|
||
|
|
"log"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/alexedwards/scs/v2"
|
||
|
|
"go.elara.ws/chaistore"
|
||
|
|
"github.com/chaisql/chai"
|
||
|
|
)
|
||
|
|
|
||
|
|
var sessionManager *scs.SessionManager
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
// Open a Chai database.
|
||
|
|
db, err := chai.Open("chai_db")
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
defer db.Close()
|
||
|
|
|
||
|
|
// Initialize a new session manager and configure it to use chaistore as the session store.
|
||
|
|
sessionManager = scs.New()
|
||
|
|
sessionManager.Store = chaistore.New(db)
|
||
|
|
|
||
|
|
mux := http.NewServeMux()
|
||
|
|
mux.HandleFunc("/put", putHandler)
|
||
|
|
mux.HandleFunc("/get", getHandler)
|
||
|
|
|
||
|
|
http.ListenAndServe(":4000", sessionManager.LoadAndSave(mux))
|
||
|
|
}
|
||
|
|
|
||
|
|
func putHandler(w http.ResponseWriter, r *http.Request) {
|
||
|
|
sessionManager.Put(r.Context(), "message", "Hello from a session!")
|
||
|
|
}
|
||
|
|
|
||
|
|
func getHandler(w http.ResponseWriter, r *http.Request) {
|
||
|
|
msg := sessionManager.GetString(r.Context(), "message")
|
||
|
|
io.WriteString(w, msg)
|
||
|
|
}
|
||
|
|
```
|
||
|
|
|
||
|
|
## Expired Session Cleanup
|
||
|
|
|
||
|
|
This package provides a background 'cleanup' goroutine to delete expired session data. This stops the database table from holding on to invalid sessions indefinitely and growing unnecessarily large. By default the cleanup runs every 5 minutes. You can change this by using the `NewWithCleanupInterval()` function to initialize your session store. For example:
|
||
|
|
|
||
|
|
```go
|
||
|
|
// Run a cleanup every 30 minutes.
|
||
|
|
chaistore.NewWithCleanupInterval(db, 30*time.Minute)
|
||
|
|
|
||
|
|
// Disable the cleanup goroutine by setting the cleanup interval to zero.
|
||
|
|
chaistore.NewWithCleanupInterval(db, 0)
|
||
|
|
```
|
||
|
|
|
||
|
|
### Terminating the Cleanup Goroutine
|
||
|
|
|
||
|
|
It's rare that the cleanup goroutine needs to be terminated --- it is generally intended to be long-lived and run for the lifetime of your application.
|
||
|
|
|
||
|
|
However, there may be occasions when your use of a session store instance is transient. A common example would be using it in a short-lived test function. In this scenario, the cleanup goroutine (which will run forever) will prevent the session store instance from being garbage collected even after the test function has finished. You can prevent this by either disabling the cleanup goroutine altogether (as described above) or by stopping it using the `StopCleanup()` method. For example:
|
||
|
|
|
||
|
|
```go
|
||
|
|
func TestExample(t *testing.T) {
|
||
|
|
db, err := chai.Open("chai_db")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer db.Close()
|
||
|
|
|
||
|
|
store := chaistore.New(db)
|
||
|
|
defer store.StopCleanup()
|
||
|
|
|
||
|
|
sessionManager = scs.New()
|
||
|
|
sessionManager.Store = store
|
||
|
|
|
||
|
|
// Run test...
|
||
|
|
}
|
||
|
|
```
|