package http_proxy

import (
	"github.com/gin-gonic/gin"
)

type (
	RpcContext struct {
		Context map[string]string `json:"context"`
	}
	//Context http_proxy 上下文，对外屏蔽内部使用的框架, 便于框架内部迭代而不影响外部api
	Context struct {
		*gin.Context
		rpcCtx *RpcContext
	}
)

func (ctx *Context) SetRpcContext(key string, value string) {
	if ctx.rpcCtx == nil {
		ctx.rpcCtx = &RpcContext{
			Context: make(map[string]string),
		}
	}

	ctx.rpcCtx.Context[key] = value
}

func (ctx *Context) GetRpcContext() *RpcContext {
	return ctx.rpcCtx
}