How to use getTags method of grpcext Package

Best K6 code snippet using grpcext.getTags

conn.go

Source:conn.go

...164}165// HandleRPC implements the grpcstats.Handler interface166func (h statsHandler) HandleRPC(ctx context.Context, stat grpcstats.RPCStats) {167	state := h.vu.State()168	tags := getTags(ctx)169	switch s := stat.(type) {170	case *grpcstats.OutHeader:171		if state.Options.SystemTags.Has(metrics.TagIP) && s.RemoteAddr != nil {172			if ip, _, err := net.SplitHostPort(s.RemoteAddr.String()); err == nil {173				tags["ip"] = ip174			}175		}176	case *grpcstats.End:177		if state.Options.SystemTags.Has(metrics.TagStatus) {178			tags["status"] = strconv.Itoa(int(status.Code(s.Error)))179		}180		mTags := map[string]string(tags)181		sampleTags := metrics.IntoSampleTags(&mTags)182		metrics.PushIfNotDone(ctx, state.Samples, metrics.ConnectedSamples{183			Samples: []metrics.Sample{184				{185					Metric: state.BuiltinMetrics.GRPCReqDuration,186					Tags:   sampleTags,187					Value:  metrics.D(s.EndTime.Sub(s.BeginTime)),188					Time:   s.EndTime,189				},190			},191		})192	}193	// (rogchap) Re-using --http-debug flag as gRPC is technically still HTTP194	if state.Options.HTTPDebug.String != "" {195		logger := state.Logger.WithField("source", "http-debug")196		httpDebugOption := state.Options.HTTPDebug.String197		DebugStat(logger, stat, httpDebugOption)198	}199}200// DebugStat prints debugging information based on RPCStats.201func DebugStat(logger logrus.FieldLogger, stat grpcstats.RPCStats, httpDebugOption string) {202	switch s := stat.(type) {203	case *grpcstats.OutHeader:204		logger.Infof("Out Header:\nFull Method: %s\nRemote Address: %s\n%s\n",205			s.FullMethod, s.RemoteAddr, formatMetadata(s.Header))206	case *grpcstats.OutTrailer:207		if len(s.Trailer) > 0 {208			logger.Infof("Out Trailer:\n%s\n", formatMetadata(s.Trailer))209		}210	case *grpcstats.OutPayload:211		if httpDebugOption == "full" {212			logger.Infof("Out Payload:\nWire Length: %d\nSent Time: %s\n%s\n\n",213				s.WireLength, s.SentTime, formatPayload(s.Payload))214		}215	case *grpcstats.InHeader:216		if len(s.Header) > 0 {217			logger.Infof("In Header:\nWire Length: %d\n%s\n", s.WireLength, formatMetadata(s.Header))218		}219	case *grpcstats.InTrailer:220		if len(s.Trailer) > 0 {221			logger.Infof("In Trailer:\nWire Length: %d\n%s\n", s.WireLength, formatMetadata(s.Trailer))222		}223	case *grpcstats.InPayload:224		if httpDebugOption == "full" {225			logger.Infof("In Payload:\nWire Length: %d\nReceived Time: %s\n%s\n\n",226				s.WireLength, s.RecvTime, formatPayload(s.Payload))227		}228	}229}230func formatMetadata(md metadata.MD) string {231	var sb strings.Builder232	for k, v := range md {233		sb.WriteString(k)234		sb.WriteString(": ")235		sb.WriteString(strings.Join(v, ", "))236		sb.WriteRune('\n')237	}238	return sb.String()239}240func formatPayload(payload interface{}) string {241	msg, ok := payload.(proto.Message)242	if !ok {243		// check to see if we are dealing with a APIv1 message244		msgV1, ok := payload.(protov1.Message)245		if !ok {246			return ""247		}248		msg = protov1.MessageV2(msgV1)249	}250	marshaler := prototext.MarshalOptions{251		Multiline: true,252		Indent:    "  ",253	}254	b, err := marshaler.Marshal(msg)255	if err != nil {256		return ""257	}258	return string(b)259}260type ctxKeyTags struct{}261type reqtags map[string]string262func withTags(ctx context.Context, tags reqtags) context.Context {263	if tags == nil {264		tags = make(map[string]string)265	}266	return context.WithValue(ctx, ctxKeyTags{}, tags)267}268func getTags(ctx context.Context) reqtags {269	v := ctx.Value(ctxKeyTags{})270	if v == nil {271		return make(map[string]string)272	}273	return v.(reqtags)274}...

getTags

Using AI Code Generation

1import (2const (3func main() {4    conn, err := grpc.Dial(address, grpc.WithInsecure())5    if err != nil {6        log.Fatalf("did not connect: %v", err)7    }8    defer conn.Close()9    c := pb.NewGreeterClient(conn)10    ctx, cancel := context.WithTimeout(context.Background(), time.Second)11    defer cancel()12    r, err := c.SayHello(ctx, &pb.HelloRequest{Name: name})13    if err != nil {14        log.Fatalf("could not greet: %v", err)15    }16    log.Printf("Greeting: %s", r.Message)17    grpclog.Infof("Greeting: %s", r.Message)18    tags := grpcext.GetTags(ctx)19    fmt.Println("Tags: ", tags)20}

getTags

Using AI Code Generation

1import (2func main() {3    creds, err := credentials.NewClientTLSFromFile("cert.pem", "localhost")4    if err != nil {5        log.Fatalf("could not load tls cert: %s", err)6    }7    conn, err = grpc.Dial(":9000", grpc.WithTransportCredentials(creds))8    if err != nil {9        log.Fatalf("did not connect: %s", err)10    }11    defer conn.Close()12    c := pb.NewGrpcExtClient(conn)13    response, err := c.GetTags(context.Background(), &pb.GetTagsRequest{Id: "1"})14    if err != nil {15        log.Fatalf("Error when calling GetTags: %s", err)16    }17    log.Printf("Response from server: %s", response)18}19import (20func main() {21    creds, err := credentials.NewClientTLSFromFile("cert.pem", "localhost")22    if err != nil {23        log.Fatalf("could not load tls cert: %s", err)24    }25    conn, err = grpc.Dial(":9000", grpc.WithTransportCredentials(creds))26    if err != nil {27        log.Fatalf("did not connect: %s", err)28    }29    defer conn.Close()30    c := pb.NewGrpcExtClient(conn)31    response, err := c.GetTags(context.Background(), &pb.GetTagsRequest{Id: "1"})32    if err != nil {

getTags

Using AI Code Generation

1import (2func main() {3    conn, err := grpc.Dial("localhost:50051", grpc.WithInsecure())4    if err != nil {5        log.Fatalf("did not connect: %v", err)6    }7    defer conn.Close()8    c := pb.NewTagServiceClient(conn)9    ctx, cancel := context.WithTimeout(context.Background(), time.Second)10    defer cancel()11    r, err := c.GetTags(ctx, &pb.GetTagsRequest{})12    if err != nil {13        log.Fatalf("could not greet: %v", err)14    }15    fmt.Println(r)16}17import (18func main() {19    conn, err := grpc.Dial("localhost:50051", grpc.WithInsecure())20    if err != nil {21        log.Fatalf("did not connect: %v", err)22    }23    defer conn.Close()24    c := pb.NewTagServiceClient(conn)25    ctx, cancel := context.WithTimeout(context.Background(), time.Second)26    defer cancel()27    r, err := c.GetTags(ctx, &pb.GetTagsRequest{})28    if err != nil {29        log.Fatalf("could not greet: %v", err)30    }31    fmt.Println(r)32}33import (34func main() {35    conn, err := grpc.Dial("localhost:50051", grpc.WithInsecure())36    if err != nil {37        log.Fatalf("did not connect: %v", err)38    }

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.