87 lines
2.0 KiB
Go
87 lines
2.0 KiB
Go
package demogrpc
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/go-resty/resty/v2"
|
|
"github.com/rs/zerolog"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"go.opentelemetry.io/otel/codes"
|
|
"go.opentelemetry.io/otel/trace"
|
|
|
|
grpccodes "google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
|
|
"gitea.libretechconsulting.com/rmcguire/go-app/pkg/otel"
|
|
|
|
pb "gitea.libretechconsulting.com/rmcguire/go-server-with-otel/api/demo/app/v1alpha1"
|
|
"gitea.libretechconsulting.com/rmcguire/go-server-with-otel/pkg/config"
|
|
)
|
|
|
|
const (
|
|
defaultFactType = config.TypeRandom
|
|
factsBaseURI = "https://uselessfacts.jsph.pl/api/v2/facts"
|
|
)
|
|
|
|
type DemoGRPCServer struct {
|
|
tracer trace.Tracer
|
|
ctx context.Context
|
|
cfg *config.DemoConfig
|
|
client *resty.Client
|
|
pb.UnimplementedDemoAppServiceServer
|
|
}
|
|
|
|
func NewDemoGRPCServer(ctx context.Context, cfg *config.DemoConfig) *DemoGRPCServer {
|
|
if cfg.Opts == nil {
|
|
cfg.Opts = &config.DemoOpts{}
|
|
}
|
|
|
|
if cfg.Opts.FactType == "" {
|
|
cfg.Opts.FactType = config.TypeRandom
|
|
}
|
|
|
|
return &DemoGRPCServer{
|
|
ctx: ctx,
|
|
cfg: cfg,
|
|
tracer: otel.GetTracer(ctx, "demoGRPCServer"),
|
|
client: resty.New().SetBaseURL(factsBaseURI),
|
|
}
|
|
}
|
|
|
|
func (d *DemoGRPCServer) GetDemo(ctx context.Context, req *pb.GetDemoRequest) (
|
|
*pb.GetDemoResponse, error,
|
|
) {
|
|
ctx, span := d.tracer.Start(ctx, "getDemo", trace.WithAttributes(
|
|
attribute.String("language", req.GetLanguage()),
|
|
))
|
|
defer span.End()
|
|
|
|
r := d.client.R().SetContext(ctx)
|
|
|
|
if req.GetLanguage() != "" {
|
|
r = r.SetQueryParam("language", req.GetLanguage())
|
|
}
|
|
|
|
fact := new(RandomFact)
|
|
r.SetResult(fact)
|
|
|
|
resp, err := r.Get(fmt.Sprintf("/%s", d.cfg.Opts.FactType))
|
|
if err != nil {
|
|
span.SetStatus(codes.Error, err.Error())
|
|
span.RecordError(err)
|
|
|
|
zerolog.Ctx(d.ctx).Err(err).Send()
|
|
|
|
return nil, status.Errorf(grpccodes.Unknown, "%s: %s",
|
|
err.Error(), string(resp.Body()))
|
|
}
|
|
|
|
span.SetAttributes(attribute.String("fact", fact.Text))
|
|
span.SetStatus(codes.Ok, "")
|
|
|
|
zerolog.Ctx(d.ctx).Debug().Str("fact", fact.Text).Msg("retrieved fact")
|
|
|
|
return fact.FactToProto(), nil
|
|
}
|