stash-box/pkg/api/factory.go

package api

import (
	"context"
	"net/http"

	"github.com/stashapp/stash-box/pkg/models"
)

type RepoProvider interface {
	// IMPORTANT: the returned Repo object MUST NOT be shared between goroutines.
	// that is: call Repo for each new request/goroutine
	Repo() models.Repo
}

// creates a new Repo (with its own transaction boundary) for each incoming request
func repoMiddleware(provider RepoProvider) func(http.Handler) http.Handler {
	return func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			r = r.WithContext(context.WithValue(r.Context(), ContextRepo, provider.Repo()))

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

func getRepo(ctx context.Context) models.Repo {
	return ctx.Value(ContextRepo).(models.Repo)
}