to filter client stream span (#270)

* to filter client stream span

* delete stream test

* delete stream test

Author: NameHaibinZhang

README.md

@@ -88,7 +88,7 @@ Also there are several [**documents**](./docs) that you may find useful for eith
 | go-redis      | https://github.com/redis/go-redis              | v9.0.5                | v9.5.1                |
 | go-redis v8   | https://github.com/redis/go-redis              | v8.11.0               | v8.11.5               |
 | gorm          | https://github.com/go-gorm/gorm                | v1.22.0               | v1.25.9               |
-| grpc          | https://google.golang.org/grpc                 | v1.44.0               | v1.67.0               |
+| grpc          | https://google.golang.org/grpc                 | v1.44.0               | v1.68.2               |
 | hertz         | https://github.com/cloudwego/hertz             | v0.8.0                | v0.9.2                |
 | kitex         | https://github.com/cloudwego/kitex             | v0.5.1                | v0.11.3               |
 | kratos        | https://github.com/go-kratos/kratos            | v2.6.3                | v2.8.2                |

docs/supported-libraries.md

@@ -12,9 +12,9 @@
 | go-redis     | https://github.com/redis/go-redis              | v9.0.5                | v9.5.1                |
 | go-redis v8  | https://github.com/redis/go-redis              | v8.11.0               | v8.11.5               |
 | gorm         | https://github.com/go-gorm/gorm                | v1.22.0               | v1.25.9               |
-| grpc         | https://google.golang.org/grpc                 | v1.44.0               | v1.67.0               |
+| grpc         | https://google.golang.org/grpc                 | v1.44.0               | v1.68.2               |
 | hertz        | https://github.com/cloudwego/hertz             | v0.8.0                | v0.9.2                |
-| kitex        | https://github.com/cloudwego/kitex             | v0.5.1                |  v0.11.3              |
+| kitex        | https://github.com/cloudwego/kitex             | v0.5.1                | v0.11.3               |
 | kratos       | https://github.com/go-kratos/kratos            | v2.6.3                | v2.8.2                |
 | log          | https://pkg.go.dev/log                         | -                     | -                     |
 | logrus       | https://github.com/sirupsen/logrus             | v1.5.0                | v1.9.3                |
@@ -25,4 +25,8 @@
 | redigo       | https://github.com/gomodule/redigo             | v1.9.0                | v1.9.2                |
 | slog         | https://pkg.go.dev/log/slog                    | -                     | -                     |
 | zap          | https://github.com/uber-go/zap                 | v1.20.0               | v1.27.0               |
-| zerolog      | https://github.com/rs/zerolog                  | v1.10.0               | v1.33.0               |
+| zerolog      | https://github.com/rs/zerolog                  | v1.10.0               | v1.33.0               |
+
+### Notice
+
+#### grpc stream is not support yet

pkg/data/default.json

@@ -512,6 +512,15 @@
     "OnExit": "grpcServerOnExit",
     "Path": "github.com/alibaba/opentelemetry-go-auto-instrumentation/pkg/rules/grpc"
   },
+  {
+    "Version": "[1.44.0,)",
+    "ImportPath": "google.golang.org/grpc",
+    "ReceiverType": "*ClientConn",
+    "Function": "NewStream",
+    "OnEnter": "grpcClientNewStreamOnEnter",
+    "OnExit": "grpcClientNewStreamOnExit",
+    "Path": "github.com/alibaba/opentelemetry-go-auto-instrumentation/pkg/rules/grpc"
+  },
   {
     "ImportPath": "github.com/cloudwego/hertz/pkg/app/server",
     "Function": "New",

pkg/rules/grpc/grpc_client_new_stream.go

@@ -0,0 +1,32 @@
+// Copyright (c) 2024 Alibaba Group Holding Ltd.
+//
+// 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 grpc
+
+import (
+	"context"
+	"github.com/alibaba/opentelemetry-go-auto-instrumentation/pkg/api"
+	"google.golang.org/grpc"
+)
+
+func grpcClientNewStreamOnEnter(call api.CallContext, cc *grpc.ClientConn, ctx context.Context, desc *grpc.StreamDesc, method string, opts ...grpc.CallOption) {
+	var stream_filter bool
+	stream_filter = true
+	ctx = context.WithValue(ctx, "stream_filter", &stream_filter)
+	call.SetParam(1, ctx)
+}
+
+func grpcClientNewStreamOnExit(call api.CallContext, cs grpc.ClientStream, err error) {
+	return
+}

pkg/rules/grpc/grpc_new_client_setup.go

@@ -49,6 +49,10 @@ func NewClientNewHandler(opts ...Option) stats.Handler {
 
 // TagRPC can attach some information to the given context.
 func (h *clientNewHandler) TagRPC(ctx context.Context, info *stats.RPCTagInfo) context.Context {
+	filter, _ := ctx.Value("stream_filter").(*bool)
+	if filter != nil && *filter {
+		return ctx
+	}
 	nCtx := grpcClientInstrument.Start(ctx, grpcRequest{
 		methodName:    info.FullMethodName,
 		serverAddress: h.serverAddr,