github.com/kaydxh/golang@v0.0.131/go/net/grpc/grpc_stats.handler.go (about)

     1  /*
     2   *Copyright (c) 2023, kaydxh
     3   *
     4   *Permission is hereby granted, free of charge, to any person obtaining a copy
     5   *of this software and associated documentation files (the "Software"), to deal
     6   *in the Software without restriction, including without limitation the rights
     7   *to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
     8   *copies of the Software, and to permit persons to whom the Software is
     9   *furnished to do so, subject to the following conditions:
    10   *
    11   *The above copyright notice and this permission notice shall be included in all
    12   *copies or substantial portions of the Software.
    13   *
    14   *THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
    15   *IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
    16   *FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
    17   *AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
    18   *LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
    19   *OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
    20   *SOFTWARE.
    21   */
    22  package grpc
    23  
    24  import (
    25  	"context"
    26  
    27  	logs_ "github.com/kaydxh/golang/pkg/logs"
    28  	"google.golang.org/grpc/stats"
    29  )
    30  
    31  type statHandler struct {
    32  }
    33  
    34  // TagRPC can attach some information to the given context.
    35  // The context used for the rest lifetime of the RPC will be derived from
    36  // the returned context.
    37  func (h *statHandler) TagRPC(ctx context.Context, info *stats.RPCTagInfo) context.Context {
    38  	return ctx
    39  }
    40  
    41  // HandleRPC processes the RPC stats.
    42  func (h *statHandler) HandleRPC(ctx context.Context, s stats.RPCStats) {
    43  	logger := logs_.GetLogger(ctx)
    44  	switch v := s.(type) {
    45  	case *stats.OutHeader:
    46  		logger.WithField("local_addr", v.LocalAddr).WithField("remote_addr", v.RemoteAddr).Infof("HandleRPC method %v", v.FullMethod)
    47  	}
    48  
    49  }
    50  
    51  // TagConn can attach some information to the given context.
    52  // The returned context will be used for stats handling.
    53  // For conn stats handling, the context used in HandleConn for this
    54  // connection will be derived from the context returned.
    55  // For RPC stats handling,
    56  //  - On server side, the context used in HandleRPC for all RPCs on this
    57  // connection will be derived from the context returned.
    58  //  - On client side, the context is not derived from the context returned.
    59  func (s *statHandler) TagConn(ctx context.Context, info *stats.ConnTagInfo) context.Context {
    60  	logger := logs_.GetLogger(ctx)
    61  	logger.WithField("local_addr", info.LocalAddr).WithField("remote_addr", info.RemoteAddr).Debugf("tag conn")
    62  	return ctx
    63  }
    64  
    65  // HandleConn processes the Conn stats.
    66  func (s *statHandler) HandleConn(context.Context, stats.ConnStats) {
    67  }