api_context.go

// Copyright 2019 The ChubaoFS Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
// implied. See the License for the specific language governing
// permissions and limitations under the License.

package objectnode

import (
	"net/http"
	"strconv"

	"github.com/chubaofs/chubaofs/proto"

	"github.com/gorilla/mux"
)

const (
	ContextKeyRequestID     = "ctx_request_id"
	ContextKeyRequestAction = "ctx_request_action"
	ContextKeyStatusCode    = "status_code"
)

func SetRequestID(r *http.Request, requestID string) {
	mux.Vars(r)[ContextKeyRequestID] = requestID
}

func GetRequestID(r *http.Request) (id string) {
	return mux.Vars(r)[ContextKeyRequestID]
}

func SetRequestAction(r *http.Request, action proto.Action) {
	mux.Vars(r)[ContextKeyRequestAction] = action.String()
}

func GetActionFromContext(r *http.Request) (action proto.Action) {
	return proto.ParseAction(mux.Vars(r)[ContextKeyRequestAction])
}

func SetResponseStatusCode(r *http.Request, code ErrorCode) {
	mux.Vars(r)[ContextKeyStatusCode] = strconv.Itoa(code.StatusCode)
}

func GetStatusCodeFromContext(r *http.Request) string {
	return mux.Vars(r)[ContextKeyStatusCode]
}