Change context to correct package (#7331)

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>

go.mod

@@ -41,7 +41,6 @@ require (
 	go.etcd.io/etcd/server/v3 v3.5.0
 	go.uber.org/zap v1.17.0
 	golang.org/x/exp v0.0.0-20200224162631-6cc2880d07d6
-	golang.org/x/net v0.0.0-20210405180319-a5a99cb37ef4
 	google.golang.org/grpc v1.38.0
 	gopkg.in/natefinch/lumberjack.v2 v2.0.0
 )

internal/datacoord/cluster.go

@@ -12,6 +12,7 @@
 package datacoord
 
 import (
+	"context"
 	"fmt"
 	"sync"
 	"time"
@@ -25,7 +26,6 @@ import (
 	"github.com/milvus-io/milvus/internal/proto/datapb"
 	"github.com/milvus-io/milvus/internal/types"
 	"go.uber.org/zap"
-	"golang.org/x/net/context"
 )
 
 const clusterPrefix = "cluster-prefix/"

internal/util/sessionutil/session_util_test.go

@@ -1,6 +1,7 @@
 package sessionutil
 
 import (
+	"context"
 	"fmt"
 	"math/rand"
 	"strconv"
@@ -12,7 +13,6 @@ import (
 	etcdkv "github.com/milvus-io/milvus/internal/kv/etcd"
 	"github.com/milvus-io/milvus/internal/util/paramtable"
 	"github.com/stretchr/testify/assert"
-	"golang.org/x/net/context"
 )
 
 var Params paramtable.BaseTable