标签：http hwlog -- maxConcur req ctx context go

//  Copyright(C) 2021. Huawei Technologies Co.,Ltd.  All rights reserved.

// Package limiter implement a token bucket limiter
package limiter

import (
   "context"
   "huawei.com/npu-exporter/hwlog"
   "huawei.com/npu-exporter/utils"
   "math"
   "net/http"
   "time"
)

const (
   kilo = 1000.0
)

type limitHandler struct {
   concurrency chan struct{}
   httpHandler http.Handler
   log         bool
}

// ServeHTTP implement http.Handler
func (h *limitHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
   ctx := context.TODO()
   reqID := req.Header.Get(hwlog.ReqID.String())
   if reqID != "" {
      ctx = context.WithValue(context.Background(), hwlog.ReqID, reqID)
   }
   id := req.Header.Get(hwlog.UserID.String())
   if id != "" {
      ctx = context.WithValue(ctx, hwlog.UserID, id)
   }
   path := req.URL.Path
   clientIP := utils.ClientIP(req)
   clientUserAgent := req.UserAgent()
   select {
   case _, ok := <-h.concurrency:
      if !ok {
         return
      }
      start := time.Now()
      h.httpHandler.ServeHTTP(w, req)
      stop := time.Since(start)
      latency := int(math.Ceil(float64(stop.Nanoseconds()) / kilo / kilo))
      if h.log {
         hwlog.RunLog.InfofWithCtx(ctx, "%s %s: %s <%3d> (%dms) |%15s |%s ", req.Proto, req.Method, path,
            http.StatusOK, latency, clientIP, clientUserAgent)
      }
      h.concurrency <- struct{}{}
   default:
      hwlog.RunLog.WarnfWithCtx(ctx, "Reject Request:%s: %s <%3d> |%15s |%s ", req.Method, path,
         http.StatusServiceUnavailable, clientIP, clientUserAgent)
      http.Error(w, "503 too busy", http.StatusServiceUnavailable)
   }
}

// NewLimitHandler new a bucket-token limiter
func NewLimitHandler(maxConcur, maxConcurrency int, handler http.Handler, printLog bool) http.Handler {
   if maxConcur < 1 || maxConcur > maxConcurrency {
      hwlog.RunLog.Fatal("maxConcurrency parameter error")
   }
   h := &limitHandler{
      concurrency: make(chan struct{}, maxConcur),
      httpHandler: handler,
      log:         printLog,
   }
   for i := 0; i < maxConcur; i++ {
      h.concurrency <- struct{}{}
   }
   return h
}

标签：http,hwlog,--,maxConcur,req,ctx,context,go
来源： https://www.cnblogs.com/gongxianjin/p/16683079.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。