fix(agent): fixing connection error check (#3180)

* fix(agent): fixing connection error check * fix(agent): fixing connection error check
kubeshop · Sep 22, 2023 · 488a1ce · 488a1ce
1 parent 94d1c7f
commit 488a1ce
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/agent/workers/poller.go b/agent/workers/poller.go
@@ -10,9 +10,9 @@ import (
 	"github.com/fluidtruck/deepcopy"
 	"github.com/kubeshop/tracetest/agent/client"
 	"github.com/kubeshop/tracetest/agent/proto"
-	"github.com/kubeshop/tracetest/agent/workers/datastores/connection"
 	"github.com/kubeshop/tracetest/server/datastore"
 	"github.com/kubeshop/tracetest/server/tracedb"
+	"github.com/kubeshop/tracetest/server/tracedb/connection"
 	"github.com/kubeshop/tracetest/server/traces"
 	"go.opentelemetry.io/otel/trace"
 )

diff --git a/agent/workers/poller/inmemory_datastore.go b/agent/workers/poller/inmemory_datastore.go
@@ -4,9 +4,9 @@ import (
 	"context"
 
 	"github.com/kubeshop/tracetest/agent/collector"
-	"github.com/kubeshop/tracetest/agent/workers/datastores/connection"
 	"github.com/kubeshop/tracetest/server/pkg/id"
 	"github.com/kubeshop/tracetest/server/tracedb"
+	"github.com/kubeshop/tracetest/server/tracedb/connection"
 	"github.com/kubeshop/tracetest/server/traces"
 	"go.opentelemetry.io/otel/trace"
 )