20 lines
322 B
Go
20 lines
322 B
Go
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
|
|
}
|