Skip to content

Commit

Permalink
minor changes
Browse files Browse the repository at this point in the history
Signed-off-by: chahatsagarmain <chahatsagar2003@gmail.com>
  • Loading branch information
chahatsagarmain committed Oct 23, 2024
1 parent 34da943 commit 5f80b28
Show file tree
Hide file tree
Showing 4 changed files with 14 additions and 14 deletions.
4 changes: 2 additions & 2 deletions crossdock/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ import (
const (
behaviorEndToEnd = "endtoend"

envCollectorSamplingHostPort = "JAEGER_COLLECTOR_SAMPLING_HOST_PORT"
envCollectorSamplingHostPort = "JAEGER_COLLECTOR_HC_HOST_PORT"
envQueryHostPort = "JAEGER_QUERY_HOST_PORT"

envQueryHealthcheckHostPort = "JAEGER_QUERY_HC_HOST_PORT"
Expand Down Expand Up @@ -78,7 +78,7 @@ func (h *clientHandler) initialize() {
httpHealthCheck(logger, "jaeger-collector", "http://"+collectorHealthcheckHostPort)

queryService := services.NewQueryService("http://"+queryHostPort, logger)
samplingService := services.NewAgentService("http://"+collectorSamplingHostPort, logger)
samplingService := services.NewCollectorService("http://"+collectorSamplingHostPort, logger)

traceHandler := services.NewTraceHandler(queryService, samplingService, logger)
behaviors := crossdock.Behaviors{
Expand Down
14 changes: 7 additions & 7 deletions crossdock/services/agent.go
Original file line number Diff line number Diff line change
Expand Up @@ -18,19 +18,19 @@ import (

var errSamplingRateMissing = errors.New("sampling rate is missing")

// AgentService is the service used to report traces to the collector.
type AgentService interface {
// CollectorService is the service used to report traces to the collector.
type CollectorService interface {
GetSamplingRate(service, operation string) (float64, error)
}

type agentService struct {
type collectorService struct {
url string
logger *zap.Logger
}

// NewAgentService returns an instance of AgentService.
func NewAgentService(url string, logger *zap.Logger) AgentService {
return &agentService{
// NewCollectorService returns an instance of CollectorService.
func NewCollectorService(url string, logger *zap.Logger) CollectorService {
return &collectorService{
url: url,
logger: logger,
}
Expand All @@ -41,7 +41,7 @@ func getSamplingURL(url string) string {
}

// GetSamplingRate returns the sampling rate for the service-operation from the agent service.
func (s *agentService) GetSamplingRate(service, operation string) (float64, error) {
func (s *collectorService) GetSamplingRate(service, operation string) (float64, error) {
url := fmt.Sprintf(getSamplingURL(s.url), getTracerServiceName(service))
resp, err := http.Get(url)
if err != nil {
Expand Down
6 changes: 3 additions & 3 deletions crossdock/services/agent_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -82,11 +82,11 @@ func TestGetSamplingRate(t *testing.T) {
defer server.Close()

// Test with no http server
agent := NewAgentService("", zap.NewNop())
agent := NewCollectorService("", zap.NewNop())
_, err := agent.GetSamplingRate("svc", "op")
require.Error(t, err)

agent = NewAgentService(server.URL, zap.NewNop())
agent = NewCollectorService(server.URL, zap.NewNop())
rate, err := agent.GetSamplingRate("svc", "op")
require.NoError(t, err)
assert.InDelta(t, 1.0, rate, 0.01)
Expand All @@ -100,7 +100,7 @@ func TestGetSamplingRateReadAllErr(t *testing.T) {
w.Header().Set("Content-Length", "1")
}))
defer server.Close()
agent := NewAgentService(server.URL, zap.NewNop())
agent := NewCollectorService(server.URL, zap.NewNop())
_, err := agent.GetSamplingRate("svc", "op")
require.EqualError(t, err, "unexpected EOF")
}
4 changes: 2 additions & 2 deletions crossdock/services/tracehandler.go
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ type testFunc func(service string, request *traceRequest) ([]*ui.Trace, error)
// TraceHandler handles creating traces and verifying them
type TraceHandler struct {
query QueryService
agent AgentService
agent CollectorService
logger *zap.Logger
getClientURL func(service string) string
getTags func() map[string]string
Expand All @@ -57,7 +57,7 @@ type TraceHandler struct {
}

// NewTraceHandler returns a TraceHandler that can create traces and verify them
func NewTraceHandler(query QueryService, agent AgentService, logger *zap.Logger) *TraceHandler {
func NewTraceHandler(query QueryService, agent CollectorService, logger *zap.Logger) *TraceHandler {
return &TraceHandler{
query: query,
agent: agent,
Expand Down

0 comments on commit 5f80b28

Please sign in to comment.