Function updateRpcFocusStats

grpc_util.go:282–322

(startTime time.Time, mgr *cbgt.Manager,
	req *pb.SearchRequest, ctx context.Context, err error)

}

func updateRpcFocusStats(startTime time.Time, mgr *cbgt.Manager,
	req *pb.SearchRequest, ctx context.Context, err error) {
	focusStats := GrpcPathStats.FocusStats(req.IndexName)
	if focusStats != nil {
		// check whether its a client request and track only in the coordinating node
		if _, er := extractMetaHeader(ctx, rpcClusterActionKey); er != nil {
			// co-ordinating node
			atomic.AddUint64(&focusStats.TotGrpcRequest, 1)
			atomic.AddUint64(&focusStats.TotGrpcRequestTimeNS,
				uint64(time.Now().Sub(startTime)))

			slowQueryLogTimeoutV := mgr.Options()["slowQueryLogTimeout"]
			if slowQueryLogTimeoutV != "" {
				var slowQueryLogTimeout time.Duration
				slowQueryLogTimeout, err = time.ParseDuration(slowQueryLogTimeoutV)
				if err == nil {
					d := time.Since(startTime)
					if d > slowQueryLogTimeout {
						log.Warnf("grpc_util: slow-query index: %s,"+
							" query: %s, duration: %v, err: %v",
							req.IndexName, string(req.Contents), d, err)

						atomic.AddUint64(&focusStats.TotGrpcRequestSlow, 1)
					}
				}
			}

			if err != nil {
				atomic.AddUint64(&focusStats.TotGrpcRequestErr, 1)
				if err == context.DeadlineExceeded {
					atomic.AddUint64(&focusStats.TotGrpcRequestTimeout, 1)
				}
			}
		} else {
			// not a co-ordinating node
			atomic.AddUint64(&focusStats.TotGrpcInternalRequest, 1)
			atomic.AddUint64(&focusStats.TotGrpcInternalRequestTimeNS,
				uint64(time.Now().Sub(startTime)))
		}
	}
}

// httpStatusCodes map the canonical gRPC codes to HTTP status codes
// as per https://github.com/grpc/grpc/blob/master/doc/statuscodes.md