move concurrent http transport to exist file

jaydensun · Nov 8, 2023 · bf53a92 · bf53a92
1 parent cf1a0fe
commit bf53a92
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 31 deletions.
diff --git a/idl b/idl
diff --git a/transport/concurrent_http.go b/transport/concurrent_http.go
diff --git a/transport/http.go b/transport/http.go
@@ -24,7 +24,6 @@ import (
 
 	"github.com/uber/jaeger-client-go/thrift"
 
-	"github.com/uber/jaeger-client-go"
 	j "github.com/uber/jaeger-client-go/thrift-gen/jaeger"
 )
 
@@ -172,3 +171,27 @@ func serializeThrift(obj thrift.TStruct) (*bytes.Buffer, error) {
 	}
 	return t.Buffer, nil
 }
+
+type ConcurrentHttpTransport struct {
+	*HTTPTransport
+	pool chan int
+}
+
+func NewConcurrentHttpTransport(httpTransport *HTTPTransport) *ConcurrentHttpTransport {
+	ctp := &ConcurrentHttpTransport{
+		HTTPTransport: httpTransport,
+	}
+	ctp.pool = make(chan int, 100)
+	for i := 0; i < 100; i++ {
+		ctp.pool <- 0
+	}
+	return ctp
+}
+
+func (c *ConcurrentHttpTransport) send(spans []*j.Span) error {
+	<-c.pool
+	defer func() {
+		c.pool <- 0
+	}()
+	return c.HTTPTransport.send(spans)
+}