From f4368074a7cc07ad305ae9fb9b08d6b8dd3025eb Mon Sep 17 00:00:00 2001 From: Anhgelus Morhtuuzh Date: Thu, 22 Jan 2026 16:04:19 +0100 Subject: feat(context): create custom with config data in --- common/context.go | 51 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 51 insertions(+) create mode 100644 common/context.go (limited to 'common/context.go') diff --git a/common/context.go b/common/context.go new file mode 100644 index 0000000..3a79264 --- /dev/null +++ b/common/context.go @@ -0,0 +1,51 @@ +package common + +import ( + "context" + + "gorm.io/gorm" +) + +type key uint8 + +const ( + keyDB key = 0 + keyDebug key = 1 + keyAuthor key = 2 +) + +func SetDB(ctx context.Context, db *gorm.DB) context.Context { + return context.WithValue(ctx, keyDB, db) +} + +func GetDB(ctx context.Context) *gorm.DB { + raw := ctx.Value(keyDB) + if raw == nil { + return nil + } + return raw.(*gorm.DB) +} + +func SetDebug(ctx context.Context, b bool) context.Context { + return context.WithValue(ctx, keyDebug, b) +} + +func IsDebug(ctx context.Context) bool { + raw := ctx.Value(keyDebug) + if raw == nil { + return false + } + return raw.(bool) +} + +func SetAuthor(ctx context.Context, s string) context.Context { + return context.WithValue(ctx, keyAuthor, s) +} + +func GetAuthor(ctx context.Context) string { + raw := ctx.Value(keyAuthor) + if raw == nil { + return "" + } + return raw.(string) +} -- cgit v1.2.3