wipfeat/multi-user

1 files changed, 20 insertions, 0 deletions

diff --git a/backend/context/user.go b/backend/context/user.go
new file mode 100644
index 0000000..b8987e4
--- /dev/null
+++ b/backend/context/user.go
@@ -0,0 +1,20 @@
+package context
+
+import (
+	"context"
+)
+
+type contextKey string
+
+const userIDContextKey contextKey = "user_id"
+
+// SetUserID adds the user ID to the context
+func SetUserID(ctx context.Context, userID int64) context.Context {
+	return context.WithValue(ctx, userIDContextKey, userID)
+}
+
+// GetUserID retrieves the user ID from the request context
+func GetUserID(ctx context.Context) (int64, bool) {
+	userID, ok := ctx.Value(userIDContextKey).(int64)
+	return userID, ok
+}