diff --git a/internal/util/rocksmq/client/rocksmq/client_impl.go b/internal/util/rocksmq/client/rocksmq/client_impl.go index 5318bcf95d..c90d422f98 100644 --- a/internal/util/rocksmq/client/rocksmq/client_impl.go +++ b/internal/util/rocksmq/client/rocksmq/client_impl.go @@ -17,6 +17,7 @@ import ( "github.com/milvus-io/milvus/internal/log" server "github.com/milvus-io/milvus/internal/util/rocksmq/server/rocksmq" + "go.uber.org/zap" ) diff --git a/internal/util/rocksmq/client/rocksmq/consumer_impl.go b/internal/util/rocksmq/client/rocksmq/consumer_impl.go index 132a4df4ba..4821dd669b 100644 --- a/internal/util/rocksmq/client/rocksmq/consumer_impl.go +++ b/internal/util/rocksmq/client/rocksmq/consumer_impl.go @@ -15,6 +15,7 @@ import ( "sync" "github.com/milvus-io/milvus/internal/log" + "go.uber.org/zap" ) diff --git a/internal/util/rocksmq/client/rocksmq/producer_impl.go b/internal/util/rocksmq/client/rocksmq/producer_impl.go index 8fd9f5de90..4eecc2a35b 100644 --- a/internal/util/rocksmq/client/rocksmq/producer_impl.go +++ b/internal/util/rocksmq/client/rocksmq/producer_impl.go @@ -14,6 +14,7 @@ package rocksmq import ( "github.com/milvus-io/milvus/internal/log" server "github.com/milvus-io/milvus/internal/util/rocksmq/server/rocksmq" + "go.uber.org/zap" ) diff --git a/internal/util/rocksmq/client/rocksmq/test_helper.go b/internal/util/rocksmq/client/rocksmq/test_helper.go index d5bd434338..56a3504063 100644 --- a/internal/util/rocksmq/client/rocksmq/test_helper.go +++ b/internal/util/rocksmq/client/rocksmq/test_helper.go @@ -21,6 +21,7 @@ import ( "github.com/milvus-io/milvus/internal/log" rocksmq "github.com/milvus-io/milvus/internal/util/rocksmq/server/rocksmq" server "github.com/milvus-io/milvus/internal/util/rocksmq/server/rocksmq" + "go.uber.org/zap" ) diff --git a/internal/util/rocksmq/server/rocksmq/global_rmq.go b/internal/util/rocksmq/server/rocksmq/global_rmq.go index 4e6a862bf6..86912a811b 100644 --- a/internal/util/rocksmq/server/rocksmq/global_rmq.go +++ b/internal/util/rocksmq/server/rocksmq/global_rmq.go @@ -18,11 +18,11 @@ import ( "sync/atomic" "github.com/milvus-io/milvus/internal/allocator" + rocksdbkv "github.com/milvus-io/milvus/internal/kv/rocksdb" "github.com/milvus-io/milvus/internal/log" "github.com/milvus-io/milvus/internal/util/paramtable" - "go.uber.org/zap" - rocksdbkv "github.com/milvus-io/milvus/internal/kv/rocksdb" + "go.uber.org/zap" ) // Rmq is global rocksmq instance that will be initialized only once diff --git a/internal/util/rocksmq/server/rocksmq/global_rmq_test.go b/internal/util/rocksmq/server/rocksmq/global_rmq_test.go index db0e91ac33..257b2d49e1 100644 --- a/internal/util/rocksmq/server/rocksmq/global_rmq_test.go +++ b/internal/util/rocksmq/server/rocksmq/global_rmq_test.go @@ -18,8 +18,8 @@ import ( "testing" "github.com/milvus-io/milvus/internal/allocator" - etcdkv "github.com/milvus-io/milvus/internal/kv/etcd" + "github.com/stretchr/testify/assert" ) diff --git a/internal/util/rocksmq/server/rocksmq/rocksmq_impl.go b/internal/util/rocksmq/server/rocksmq/rocksmq_impl.go index be2aaad422..f1eb96e5fe 100644 --- a/internal/util/rocksmq/server/rocksmq/rocksmq_impl.go +++ b/internal/util/rocksmq/server/rocksmq/rocksmq_impl.go @@ -22,12 +22,12 @@ import ( "github.com/milvus-io/milvus/internal/allocator" "github.com/milvus-io/milvus/internal/kv" + rocksdbkv "github.com/milvus-io/milvus/internal/kv/rocksdb" "github.com/milvus-io/milvus/internal/log" "github.com/milvus-io/milvus/internal/util/typeutil" + "github.com/tecbot/gorocksdb" "go.uber.org/zap" - - rocksdbkv "github.com/milvus-io/milvus/internal/kv/rocksdb" ) // UniqueID is the type of message ID diff --git a/internal/util/rocksmq/server/rocksmq/rocksmq_impl_test.go b/internal/util/rocksmq/server/rocksmq/rocksmq_impl_test.go index b369a1718d..afd7196656 100644 --- a/internal/util/rocksmq/server/rocksmq/rocksmq_impl_test.go +++ b/internal/util/rocksmq/server/rocksmq/rocksmq_impl_test.go @@ -21,10 +21,10 @@ import ( "time" "github.com/milvus-io/milvus/internal/allocator" - "github.com/milvus-io/milvus/internal/util/paramtable" - etcdkv "github.com/milvus-io/milvus/internal/kv/etcd" rocksdbkv "github.com/milvus-io/milvus/internal/kv/rocksdb" + "github.com/milvus-io/milvus/internal/util/paramtable" + "github.com/stretchr/testify/assert" ) diff --git a/internal/util/rocksmq/server/rocksmq/rocksmq_retention.go b/internal/util/rocksmq/server/rocksmq/rocksmq_retention.go index 58844d4491..d2ef07c005 100644 --- a/internal/util/rocksmq/server/rocksmq/rocksmq_retention.go +++ b/internal/util/rocksmq/server/rocksmq/rocksmq_retention.go @@ -21,6 +21,7 @@ import ( rocksdbkv "github.com/milvus-io/milvus/internal/kv/rocksdb" "github.com/milvus-io/milvus/internal/log" + "github.com/tecbot/gorocksdb" "go.uber.org/zap" ) diff --git a/internal/util/rocksmq/server/rocksmq/rocksmq_retention_test.go b/internal/util/rocksmq/server/rocksmq/rocksmq_retention_test.go index 55347fad8a..57496e7839 100644 --- a/internal/util/rocksmq/server/rocksmq/rocksmq_retention_test.go +++ b/internal/util/rocksmq/server/rocksmq/rocksmq_retention_test.go @@ -20,6 +20,7 @@ import ( "time" "github.com/milvus-io/milvus/internal/log" + "github.com/stretchr/testify/assert" "go.uber.org/zap" )