SCS v2 Manual Load and Save

gist.go

package main

import (
	"io"
	"net/http"

	"github.com/alexedwards/scs/v2"
)

var session = scs.NewSession()

func main() {
	mux := http.NewServeMux()
	mux.HandleFunc("/put", putMiddleware(putHandler))
	mux.HandleFunc("/get", getHandler)

	http.ListenAndServe(":4000", mux)
}

func putMiddleware(next http.HandlerFunc) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		ctx, err := session.Load(r.Context(), r.Header.Get("X-Session"))
		if err != nil {
			http.Error(w, err.Error(), 500)
			return
		}

		session.Put(ctx, "middlewareMessage", "Hello from putMiddleware!")

		next.ServeHTTP(w, r.WithContext(ctx))
	}
}

func putHandler(w http.ResponseWriter, r *http.Request) {
	ctx, err := session.Load(r.Context(), r.Header.Get("X-Session"))
	if err != nil {
		http.Error(w, err.Error(), 500)
		return
	}

	session.Put(ctx, "handlerMessage", "Hello from putHandler!")

	token, expiry, err := session.Commit(ctx)
	if err != nil {
		http.Error(w, err.Error(), 500)
		return
	}
	w.Header().Set("X-Session", token)
	w.Header().Set("X-Session-Expiry", expiry.Format(http.TimeFormat))

	msg := session.GetString(ctx, "middlewareMessage")
	io.WriteString(w, msg)
}

func getHandler(w http.ResponseWriter, r *http.Request) {
	ctx, err := session.Load(r.Context(), r.Header.Get("X-Session"))
	if err != nil {
		http.Error(w, err.Error(), 500)
		return
	}

	msg := session.GetString(ctx, "handlerMessage")
	io.WriteString(w, msg)
}