package db

import (
	"context"
	"database/sql"
)

func Inject(ctx context.Context, db *sql.DB) context.Context {
	return context.WithValue(ctx, "__db__", db)
}

func From(ctx context.Context) *sql.DB {
	return Extract(ctx)
}

func Extract(ctx context.Context) *sql.DB {
	v, _ := ctx.Value("__db__").(*sql.DB)
	return v
}