add Limited method

clubpay · May 12, 2022 · 41b7c90 · 41b7c90
1 parent f276d19
commit 41b7c90
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/ctx.go b/ctx.go
@@ -172,6 +172,12 @@ func (ctx *Context) HasError() bool {
 	return ctx.err != nil
 }
 
+// Limited returns a LimitedContext. This is useful when you don't want to give all
+// capabilities of the Context to some other function/method.
+func (ctx *Context) Limited() *LimitedContext {
+	return newLimitedContext(ctx)
+}
+
 func (ctx *Context) reset() {
 	for k := range ctx.kv {
 		delete(ctx.kv, k)

diff --git a/ctx_limited.go b/ctx_limited.go
@@ -54,7 +54,3 @@ func (ctx *LimitedContext) Route() string {
 func (ctx *LimitedContext) ServiceName() string {
 	return ctx.ctx.ServiceName()
 }
-
-func (ctx *LimitedContext) Cluster() Cluster {
-	panic("not implemented")
-}