fix: solve lint error,implement error

This commit is contained in:
weloe
2023-05-02 19:45:46 +08:00
parent c04ab085b4
commit 4479cccf73
3 changed files with 14 additions and 11 deletions

View File

@@ -2,6 +2,7 @@ package go_http_context
import (
"context"
"github.com/weloe/token-go/ctx"
"net/http"
)
@@ -13,18 +14,18 @@ func NewReqStorage(req *http.Request) *HttpReqStorage {
return &HttpReqStorage{source: req.Context()}
}
func (r HttpReqStorage) Source() interface{} {
func (r *HttpReqStorage) Source() interface{} {
return r.source
}
func (r HttpReqStorage) Get(key string) interface{} {
func (r *HttpReqStorage) Get(key ctx.StorageKey) interface{} {
return r.source.Value(key)
}
func (r HttpReqStorage) Set(key string, value string) {
func (r *HttpReqStorage) Set(key ctx.StorageKey, value string) {
r.source = context.WithValue(r.source, key, value)
}
func (r HttpReqStorage) Delete(key string) {
func (r *HttpReqStorage) Delete(key ctx.StorageKey) {
r.source = context.WithValue(r.source, key, nil)
}