Add opentelemetry

Author: Aristat

README.md

@@ -45,10 +45,11 @@ Packages which use in this example project
 10. [gqlgen](https://github.com/99designs/gqlgen) - graphql server library
 11. [protobuf](https://pkg.go.dev/google.golang.org/protobuf) - Google's data interchange format
 12. [grpc](google.golang.org/grpc) - RPC framework
-13. [jaeger](https://github.com/uber/jaeger-client-go) - Jaeger Bindings for Go OpenTracing API
-14. [casbin](https://github.com/casbin/casbin) - Supports access control
-15. [dataloaden](https://github.com/vektah/dataloaden) - DataLoader for graphql
-16. [nats](https://github.com/nats-io/nats.go) - Golang client for NATS, the cloud native messaging system
+13. [opentelemetry](https://github.com/open-telemetry/opentelemetry-go) - OpenTelemetry
+14. [jaeger](https://github.com/uber/jaeger-client-go) - Jaeger Bindings for Go OpenTelemetry API
+15. [casbin](https://github.com/casbin/casbin) - Supports access control
+16. [dataloaden](https://github.com/vektah/dataloaden) - DataLoader for graphql
+17. [nats](https://github.com/nats-io/nats.go) - Golang client for NATS, the cloud native messaging system
 
 # Installing
 

app/common/jaeger.go

@@ -1,39 +1,38 @@
 package common
 
 import (
-	"log"
+	"github.com/aristat/golang-example-app/app/tracing"
+	"go.opentelemetry.io/otel/exporters/jaeger"
+	"go.opentelemetry.io/otel/sdk/resource"
+	tracesdk "go.opentelemetry.io/otel/sdk/trace"
+	semconv "go.opentelemetry.io/otel/semconv/v1.4.0"
 
 	"github.com/spf13/viper"
-
-	"github.com/opentracing/opentracing-go"
-	"github.com/uber/jaeger-client-go"
-
-	jaegerConfig "github.com/uber/jaeger-client-go/config"
 )
 
-func GenerateTracerForTestClient(serviceName string, cfg *viper.Viper) opentracing.Tracer {
-	jaegerCfg := jaegerConfig.Configuration{
-		Sampler: &jaegerConfig.SamplerConfig{
-			Type:  "const",
-			Param: 1,
-		},
-		Reporter: &jaegerConfig.ReporterConfig{
-			LogSpans: false,
-		},
+func GenerateTracerForTestClient(serviceName string, cfg *viper.Viper) (*tracesdk.TracerProvider, error) {
+	configuration := tracing.Configuration{}
+	err := cfg.UnmarshalKey("tracing.jaeger", &configuration)
+	if err != nil {
+		return nil, err
 	}
 
-	e := cfg.UnmarshalKey("tracing.jaeger", &jaegerCfg)
-	if e != nil {
-		log.Fatal("Jaeger initialize config error")
-	}
-
-	// redefine for test services
-	jaegerCfg.ServiceName = serviceName
-
-	tracer, _, e := jaegerCfg.NewTracer(jaegerConfig.Logger(jaeger.StdLogger))
-	if e != nil {
-		log.Fatal("Jaeger initialize client error")
+	exp, err := jaeger.New(
+		jaeger.WithAgentEndpoint(
+			jaeger.WithAgentHost(configuration.AgentHost),
+			jaeger.WithAgentPort(configuration.AgentPort),
+		),
+	)
+	if err != nil {
+		return nil, err
 	}
 
-	return tracer
+	tp := tracesdk.NewTracerProvider(
+		tracesdk.WithBatcher(exp),
+		tracesdk.WithResource(resource.NewWithAttributes(
+			semconv.SchemaURL,
+			semconv.ServiceNameKey.String(serviceName),
+		)),
+	)
+	return tp, nil
 }

app/config/wire_gen.go

@@ -18,7 +18,7 @@ func Cfg(cfg *viper.Viper) (Config, func(), error) {
 	c := Config{LogLevel: logger.LevelDebug}
 	e := cfg.UnmarshalKey("db", &c)
 	if e != nil {
-		return c, func() {}, nil
+		return c, func() {}, e
 	}
 	return c, func() {}, nil
 }

app/db/provider.go

@@ -16,7 +16,7 @@ import (
 	gqlgen "github.com/99designs/gqlgen/graphql"
 	"github.com/aristat/golang-example-app/app/logger"
 	"github.com/aristat/golang-example-app/generated/graphql"
-	"github.com/go-chi/chi"
+	"github.com/go-chi/chi/v5"
 	"github.com/vektah/gqlparser/v2/gqlerror"
 )
 

app/db/wire_gen.go

@@ -4,11 +4,14 @@ import (
 	"context"
 	"errors"
 
-	grpc_opentracing "github.com/grpc-ecosystem/go-grpc-middleware/tracing/opentracing"
+	"go.opentelemetry.io/otel/propagation"
+
+	"go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc"
+	tracesdk "go.opentelemetry.io/otel/sdk/trace"
+
+	"google.golang.org/grpc/credentials/insecure"
 
 	"github.com/aristat/golang-example-app/app/logger"
-	"github.com/aristat/golang-example-app/app/tracing"
-	grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/keepalive"
 )
@@ -20,7 +23,7 @@ var errCfgInvalid = errors.New("cfg is not present or invalid")
 // PoolManager
 type PoolManager struct {
 	ctx     context.Context
-	tracing tracing.Tracer
+	tracing *tracesdk.TracerProvider
 	cfg     *Config
 	logger  logger.Logger
 }
@@ -34,7 +37,7 @@ func (p *PoolManager) NewPool(service string) (_ *Pool, loaded bool, _ error) {
 	}
 
 	var opts []grpc.DialOption
-	opts = append(opts, grpc.WithInsecure())
+	opts = append(opts, grpc.WithTransportCredentials(insecure.NewCredentials()))
 
 	cl := s.ClientParameters
 	if cl == nil {
@@ -47,14 +50,17 @@ func (p *PoolManager) NewPool(service string) (_ *Pool, loaded bool, _ error) {
 
 	opts = append(opts, grpc.WithKeepaliveParams(*cl))
 	opts = append(opts,
-		grpc.WithUnaryInterceptor(grpc_middleware.ChainUnaryClient(
+		grpc.WithChainUnaryInterceptor(
 			logger.UnaryClientInterceptor(p.logger, true),
-			grpc_opentracing.UnaryClientInterceptor(grpc_opentracing.WithTracer(p.tracing)),
-		)))
-	opts = append(opts, grpc.WithStreamInterceptor(grpc_middleware.ChainStreamClient(
-		logger.StreamClientInterceptor(p.logger, true),
-		grpc_opentracing.StreamClientInterceptor(grpc_opentracing.WithTracer(p.tracing)),
-	)))
+			otelgrpc.UnaryClientInterceptor(otelgrpc.WithTracerProvider(p.tracing), otelgrpc.WithPropagators(propagation.TraceContext{})),
+		),
+	)
+	opts = append(opts,
+		grpc.WithChainStreamInterceptor(
+			logger.StreamClientInterceptor(p.logger, true),
+			otelgrpc.StreamClientInterceptor(otelgrpc.WithTracerProvider(p.tracing), otelgrpc.WithPropagators(propagation.TraceContext{})),
+		),
+	)
 
 	pool, l := NewPool(p.ctx, service, s.Target,
 		MaxConn(s.MaxConn),
@@ -68,6 +74,6 @@ func (p *PoolManager) NewPool(service string) (_ *Pool, loaded bool, _ error) {
 }
 
 // NewPoolManager
-func NewPoolManager(ctx context.Context, tracing tracing.Tracer, logger logger.Logger, cfg *Config) *PoolManager {
+func NewPoolManager(ctx context.Context, tracing *tracesdk.TracerProvider, logger logger.Logger, cfg *Config) *PoolManager {
 	return &PoolManager{ctx: ctx, tracing: tracing, cfg: cfg, logger: logger}
 }

app/graphql/graphql.go

@@ -5,9 +5,10 @@ import (
 	"sync"
 	"time"
 
+	tracesdk "go.opentelemetry.io/otel/sdk/trace"
+
 	"github.com/aristat/golang-example-app/app/logger"
 
-	"github.com/aristat/golang-example-app/app/tracing"
 	"github.com/google/wire"
 	"github.com/spf13/viper"
 	"google.golang.org/grpc/keepalive"
@@ -50,7 +51,7 @@ type Config struct {
 }
 
 // Provider
-func Provider(ctx context.Context, tracing tracing.Tracer, logger logger.Logger, cfg *Config) (*PoolManager, func(), error) {
+func Provider(ctx context.Context, tracing *tracesdk.TracerProvider, logger logger.Logger, cfg *Config) (*PoolManager, func(), error) {
 	mutexPM.Lock()
 	defer mutexPM.Unlock()
 	if pm != nil {

app/graphql_resolver/wire_gen.go

@@ -5,7 +5,7 @@ import (
 	"net/http"
 	"sync"
 
-	"github.com/go-chi/chi"
+	"github.com/go-chi/chi/v5"
 
 	"github.com/go-session/session"
 

app/grpc/manager.go

@@ -5,7 +5,7 @@ import (
 	"time"
 
 	"github.com/aristat/golang-example-app/app/logger"
-	"github.com/go-chi/chi/middleware"
+	"github.com/go-chi/chi/v5/middleware"
 )
 
 func Logger(l logger.Logger) func(next http.Handler) http.Handler {

app/grpc/provider.go

@@ -3,31 +3,32 @@ package http
 import (
 	"context"
 
+	"github.com/riandyrn/otelchi"
+
 	"github.com/aristat/golang-example-app/app/dataloader"
 
 	products_router "github.com/aristat/golang-example-app/app/http_routers/products-router"
 
 	"github.com/aristat/golang-example-app/app/auth"
 
 	"github.com/aristat/golang-example-app/app/graphql"
-	"github.com/go-chi/chi"
-	"github.com/go-chi/chi/middleware"
-	"github.com/opentracing/opentracing-go"
+	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/chi/v5/middleware"
 
 	"github.com/aristat/golang-example-app/app/logger"
 
 	"github.com/google/wire"
 	"github.com/spf13/viper"
 )
 
-var mux *chi.Mux
+var muxRouter *chi.Mux
 
 // Cfg
 func Cfg(cfg *viper.Viper) (Config, func(), error) {
 	c := Config{}
 	e := cfg.UnmarshalKey("http", &c)
 	if e != nil {
-		return c, func() {}, nil
+		return c, func() {}, e
 	}
 	c.Debug = cfg.GetBool("debug")
 	return c, func() {}, nil
@@ -39,21 +40,21 @@ func CfgTest() (Config, func(), error) {
 }
 
 // Mux
-func Mux(managers Managers, log logger.Logger, tracer opentracing.Tracer) (*chi.Mux, func(), error) {
-	if mux != nil {
-		return mux, func() {}, nil
+func Mux(managers Managers, log logger.Logger) (*chi.Mux, func(), error) {
+	if muxRouter != nil {
+		return muxRouter, func() {}, nil
 	}
 
-	mux = chi.NewRouter()
-	mux.Use(middleware.RequestID)
-	mux.Use(Logger(log))
-	mux.Use(Tracer(tracer))
-	mux.Use(dataloader.LoaderMiddleware)
+	muxRouter = chi.NewRouter()
+	muxRouter.Use(middleware.RequestID)
+	muxRouter.Use(Logger(log))
+	muxRouter.Use(otelchi.Middleware("http-server", otelchi.WithChiRoutes(muxRouter)))
+	muxRouter.Use(dataloader.LoaderMiddleware)
 
-	managers.products.Router.Run(mux)
-	managers.graphql.Routers(mux.With(managers.authMiddleware.JWTHandler))
+	managers.products.Router.Run(muxRouter)
+	managers.graphql.Routers(muxRouter.With(managers.authMiddleware.JWTHandler))
 
-	return mux, func() {}, nil
+	return muxRouter, func() {}, nil
 }
 
 // ServiceManagers