Repository: milvus-io/milvus-sdk-go
Pull request #461 (2023-05-23 04:12:28)

Support JSON datatype
See also #432
Related to milvus-io/milvus#23389
/kind enhancement

sre-ci-robot: [APPROVALNOTIFIER] This PR is **APPROVED**

This pull-request has been approved by: *<a href="https://github.com/milvus-io/milvus-sdk-go/pull/461#" title="Author self-approved">congqixia</a>*

The full list of commands accepted by this bot can be found [here](https://go.k8s.io/bot-commands?repo=milvus-io%2Fmilvus-sdk-go).

The pull request process is described [here](https://git.k8s.io/community/contributors/guide/owners.md#the-code-review-process)

<details >
Needs approval from an approver in each of these files:

- ~~[OWNERS](https://github.com/milvus-io/milvus-sdk-go/blob/v2.2.x/OWNERS)~~ [congqixia]

Approvers can indicate their approval by writing `/approve` in a comment
Approvers can cancel approval by writing `/approve cancel` in a comment
</details>
<!-- META={"approvers":[]} -->
codecov[bot]: ## [Codecov](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/461?src=pr&el=h1&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io) Report
> Merging [#461](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/461?src=pr&el=desc&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io) (0cff0a4) into [v2.2.x](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/commit/8b02b8e5905d0a88baee8c0daac17107e536a706?el=desc&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io) (8b02b8e) will **increase** coverage by `0.05%`.
> The diff coverage is `91.17%`.

[![Impacted file tree graph](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/461/graphs/tree.svg?width=650&height=150&src=pr&token=WJK43PV390&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io)](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/461?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io)

```diff
@@            Coverage Diff             @@
##           v2.2.x     #461      +/-   ##
==========================================
+ Coverage   86.69%   86.74%   +0.05%     
==========================================
  Files          33       34       +1     
  Lines        2932     2966      +34     
==========================================
+ Hits         2542     2573      +31     
- Misses        300      303       +3     
  Partials       90       90              
```


| [Impacted Files](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/461?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io) | Coverage Δ | |
|---|---|---|
| [client/admin.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/461?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L2FkbWluLmdv) | `83.33% <ø> (ø)` | |
| [client/alias.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/461?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L2FsaWFzLmdv) | `100.00% <ø> (ø)` | |
| [client/authentication.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/461?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L2F1dGhlbnRpY2F0aW9uLmdv) | `100.00% <ø> (ø)` | |
| [client/collection.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/461?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L2NvbGxlY3Rpb24uZ28=) | `85.40% <ø> (ø)` | |
| [client/data.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/461?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L2RhdGEuZ28=) | `72.72% <ø> (ø)` | |
| [client/database.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/461?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L2RhdGFiYXNlLmdv) | `92.85% <ø> (ø)` | |
| [client/index.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/461?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L2luZGV4Lmdv) | `77.95% <ø> (ø)` | |
| [client/maintainance.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/461?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L21haW50YWluYW5jZS5nbw==) | `95.55% <ø> (ø)` | |
| [client/options.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/461?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L29wdGlvbnMuZ28=) | `79.16% <ø> (ø)` | |
| [client/partition.go](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/461?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io#diff-Y2xpZW50L3BhcnRpdGlvbi5nbw==) | `82.67% <ø> (ø)` | |
| ... and [5 more](https://app.codecov.io/gh/milvus-io/milvus-sdk-go/pull/461?src=pr&el=tree-more&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=milvus-io) | |

yah01: /lgtm

Patch:
diff --git a/client/client_grpc_collection.go b/client/collection.go
similarity index 100%
rename from client/client_grpc_collection.go
rename to client/collection.go
diff --git a/client/client_grpc_collection_test.go b/client/collection_test.go
similarity index 100%
rename from client/client_grpc_collection_test.go
rename to client/collection_test.go
diff --git a/client/client_grpc_data.go b/client/data.go
similarity index 100%
rename from client/client_grpc_data.go
rename to client/data.go
diff --git a/client/client_grpc_data_test.go b/client/data_test.go
similarity index 100%
rename from client/client_grpc_data_test.go
rename to client/data_test.go
diff --git a/client/client_grpc_database.go b/client/database.go
similarity index 100%
rename from client/client_grpc_database.go
rename to client/database.go
diff --git a/client/client_grpc_database_test.go b/client/database_test.go
similarity index 100%
rename from client/client_grpc_database_test.go
rename to client/database_test.go
diff --git a/client/client_grpc_index.go b/client/index.go
similarity index 100%
rename from client/client_grpc_index.go
rename to client/index.go
diff --git a/client/client_grpc_index_test.go b/client/index_test.go
similarity index 100%
rename from client/client_grpc_index_test.go
rename to client/index_test.go
diff --git a/client/client_grpc_control.go b/client/maintainance.go
similarity index 100%
rename from client/client_grpc_control.go
rename to client/maintainance.go
diff --git a/client/client_grpc_control_test.go b/client/maintainance_test.go
similarity index 100%
rename from client/client_grpc_control_test.go
rename to client/maintainance_test.go
diff --git a/client/client_grpc_options.go b/client/options.go
similarity index 100%
rename from client/client_grpc_options.go
rename to client/options.go
diff --git a/client/client_grpc_options_test.go b/client/options_test.go
similarity index 100%
rename from client/client_grpc_options_test.go
rename to client/options_test.go
diff --git a/client/client_grpc_partition.go b/client/partition.go
similarity index 100%
rename from client/client_grpc_partition.go
rename to client/partition.go
diff --git a/client/client_grpc_partition_test.go b/client/partition_test.go
similarity index 100%
rename from client/client_grpc_partition_test.go
rename to client/partition_test.go
diff --git a/client/client_grpc_rbac.go b/client/rbac.go
similarity index 100%
rename from client/client_grpc_rbac.go
rename to client/rbac.go
diff --git a/client/client_grpc_rbac_test.go b/client/rbac_test.go
similarity index 100%
rename from client/client_grpc_rbac_test.go
rename to client/rbac_test.go
diff --git a/client/client_grpc_rg.go b/client/resource_group.go
similarity index 100%
rename from client/client_grpc_rg.go
rename to client/resource_group.go
diff --git a/client/client_grpc_rg_test.go b/client/resource_group_test.go
similarity index 100%
rename from client/client_grpc_rg_test.go
rename to client/resource_group_test.go
diff --git a/client/client_grpc_row.go b/client/row.go
similarity index 100%
rename from client/client_grpc_row.go
rename to client/row.go
diff --git a/client/client_grpc_row_test.go b/client/row_test.go
similarity index 100%
rename from client/client_grpc_row_test.go
rename to client/row_test.go
diff --git a/entity/columns_json.go b/entity/columns_json.go
new file mode 100644
index 0000000..5f3f9f2
--- /dev/null
+++ b/entity/columns_json.go
@@ -0,0 +1,92 @@
+package entity
+
+import (
+	"fmt"
+
+	"github.com/cockroachdb/errors"
+	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+)
+
+var _ (Column) = (*ColumnJSONBytes)(nil)
+
+// ColumnJSONBytes column type for JSON.
+// all items are marshaled json bytes.
+type ColumnJSONBytes struct {
+	name   string
+	values [][]byte
+}
+
+// Name returns column name.
+func (c *ColumnJSONBytes) Name() string {
+	return c.name
+}
+
+// Type returns column FieldType.
+func (c *ColumnJSONBytes) Type() FieldType {
+	return FieldTypeJSON
+}
+
+// Len returns column values length.
+func (c *ColumnJSONBytes) Len() int {
+	return len(c.values)
+}
+
+// Get returns value at index as interface{}.
+func (c *ColumnJSONBytes) Get(idx int) (interface{}, error) {
+	if idx < 0 || idx > c.Len() {
+		return nil, errors.New("index out of range")
+	}
+	return c.values[idx], nil
+}
+
+// FieldData return column data mapped to schema.FieldData.
+func (c *ColumnJSONBytes) FieldData() *schema.FieldData {
+	fd := &schema.FieldData{
+		Type:      schema.DataType_JSON,
+		FieldName: c.name,
+	}
+
+	fd.Field = &schema.FieldData_Scalars{
+		Scalars: &schema.ScalarField{
+			Data: &schema.ScalarField_JsonData{
+				JsonData: &schema.JSONArray{
+					Data: c.values,
+				},
+			},
+		},
+	}
+
+	return fd
+}
+
+// ValueByIdx returns value of the provided index.
+func (c *ColumnJSONBytes) ValueByIdx(idx int) ([]byte, error) {
+	if idx < 0 || idx >= c.Len() {
+		return nil, errors.New("index out of range")
+	}
+	return c.values[idx], nil
+}
+
+// AppendValue append value into column.
+func (c *ColumnJSONBytes) AppendValue(i interface{}) error {
+	v, ok := i.([]byte)
+	if !ok {
+		return fmt.Errorf("invalid type, expected []byte, got %T", i)
+	}
+	c.values = append(c.values, v)
+
+	return nil
+}
+
+// Data returns column data.
+func (c *ColumnJSONBytes) Data() [][]byte {
+	return c.values
+}
+
+// NewColumnJSONBytes composes a Column with json bytes.
+func NewColumnJSONBytes(name string, values [][]byte) *ColumnJSONBytes {
+	return &ColumnJSONBytes{
+		name:   name,
+		values: values,
+	}
+}
diff --git a/entity/schema.go b/entity/schema.go
index e1ca53c..a412aed 100644
--- a/entity/schema.go
+++ b/entity/schema.go
@@ -163,6 +163,8 @@ func (t FieldType) Name() string {
 		return "String"
 	case FieldTypeVarChar:
 		return "VarChar"
+	case FieldTypeJSON:
+		return "JSON"
 	case FieldTypeBinaryVector:
 		return "BinaryVector"
 	case FieldTypeFloatVector:
@@ -193,6 +195,8 @@ func (t FieldType) String() string {
 		return "string"
 	case FieldTypeVarChar:
 		return "string"
+	case FieldTypeJSON:
+		return "JSON"
 	case FieldTypeBinaryVector:
 		return "[]byte"
 	case FieldTypeFloatVector:
@@ -223,6 +227,8 @@ func (t FieldType) PbFieldType() (string, string) {
 		return "String", "string"
 	case FieldTypeVarChar:
 		return "VarChar", "string"
+	case FieldTypeJSON:
+		return "JSON", "JSON"
 	case FieldTypeBinaryVector:
 		return "[]byte", ""
 	case FieldTypeFloatVector:
@@ -256,6 +262,9 @@ const (
 	FieldTypeString FieldType = 20
 	// FieldTypeVarChar field type varchar
 	FieldTypeVarChar FieldType = 21 // variable-length strings with a specified maximum length
+	// FieldTypeArray FieldType = 22
+	// FieldTypeJSON field type JSON
+	FieldTypeJSON FieldType = 23
 	// FieldTypeBinaryVector field type binary vector
 	FieldTypeBinaryVector FieldType = 100
 	// FieldTypeFloatVector field type float vector
diff --git a/go.mod b/go.mod
index 510d38c..a14dc88 100644
--- a/go.mod
+++ b/go.mod
@@ -7,7 +7,7 @@ require (
 	github.com/go-faker/faker/v4 v4.1.0
 	github.com/golang/protobuf v1.5.2
 	github.com/grpc-ecosystem/go-grpc-middleware v1.3.0
-	github.com/milvus-io/milvus-proto/go-api v0.0.0-20230426074830-1b5e205e11ab
+	github.com/milvus-io/milvus-proto/go-api v0.0.0-20230522080721-2975bfe7a190
 	github.com/stretchr/testify v1.8.1
 	google.golang.org/grpc v1.48.0
 	google.golang.org/grpc/examples v0.0.0-20220617181431-3e7b97febc7f
@@ -29,6 +29,6 @@ require (
 	golang.org/x/sys v0.0.0-20220422013727-9388b58f7150 // indirect
 	golang.org/x/text v0.3.7 // indirect
 	google.golang.org/genproto v0.0.0-20220503193339-ba3ae3f07e29 // indirect
-	google.golang.org/protobuf v1.28.1 // indirect
+	google.golang.org/protobuf v1.28.1
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )
diff --git a/go.sum b/go.sum
index 4aa9f80..bce7e80 100644
--- a/go.sum
+++ b/go.sum
@@ -159,6 +159,8 @@ github.com/mediocregopher/radix/v3 v3.4.2/go.mod h1:8FL3F6UQRXHXIBSPUs5h0RybMF8i
 github.com/microcosm-cc/bluemonday v1.0.2/go.mod h1:iVP4YcDBq+n/5fb23BhYFvIMq/leAFZyRl6bYmGDlGc=
 github.com/milvus-io/milvus-proto/go-api v0.0.0-20230426074830-1b5e205e11ab h1:8XkGKAgMvftgknyMoow5DmIjHZ28A7+U5tECi+MqJmk=
 github.com/milvus-io/milvus-proto/go-api v0.0.0-20230426074830-1b5e205e11ab/go.mod h1:148qnlmZ0Fdm1Fq+Mj/OW2uDoEP25g3mjh0vMGtkgmk=
+github.com/milvus-io/milvus-proto/go-api v0.0.0-20230522080721-2975bfe7a190 h1:ZREJhOMgAvXs+K0ain51ibxAtCB8Lnn3EBZFHEykIlk=
+github.com/milvus-io/milvus-proto/go-api v0.0.0-20230522080721-2975bfe7a190/go.mod h1:148qnlmZ0Fdm1Fq+Mj/OW2uDoEP25g3mjh0vMGtkgmk=
 github.com/mitchellh/go-homedir v1.1.0/go.mod h1:SfyaCUpYCn1Vlf4IUYiD9fPX4A5wJrkLzIz1N1q0pr0=
 github.com/mitchellh/mapstructure v1.1.2/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y=
 github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
diff --git a/mocks/MilvusServiceServer.go b/mocks/MilvusServiceServer.go
index 9eaaebd..edf238a 100644
--- a/mocks/MilvusServiceServer.go
+++ b/mocks/MilvusServiceServer.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.26.1. DO NOT EDIT.
+// Code generated by mockery v2.16.0. DO NOT EDIT.
 
 package mocks
 
@@ -30,10 +30,6 @@ func (_m *MilvusServiceServer) AlterAlias(_a0 context.Context, _a1 *milvuspb.Alt
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterAliasRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterAliasRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -42,6 +38,7 @@ func (_m *MilvusServiceServer) AlterAlias(_a0 context.Context, _a1 *milvuspb.Alt
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.AlterAliasRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -57,8 +54,8 @@ type MilvusServiceServer_AlterAlias_Call struct {
 }
 
 // AlterAlias is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.AlterAliasRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.AlterAliasRequest
 func (_e *MilvusServiceServer_Expecter) AlterAlias(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_AlterAlias_Call {
 	return &MilvusServiceServer_AlterAlias_Call{Call: _e.mock.On("AlterAlias", _a0, _a1)}
 }
@@ -75,20 +72,11 @@ func (_c *MilvusServiceServer_AlterAlias_Call) Return(_a0 *commonpb.Status, _a1
 	return _c
 }
 
-func (_c *MilvusServiceServer_AlterAlias_Call) RunAndReturn(run func(context.Context, *milvuspb.AlterAliasRequest) (*commonpb.Status, error)) *MilvusServiceServer_AlterAlias_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // AlterCollection provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) AlterCollection(_a0 context.Context, _a1 *milvuspb.AlterCollectionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterCollectionRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterCollectionRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -97,6 +85,7 @@ func (_m *MilvusServiceServer) AlterCollection(_a0 context.Context, _a1 *milvusp
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.AlterCollectionRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -112,8 +101,8 @@ type MilvusServiceServer_AlterCollection_Call struct {
 }
 
 // AlterCollection is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.AlterCollectionRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.AlterCollectionRequest
 func (_e *MilvusServiceServer_Expecter) AlterCollection(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_AlterCollection_Call {
 	return &MilvusServiceServer_AlterCollection_Call{Call: _e.mock.On("AlterCollection", _a0, _a1)}
 }
@@ -130,20 +119,11 @@ func (_c *MilvusServiceServer_AlterCollection_Call) Return(_a0 *commonpb.Status,
 	return _c
 }
 
-func (_c *MilvusServiceServer_AlterCollection_Call) RunAndReturn(run func(context.Context, *milvuspb.AlterCollectionRequest) (*commonpb.Status, error)) *MilvusServiceServer_AlterCollection_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // CalcDistance provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) CalcDistance(_a0 context.Context, _a1 *milvuspb.CalcDistanceRequest) (*milvuspb.CalcDistanceResults, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.CalcDistanceResults
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CalcDistanceRequest) (*milvuspb.CalcDistanceResults, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CalcDistanceRequest) *milvuspb.CalcDistanceResults); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -152,6 +132,7 @@ func (_m *MilvusServiceServer) CalcDistance(_a0 context.Context, _a1 *milvuspb.C
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CalcDistanceRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -167,8 +148,8 @@ type MilvusServiceServer_CalcDistance_Call struct {
 }
 
 // CalcDistance is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.CalcDistanceRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.CalcDistanceRequest
 func (_e *MilvusServiceServer_Expecter) CalcDistance(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_CalcDistance_Call {
 	return &MilvusServiceServer_CalcDistance_Call{Call: _e.mock.On("CalcDistance", _a0, _a1)}
 }
@@ -185,20 +166,11 @@ func (_c *MilvusServiceServer_CalcDistance_Call) Return(_a0 *milvuspb.CalcDistan
 	return _c
 }
 
-func (_c *MilvusServiceServer_CalcDistance_Call) RunAndReturn(run func(context.Context, *milvuspb.CalcDistanceRequest) (*milvuspb.CalcDistanceResults, error)) *MilvusServiceServer_CalcDistance_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // CheckHealth provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) CheckHealth(_a0 context.Context, _a1 *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.CheckHealthResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest) *milvuspb.CheckHealthResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -207,6 +179,7 @@ func (_m *MilvusServiceServer) CheckHealth(_a0 context.Context, _a1 *milvuspb.Ch
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CheckHealthRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -222,8 +195,8 @@ type MilvusServiceServer_CheckHealth_Call struct {
 }
 
 // CheckHealth is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.CheckHealthRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.CheckHealthRequest
 func (_e *MilvusServiceServer_Expecter) CheckHealth(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_CheckHealth_Call {
 	return &MilvusServiceServer_CheckHealth_Call{Call: _e.mock.On("CheckHealth", _a0, _a1)}
 }
@@ -240,8 +213,50 @@ func (_c *MilvusServiceServer_CheckHealth_Call) Return(_a0 *milvuspb.CheckHealth
 	return _c
 }
 
-func (_c *MilvusServiceServer_CheckHealth_Call) RunAndReturn(run func(context.Context, *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error)) *MilvusServiceServer_CheckHealth_Call {
-	_c.Call.Return(run)
+// Connect provides a mock function with given fields: _a0, _a1
+func (_m *MilvusServiceServer) Connect(_a0 context.Context, _a1 *milvuspb.ConnectRequest) (*milvuspb.ConnectResponse, error) {
+	ret := _m.Called(_a0, _a1)
+
+	var r0 *milvuspb.ConnectResponse
+	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ConnectRequest) *milvuspb.ConnectResponse); ok {
+		r0 = rf(_a0, _a1)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(*milvuspb.ConnectResponse)
+		}
+	}
+
+	var r1 error
+	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ConnectRequest) error); ok {
+		r1 = rf(_a0, _a1)
+	} else {
+		r1 = ret.Error(1)
+	}
+
+	return r0, r1
+}
+
+// MilvusServiceServer_Connect_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Connect'
+type MilvusServiceServer_Connect_Call struct {
+	*mock.Call
+}
+
+// Connect is a helper method to define mock.On call
+//  - _a0 context.Context
+//  - _a1 *milvuspb.ConnectRequest
+func (_e *MilvusServiceServer_Expecter) Connect(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_Connect_Call {
+	return &MilvusServiceServer_Connect_Call{Call: _e.mock.On("Connect", _a0, _a1)}
+}
+
+func (_c *MilvusServiceServer_Connect_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.ConnectRequest)) *MilvusServiceServer_Connect_Call {
+	_c.Call.Run(func(args mock.Arguments) {
+		run(args[0].(context.Context), args[1].(*milvuspb.ConnectRequest))
+	})
+	return _c
+}
+
+func (_c *MilvusServiceServer_Connect_Call) Return(_a0 *milvuspb.ConnectResponse, _a1 error) *MilvusServiceServer_Connect_Call {
+	_c.Call.Return(_a0, _a1)
 	return _c
 }
 
@@ -250,10 +265,6 @@ func (_m *MilvusServiceServer) CreateAlias(_a0 context.Context, _a1 *milvuspb.Cr
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateAliasRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateAliasRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -262,6 +273,7 @@ func (_m *MilvusServiceServer) CreateAlias(_a0 context.Context, _a1 *milvuspb.Cr
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateAliasRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -277,8 +289,8 @@ type MilvusServiceServer_CreateAlias_Call struct {
 }
 
 // CreateAlias is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.CreateAliasRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.CreateAliasRequest
 func (_e *MilvusServiceServer_Expecter) CreateAlias(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_CreateAlias_Call {
 	return &MilvusServiceServer_CreateAlias_Call{Call: _e.mock.On("CreateAlias", _a0, _a1)}
 }
@@ -295,20 +307,11 @@ func (_c *MilvusServiceServer_CreateAlias_Call) Return(_a0 *commonpb.Status, _a1
 	return _c
 }
 
-func (_c *MilvusServiceServer_CreateAlias_Call) RunAndReturn(run func(context.Context, *milvuspb.CreateAliasRequest) (*commonpb.Status, error)) *MilvusServiceServer_CreateAlias_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // CreateCollection provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) CreateCollection(_a0 context.Context, _a1 *milvuspb.CreateCollectionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateCollectionRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateCollectionRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -317,6 +320,7 @@ func (_m *MilvusServiceServer) CreateCollection(_a0 context.Context, _a1 *milvus
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateCollectionRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -332,8 +336,8 @@ type MilvusServiceServer_CreateCollection_Call struct {
 }
 
 // CreateCollection is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.CreateCollectionRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.CreateCollectionRequest
 func (_e *MilvusServiceServer_Expecter) CreateCollection(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_CreateCollection_Call {
 	return &MilvusServiceServer_CreateCollection_Call{Call: _e.mock.On("CreateCollection", _a0, _a1)}
 }
@@ -350,20 +354,11 @@ func (_c *MilvusServiceServer_CreateCollection_Call) Return(_a0 *commonpb.Status
 	return _c
 }
 
-func (_c *MilvusServiceServer_CreateCollection_Call) RunAndReturn(run func(context.Context, *milvuspb.CreateCollectionRequest) (*commonpb.Status, error)) *MilvusServiceServer_CreateCollection_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // CreateCredential provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) CreateCredential(_a0 context.Context, _a1 *milvuspb.CreateCredentialRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateCredentialRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateCredentialRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -372,6 +367,7 @@ func (_m *MilvusServiceServer) CreateCredential(_a0 context.Context, _a1 *milvus
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateCredentialRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -387,8 +383,8 @@ type MilvusServiceServer_CreateCredential_Call struct {
 }
 
 // CreateCredential is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.CreateCredentialRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.CreateCredentialRequest
 func (_e *MilvusServiceServer_Expecter) CreateCredential(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_CreateCredential_Call {
 	return &MilvusServiceServer_CreateCredential_Call{Call: _e.mock.On("CreateCredential", _a0, _a1)}
 }
@@ -405,20 +401,11 @@ func (_c *MilvusServiceServer_CreateCredential_Call) Return(_a0 *commonpb.Status
 	return _c
 }
 
-func (_c *MilvusServiceServer_CreateCredential_Call) RunAndReturn(run func(context.Context, *milvuspb.CreateCredentialRequest) (*commonpb.Status, error)) *MilvusServiceServer_CreateCredential_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // CreateDatabase provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) CreateDatabase(_a0 context.Context, _a1 *milvuspb.CreateDatabaseRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateDatabaseRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateDatabaseRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -427,6 +414,7 @@ func (_m *MilvusServiceServer) CreateDatabase(_a0 context.Context, _a1 *milvuspb
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateDatabaseRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -442,8 +430,8 @@ type MilvusServiceServer_CreateDatabase_Call struct {
 }
 
 // CreateDatabase is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.CreateDatabaseRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.CreateDatabaseRequest
 func (_e *MilvusServiceServer_Expecter) CreateDatabase(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_CreateDatabase_Call {
 	return &MilvusServiceServer_CreateDatabase_Call{Call: _e.mock.On("CreateDatabase", _a0, _a1)}
 }
@@ -460,20 +448,11 @@ func (_c *MilvusServiceServer_CreateDatabase_Call) Return(_a0 *commonpb.Status,
 	return _c
 }
 
-func (_c *MilvusServiceServer_CreateDatabase_Call) RunAndReturn(run func(context.Context, *milvuspb.CreateDatabaseRequest) (*commonpb.Status, error)) *MilvusServiceServer_CreateDatabase_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // CreateIndex provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) CreateIndex(_a0 context.Context, _a1 *milvuspb.CreateIndexRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateIndexRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateIndexRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -482,6 +461,7 @@ func (_m *MilvusServiceServer) CreateIndex(_a0 context.Context, _a1 *milvuspb.Cr
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateIndexRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -497,8 +477,8 @@ type MilvusServiceServer_CreateIndex_Call struct {
 }
 
 // CreateIndex is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.CreateIndexRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.CreateIndexRequest
 func (_e *MilvusServiceServer_Expecter) CreateIndex(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_CreateIndex_Call {
 	return &MilvusServiceServer_CreateIndex_Call{Call: _e.mock.On("CreateIndex", _a0, _a1)}
 }
@@ -515,20 +495,11 @@ func (_c *MilvusServiceServer_CreateIndex_Call) Return(_a0 *commonpb.Status, _a1
 	return _c
 }
 
-func (_c *MilvusServiceServer_CreateIndex_Call) RunAndReturn(run func(context.Context, *milvuspb.CreateIndexRequest) (*commonpb.Status, error)) *MilvusServiceServer_CreateIndex_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // CreatePartition provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) CreatePartition(_a0 context.Context, _a1 *milvuspb.CreatePartitionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreatePartitionRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreatePartitionRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -537,6 +508,7 @@ func (_m *MilvusServiceServer) CreatePartition(_a0 context.Context, _a1 *milvusp
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreatePartitionRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -552,8 +524,8 @@ type MilvusServiceServer_CreatePartition_Call struct {
 }
 
 // CreatePartition is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.CreatePartitionRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.CreatePartitionRequest
 func (_e *MilvusServiceServer_Expecter) CreatePartition(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_CreatePartition_Call {
 	return &MilvusServiceServer_CreatePartition_Call{Call: _e.mock.On("CreatePartition", _a0, _a1)}
 }
@@ -570,20 +542,11 @@ func (_c *MilvusServiceServer_CreatePartition_Call) Return(_a0 *commonpb.Status,
 	return _c
 }
 
-func (_c *MilvusServiceServer_CreatePartition_Call) RunAndReturn(run func(context.Context, *milvuspb.CreatePartitionRequest) (*commonpb.Status, error)) *MilvusServiceServer_CreatePartition_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // CreateResourceGroup provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) CreateResourceGroup(_a0 context.Context, _a1 *milvuspb.CreateResourceGroupRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateResourceGroupRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateResourceGroupRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -592,6 +555,7 @@ func (_m *MilvusServiceServer) CreateResourceGroup(_a0 context.Context, _a1 *mil
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateResourceGroupRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -607,8 +571,8 @@ type MilvusServiceServer_CreateResourceGroup_Call struct {
 }
 
 // CreateResourceGroup is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.CreateResourceGroupRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.CreateResourceGroupRequest
 func (_e *MilvusServiceServer_Expecter) CreateResourceGroup(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_CreateResourceGroup_Call {
 	return &MilvusServiceServer_CreateResourceGroup_Call{Call: _e.mock.On("CreateResourceGroup", _a0, _a1)}
 }
@@ -625,20 +589,11 @@ func (_c *MilvusServiceServer_CreateResourceGroup_Call) Return(_a0 *commonpb.Sta
 	return _c
 }
 
-func (_c *MilvusServiceServer_CreateResourceGroup_Call) RunAndReturn(run func(context.Context, *milvuspb.CreateResourceGroupRequest) (*commonpb.Status, error)) *MilvusServiceServer_CreateResourceGroup_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // CreateRole provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) CreateRole(_a0 context.Context, _a1 *milvuspb.CreateRoleRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateRoleRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateRoleRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -647,6 +602,7 @@ func (_m *MilvusServiceServer) CreateRole(_a0 context.Context, _a1 *milvuspb.Cre
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateRoleRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -662,8 +618,8 @@ type MilvusServiceServer_CreateRole_Call struct {
 }
 
 // CreateRole is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.CreateRoleRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.CreateRoleRequest
 func (_e *MilvusServiceServer_Expecter) CreateRole(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_CreateRole_Call {
 	return &MilvusServiceServer_CreateRole_Call{Call: _e.mock.On("CreateRole", _a0, _a1)}
 }
@@ -680,20 +636,11 @@ func (_c *MilvusServiceServer_CreateRole_Call) Return(_a0 *commonpb.Status, _a1
 	return _c
 }
 
-func (_c *MilvusServiceServer_CreateRole_Call) RunAndReturn(run func(context.Context, *milvuspb.CreateRoleRequest) (*commonpb.Status, error)) *MilvusServiceServer_CreateRole_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // Delete provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) Delete(_a0 context.Context, _a1 *milvuspb.DeleteRequest) (*milvuspb.MutationResult, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.MutationResult
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteRequest) (*milvuspb.MutationResult, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteRequest) *milvuspb.MutationResult); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -702,6 +649,7 @@ func (_m *MilvusServiceServer) Delete(_a0 context.Context, _a1 *milvuspb.DeleteR
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DeleteRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -717,8 +665,8 @@ type MilvusServiceServer_Delete_Call struct {
 }
 
 // Delete is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.DeleteRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.DeleteRequest
 func (_e *MilvusServiceServer_Expecter) Delete(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_Delete_Call {
 	return &MilvusServiceServer_Delete_Call{Call: _e.mock.On("Delete", _a0, _a1)}
 }
@@ -735,20 +683,11 @@ func (_c *MilvusServiceServer_Delete_Call) Return(_a0 *milvuspb.MutationResult,
 	return _c
 }
 
-func (_c *MilvusServiceServer_Delete_Call) RunAndReturn(run func(context.Context, *milvuspb.DeleteRequest) (*milvuspb.MutationResult, error)) *MilvusServiceServer_Delete_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // DeleteCredential provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) DeleteCredential(_a0 context.Context, _a1 *milvuspb.DeleteCredentialRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteCredentialRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteCredentialRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -757,6 +696,7 @@ func (_m *MilvusServiceServer) DeleteCredential(_a0 context.Context, _a1 *milvus
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DeleteCredentialRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -772,8 +712,8 @@ type MilvusServiceServer_DeleteCredential_Call struct {
 }
 
 // DeleteCredential is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.DeleteCredentialRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.DeleteCredentialRequest
 func (_e *MilvusServiceServer_Expecter) DeleteCredential(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_DeleteCredential_Call {
 	return &MilvusServiceServer_DeleteCredential_Call{Call: _e.mock.On("DeleteCredential", _a0, _a1)}
 }
@@ -790,20 +730,11 @@ func (_c *MilvusServiceServer_DeleteCredential_Call) Return(_a0 *commonpb.Status
 	return _c
 }
 
-func (_c *MilvusServiceServer_DeleteCredential_Call) RunAndReturn(run func(context.Context, *milvuspb.DeleteCredentialRequest) (*commonpb.Status, error)) *MilvusServiceServer_DeleteCredential_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // DescribeCollection provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) DescribeCollection(_a0 context.Context, _a1 *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.DescribeCollectionResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeCollectionRequest) *milvuspb.DescribeCollectionResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -812,6 +743,7 @@ func (_m *MilvusServiceServer) DescribeCollection(_a0 context.Context, _a1 *milv
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DescribeCollectionRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -827,8 +759,8 @@ type MilvusServiceServer_DescribeCollection_Call struct {
 }
 
 // DescribeCollection is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.DescribeCollectionRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.DescribeCollectionRequest
 func (_e *MilvusServiceServer_Expecter) DescribeCollection(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_DescribeCollection_Call {
 	return &MilvusServiceServer_DescribeCollection_Call{Call: _e.mock.On("DescribeCollection", _a0, _a1)}
 }
@@ -845,20 +777,11 @@ func (_c *MilvusServiceServer_DescribeCollection_Call) Return(_a0 *milvuspb.Desc
 	return _c
 }
 
-func (_c *MilvusServiceServer_DescribeCollection_Call) RunAndReturn(run func(context.Context, *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error)) *MilvusServiceServer_DescribeCollection_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // DescribeIndex provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) DescribeIndex(_a0 context.Context, _a1 *milvuspb.DescribeIndexRequest) (*milvuspb.DescribeIndexResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.DescribeIndexResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeIndexRequest) (*milvuspb.DescribeIndexResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeIndexRequest) *milvuspb.DescribeIndexResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -867,6 +790,7 @@ func (_m *MilvusServiceServer) DescribeIndex(_a0 context.Context, _a1 *milvuspb.
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DescribeIndexRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -882,8 +806,8 @@ type MilvusServiceServer_DescribeIndex_Call struct {
 }
 
 // DescribeIndex is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.DescribeIndexRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.DescribeIndexRequest
 func (_e *MilvusServiceServer_Expecter) DescribeIndex(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_DescribeIndex_Call {
 	return &MilvusServiceServer_DescribeIndex_Call{Call: _e.mock.On("DescribeIndex", _a0, _a1)}
 }
@@ -900,20 +824,11 @@ func (_c *MilvusServiceServer_DescribeIndex_Call) Return(_a0 *milvuspb.DescribeI
 	return _c
 }
 
-func (_c *MilvusServiceServer_DescribeIndex_Call) RunAndReturn(run func(context.Context, *milvuspb.DescribeIndexRequest) (*milvuspb.DescribeIndexResponse, error)) *MilvusServiceServer_DescribeIndex_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // DescribeResourceGroup provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) DescribeResourceGroup(_a0 context.Context, _a1 *milvuspb.DescribeResourceGroupRequest) (*milvuspb.DescribeResourceGroupResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.DescribeResourceGroupResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeResourceGroupRequest) (*milvuspb.DescribeResourceGroupResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeResourceGroupRequest) *milvuspb.DescribeResourceGroupResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -922,6 +837,7 @@ func (_m *MilvusServiceServer) DescribeResourceGroup(_a0 context.Context, _a1 *m
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DescribeResourceGroupRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -937,8 +853,8 @@ type MilvusServiceServer_DescribeResourceGroup_Call struct {
 }
 
 // DescribeResourceGroup is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.DescribeResourceGroupRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.DescribeResourceGroupRequest
 func (_e *MilvusServiceServer_Expecter) DescribeResourceGroup(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_DescribeResourceGroup_Call {
 	return &MilvusServiceServer_DescribeResourceGroup_Call{Call: _e.mock.On("DescribeResourceGroup", _a0, _a1)}
 }
@@ -955,20 +871,11 @@ func (_c *MilvusServiceServer_DescribeResourceGroup_Call) Return(_a0 *milvuspb.D
 	return _c
 }
 
-func (_c *MilvusServiceServer_DescribeResourceGroup_Call) RunAndReturn(run func(context.Context, *milvuspb.DescribeResourceGroupRequest) (*milvuspb.DescribeResourceGroupResponse, error)) *MilvusServiceServer_DescribeResourceGroup_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // DropAlias provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) DropAlias(_a0 context.Context, _a1 *milvuspb.DropAliasRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropAliasRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropAliasRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -977,6 +884,7 @@ func (_m *MilvusServiceServer) DropAlias(_a0 context.Context, _a1 *milvuspb.Drop
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropAliasRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -992,8 +900,8 @@ type MilvusServiceServer_DropAlias_Call struct {
 }
 
 // DropAlias is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.DropAliasRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.DropAliasRequest
 func (_e *MilvusServiceServer_Expecter) DropAlias(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_DropAlias_Call {
 	return &MilvusServiceServer_DropAlias_Call{Call: _e.mock.On("DropAlias", _a0, _a1)}
 }
@@ -1010,20 +918,11 @@ func (_c *MilvusServiceServer_DropAlias_Call) Return(_a0 *commonpb.Status, _a1 e
 	return _c
 }
 
-func (_c *MilvusServiceServer_DropAlias_Call) RunAndReturn(run func(context.Context, *milvuspb.DropAliasRequest) (*commonpb.Status, error)) *MilvusServiceServer_DropAlias_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // DropCollection provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) DropCollection(_a0 context.Context, _a1 *milvuspb.DropCollectionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropCollectionRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropCollectionRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1032,6 +931,7 @@ func (_m *MilvusServiceServer) DropCollection(_a0 context.Context, _a1 *milvuspb
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropCollectionRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1047,8 +947,8 @@ type MilvusServiceServer_DropCollection_Call struct {
 }
 
 // DropCollection is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.DropCollectionRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.DropCollectionRequest
 func (_e *MilvusServiceServer_Expecter) DropCollection(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_DropCollection_Call {
 	return &MilvusServiceServer_DropCollection_Call{Call: _e.mock.On("DropCollection", _a0, _a1)}
 }
@@ -1065,20 +965,11 @@ func (_c *MilvusServiceServer_DropCollection_Call) Return(_a0 *commonpb.Status,
 	return _c
 }
 
-func (_c *MilvusServiceServer_DropCollection_Call) RunAndReturn(run func(context.Context, *milvuspb.DropCollectionRequest) (*commonpb.Status, error)) *MilvusServiceServer_DropCollection_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // DropDatabase provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) DropDatabase(_a0 context.Context, _a1 *milvuspb.DropDatabaseRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropDatabaseRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropDatabaseRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1087,6 +978,7 @@ func (_m *MilvusServiceServer) DropDatabase(_a0 context.Context, _a1 *milvuspb.D
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropDatabaseRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1102,8 +994,8 @@ type MilvusServiceServer_DropDatabase_Call struct {
 }
 
 // DropDatabase is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.DropDatabaseRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.DropDatabaseRequest
 func (_e *MilvusServiceServer_Expecter) DropDatabase(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_DropDatabase_Call {
 	return &MilvusServiceServer_DropDatabase_Call{Call: _e.mock.On("DropDatabase", _a0, _a1)}
 }
@@ -1120,20 +1012,11 @@ func (_c *MilvusServiceServer_DropDatabase_Call) Return(_a0 *commonpb.Status, _a
 	return _c
 }
 
-func (_c *MilvusServiceServer_DropDatabase_Call) RunAndReturn(run func(context.Context, *milvuspb.DropDatabaseRequest) (*commonpb.Status, error)) *MilvusServiceServer_DropDatabase_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // DropIndex provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) DropIndex(_a0 context.Context, _a1 *milvuspb.DropIndexRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropIndexRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropIndexRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1142,6 +1025,7 @@ func (_m *MilvusServiceServer) DropIndex(_a0 context.Context, _a1 *milvuspb.Drop
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropIndexRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1157,8 +1041,8 @@ type MilvusServiceServer_DropIndex_Call struct {
 }
 
 // DropIndex is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.DropIndexRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.DropIndexRequest
 func (_e *MilvusServiceServer_Expecter) DropIndex(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_DropIndex_Call {
 	return &MilvusServiceServer_DropIndex_Call{Call: _e.mock.On("DropIndex", _a0, _a1)}
 }
@@ -1175,20 +1059,11 @@ func (_c *MilvusServiceServer_DropIndex_Call) Return(_a0 *commonpb.Status, _a1 e
 	return _c
 }
 
-func (_c *MilvusServiceServer_DropIndex_Call) RunAndReturn(run func(context.Context, *milvuspb.DropIndexRequest) (*commonpb.Status, error)) *MilvusServiceServer_DropIndex_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // DropPartition provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) DropPartition(_a0 context.Context, _a1 *milvuspb.DropPartitionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropPartitionRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropPartitionRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1197,6 +1072,7 @@ func (_m *MilvusServiceServer) DropPartition(_a0 context.Context, _a1 *milvuspb.
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropPartitionRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1212,8 +1088,8 @@ type MilvusServiceServer_DropPartition_Call struct {
 }
 
 // DropPartition is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.DropPartitionRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.DropPartitionRequest
 func (_e *MilvusServiceServer_Expecter) DropPartition(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_DropPartition_Call {
 	return &MilvusServiceServer_DropPartition_Call{Call: _e.mock.On("DropPartition", _a0, _a1)}
 }
@@ -1230,20 +1106,11 @@ func (_c *MilvusServiceServer_DropPartition_Call) Return(_a0 *commonpb.Status, _
 	return _c
 }
 
-func (_c *MilvusServiceServer_DropPartition_Call) RunAndReturn(run func(context.Context, *milvuspb.DropPartitionRequest) (*commonpb.Status, error)) *MilvusServiceServer_DropPartition_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // DropResourceGroup provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) DropResourceGroup(_a0 context.Context, _a1 *milvuspb.DropResourceGroupRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropResourceGroupRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropResourceGroupRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1252,6 +1119,7 @@ func (_m *MilvusServiceServer) DropResourceGroup(_a0 context.Context, _a1 *milvu
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropResourceGroupRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1267,8 +1135,8 @@ type MilvusServiceServer_DropResourceGroup_Call struct {
 }
 
 // DropResourceGroup is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.DropResourceGroupRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.DropResourceGroupRequest
 func (_e *MilvusServiceServer_Expecter) DropResourceGroup(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_DropResourceGroup_Call {
 	return &MilvusServiceServer_DropResourceGroup_Call{Call: _e.mock.On("DropResourceGroup", _a0, _a1)}
 }
@@ -1285,20 +1153,11 @@ func (_c *MilvusServiceServer_DropResourceGroup_Call) Return(_a0 *commonpb.Statu
 	return _c
 }
 
-func (_c *MilvusServiceServer_DropResourceGroup_Call) RunAndReturn(run func(context.Context, *milvuspb.DropResourceGroupRequest) (*commonpb.Status, error)) *MilvusServiceServer_DropResourceGroup_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // DropRole provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) DropRole(_a0 context.Context, _a1 *milvuspb.DropRoleRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropRoleRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropRoleRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1307,6 +1166,7 @@ func (_m *MilvusServiceServer) DropRole(_a0 context.Context, _a1 *milvuspb.DropR
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropRoleRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1322,8 +1182,8 @@ type MilvusServiceServer_DropRole_Call struct {
 }
 
 // DropRole is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.DropRoleRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.DropRoleRequest
 func (_e *MilvusServiceServer_Expecter) DropRole(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_DropRole_Call {
 	return &MilvusServiceServer_DropRole_Call{Call: _e.mock.On("DropRole", _a0, _a1)}
 }
@@ -1340,20 +1200,11 @@ func (_c *MilvusServiceServer_DropRole_Call) Return(_a0 *commonpb.Status, _a1 er
 	return _c
 }
 
-func (_c *MilvusServiceServer_DropRole_Call) RunAndReturn(run func(context.Context, *milvuspb.DropRoleRequest) (*commonpb.Status, error)) *MilvusServiceServer_DropRole_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // Dummy provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) Dummy(_a0 context.Context, _a1 *milvuspb.DummyRequest) (*milvuspb.DummyResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.DummyResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DummyRequest) (*milvuspb.DummyResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DummyRequest) *milvuspb.DummyResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1362,6 +1213,7 @@ func (_m *MilvusServiceServer) Dummy(_a0 context.Context, _a1 *milvuspb.DummyReq
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DummyRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1377,8 +1229,8 @@ type MilvusServiceServer_Dummy_Call struct {
 }
 
 // Dummy is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.DummyRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.DummyRequest
 func (_e *MilvusServiceServer_Expecter) Dummy(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_Dummy_Call {
 	return &MilvusServiceServer_Dummy_Call{Call: _e.mock.On("Dummy", _a0, _a1)}
 }
@@ -1395,20 +1247,11 @@ func (_c *MilvusServiceServer_Dummy_Call) Return(_a0 *milvuspb.DummyResponse, _a
 	return _c
 }
 
-func (_c *MilvusServiceServer_Dummy_Call) RunAndReturn(run func(context.Context, *milvuspb.DummyRequest) (*milvuspb.DummyResponse, error)) *MilvusServiceServer_Dummy_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // Flush provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) Flush(_a0 context.Context, _a1 *milvuspb.FlushRequest) (*milvuspb.FlushResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.FlushResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.FlushRequest) (*milvuspb.FlushResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.FlushRequest) *milvuspb.FlushResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1417,6 +1260,7 @@ func (_m *MilvusServiceServer) Flush(_a0 context.Context, _a1 *milvuspb.FlushReq
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.FlushRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1432,8 +1276,8 @@ type MilvusServiceServer_Flush_Call struct {
 }
 
 // Flush is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.FlushRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.FlushRequest
 func (_e *MilvusServiceServer_Expecter) Flush(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_Flush_Call {
 	return &MilvusServiceServer_Flush_Call{Call: _e.mock.On("Flush", _a0, _a1)}
 }
@@ -1450,20 +1294,11 @@ func (_c *MilvusServiceServer_Flush_Call) Return(_a0 *milvuspb.FlushResponse, _a
 	return _c
 }
 
-func (_c *MilvusServiceServer_Flush_Call) RunAndReturn(run func(context.Context, *milvuspb.FlushRequest) (*milvuspb.FlushResponse, error)) *MilvusServiceServer_Flush_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // FlushAll provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) FlushAll(_a0 context.Context, _a1 *milvuspb.FlushAllRequest) (*milvuspb.FlushAllResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.FlushAllResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.FlushAllRequest) (*milvuspb.FlushAllResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.FlushAllRequest) *milvuspb.FlushAllResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1472,6 +1307,7 @@ func (_m *MilvusServiceServer) FlushAll(_a0 context.Context, _a1 *milvuspb.Flush
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.FlushAllRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1487,8 +1323,8 @@ type MilvusServiceServer_FlushAll_Call struct {
 }
 
 // FlushAll is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.FlushAllRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.FlushAllRequest
 func (_e *MilvusServiceServer_Expecter) FlushAll(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_FlushAll_Call {
 	return &MilvusServiceServer_FlushAll_Call{Call: _e.mock.On("FlushAll", _a0, _a1)}
 }
@@ -1505,20 +1341,11 @@ func (_c *MilvusServiceServer_FlushAll_Call) Return(_a0 *milvuspb.FlushAllRespon
 	return _c
 }
 
-func (_c *MilvusServiceServer_FlushAll_Call) RunAndReturn(run func(context.Context, *milvuspb.FlushAllRequest) (*milvuspb.FlushAllResponse, error)) *MilvusServiceServer_FlushAll_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetCollectionStatistics provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetCollectionStatistics(_a0 context.Context, _a1 *milvuspb.GetCollectionStatisticsRequest) (*milvuspb.GetCollectionStatisticsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.GetCollectionStatisticsResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCollectionStatisticsRequest) (*milvuspb.GetCollectionStatisticsResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCollectionStatisticsRequest) *milvuspb.GetCollectionStatisticsResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1527,6 +1354,7 @@ func (_m *MilvusServiceServer) GetCollectionStatistics(_a0 context.Context, _a1
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetCollectionStatisticsRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1542,8 +1370,8 @@ type MilvusServiceServer_GetCollectionStatistics_Call struct {
 }
 
 // GetCollectionStatistics is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetCollectionStatisticsRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetCollectionStatisticsRequest
 func (_e *MilvusServiceServer_Expecter) GetCollectionStatistics(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetCollectionStatistics_Call {
 	return &MilvusServiceServer_GetCollectionStatistics_Call{Call: _e.mock.On("GetCollectionStatistics", _a0, _a1)}
 }
@@ -1560,20 +1388,11 @@ func (_c *MilvusServiceServer_GetCollectionStatistics_Call) Return(_a0 *milvuspb
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetCollectionStatistics_Call) RunAndReturn(run func(context.Context, *milvuspb.GetCollectionStatisticsRequest) (*milvuspb.GetCollectionStatisticsResponse, error)) *MilvusServiceServer_GetCollectionStatistics_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetCompactionState provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetCompactionState(_a0 context.Context, _a1 *milvuspb.GetCompactionStateRequest) (*milvuspb.GetCompactionStateResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.GetCompactionStateResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionStateRequest) (*milvuspb.GetCompactionStateResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionStateRequest) *milvuspb.GetCompactionStateResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1582,6 +1401,7 @@ func (_m *MilvusServiceServer) GetCompactionState(_a0 context.Context, _a1 *milv
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetCompactionStateRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1597,8 +1417,8 @@ type MilvusServiceServer_GetCompactionState_Call struct {
 }
 
 // GetCompactionState is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetCompactionStateRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetCompactionStateRequest
 func (_e *MilvusServiceServer_Expecter) GetCompactionState(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetCompactionState_Call {
 	return &MilvusServiceServer_GetCompactionState_Call{Call: _e.mock.On("GetCompactionState", _a0, _a1)}
 }
@@ -1615,20 +1435,11 @@ func (_c *MilvusServiceServer_GetCompactionState_Call) Return(_a0 *milvuspb.GetC
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetCompactionState_Call) RunAndReturn(run func(context.Context, *milvuspb.GetCompactionStateRequest) (*milvuspb.GetCompactionStateResponse, error)) *MilvusServiceServer_GetCompactionState_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetCompactionStateWithPlans provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetCompactionStateWithPlans(_a0 context.Context, _a1 *milvuspb.GetCompactionPlansRequest) (*milvuspb.GetCompactionPlansResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.GetCompactionPlansResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionPlansRequest) (*milvuspb.GetCompactionPlansResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionPlansRequest) *milvuspb.GetCompactionPlansResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1637,6 +1448,7 @@ func (_m *MilvusServiceServer) GetCompactionStateWithPlans(_a0 context.Context,
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetCompactionPlansRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1652,8 +1464,8 @@ type MilvusServiceServer_GetCompactionStateWithPlans_Call struct {
 }
 
 // GetCompactionStateWithPlans is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetCompactionPlansRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetCompactionPlansRequest
 func (_e *MilvusServiceServer_Expecter) GetCompactionStateWithPlans(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetCompactionStateWithPlans_Call {
 	return &MilvusServiceServer_GetCompactionStateWithPlans_Call{Call: _e.mock.On("GetCompactionStateWithPlans", _a0, _a1)}
 }
@@ -1670,20 +1482,11 @@ func (_c *MilvusServiceServer_GetCompactionStateWithPlans_Call) Return(_a0 *milv
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetCompactionStateWithPlans_Call) RunAndReturn(run func(context.Context, *milvuspb.GetCompactionPlansRequest) (*milvuspb.GetCompactionPlansResponse, error)) *MilvusServiceServer_GetCompactionStateWithPlans_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetComponentStates provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetComponentStates(_a0 context.Context, _a1 *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.ComponentStates
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) *milvuspb.ComponentStates); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1692,6 +1495,7 @@ func (_m *MilvusServiceServer) GetComponentStates(_a0 context.Context, _a1 *milv
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetComponentStatesRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1707,8 +1511,8 @@ type MilvusServiceServer_GetComponentStates_Call struct {
 }
 
 // GetComponentStates is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetComponentStatesRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetComponentStatesRequest
 func (_e *MilvusServiceServer_Expecter) GetComponentStates(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetComponentStates_Call {
 	return &MilvusServiceServer_GetComponentStates_Call{Call: _e.mock.On("GetComponentStates", _a0, _a1)}
 }
@@ -1725,20 +1529,11 @@ func (_c *MilvusServiceServer_GetComponentStates_Call) Return(_a0 *milvuspb.Comp
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetComponentStates_Call) RunAndReturn(run func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)) *MilvusServiceServer_GetComponentStates_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetFlushAllState provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetFlushAllState(_a0 context.Context, _a1 *milvuspb.GetFlushAllStateRequest) (*milvuspb.GetFlushAllStateResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.GetFlushAllStateResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetFlushAllStateRequest) (*milvuspb.GetFlushAllStateResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetFlushAllStateRequest) *milvuspb.GetFlushAllStateResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1747,6 +1542,7 @@ func (_m *MilvusServiceServer) GetFlushAllState(_a0 context.Context, _a1 *milvus
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetFlushAllStateRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1762,8 +1558,8 @@ type MilvusServiceServer_GetFlushAllState_Call struct {
 }
 
 // GetFlushAllState is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetFlushAllStateRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetFlushAllStateRequest
 func (_e *MilvusServiceServer_Expecter) GetFlushAllState(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetFlushAllState_Call {
 	return &MilvusServiceServer_GetFlushAllState_Call{Call: _e.mock.On("GetFlushAllState", _a0, _a1)}
 }
@@ -1780,20 +1576,11 @@ func (_c *MilvusServiceServer_GetFlushAllState_Call) Return(_a0 *milvuspb.GetFlu
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetFlushAllState_Call) RunAndReturn(run func(context.Context, *milvuspb.GetFlushAllStateRequest) (*milvuspb.GetFlushAllStateResponse, error)) *MilvusServiceServer_GetFlushAllState_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetFlushState provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetFlushState(_a0 context.Context, _a1 *milvuspb.GetFlushStateRequest) (*milvuspb.GetFlushStateResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.GetFlushStateResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetFlushStateRequest) (*milvuspb.GetFlushStateResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetFlushStateRequest) *milvuspb.GetFlushStateResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1802,6 +1589,7 @@ func (_m *MilvusServiceServer) GetFlushState(_a0 context.Context, _a1 *milvuspb.
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetFlushStateRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1817,8 +1605,8 @@ type MilvusServiceServer_GetFlushState_Call struct {
 }
 
 // GetFlushState is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetFlushStateRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetFlushStateRequest
 func (_e *MilvusServiceServer_Expecter) GetFlushState(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetFlushState_Call {
 	return &MilvusServiceServer_GetFlushState_Call{Call: _e.mock.On("GetFlushState", _a0, _a1)}
 }
@@ -1835,20 +1623,11 @@ func (_c *MilvusServiceServer_GetFlushState_Call) Return(_a0 *milvuspb.GetFlushS
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetFlushState_Call) RunAndReturn(run func(context.Context, *milvuspb.GetFlushStateRequest) (*milvuspb.GetFlushStateResponse, error)) *MilvusServiceServer_GetFlushState_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetImportState provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetImportState(_a0 context.Context, _a1 *milvuspb.GetImportStateRequest) (*milvuspb.GetImportStateResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.GetImportStateResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetImportStateRequest) (*milvuspb.GetImportStateResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetImportStateRequest) *milvuspb.GetImportStateResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1857,6 +1636,7 @@ func (_m *MilvusServiceServer) GetImportState(_a0 context.Context, _a1 *milvuspb
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetImportStateRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1872,8 +1652,8 @@ type MilvusServiceServer_GetImportState_Call struct {
 }
 
 // GetImportState is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetImportStateRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetImportStateRequest
 func (_e *MilvusServiceServer_Expecter) GetImportState(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetImportState_Call {
 	return &MilvusServiceServer_GetImportState_Call{Call: _e.mock.On("GetImportState", _a0, _a1)}
 }
@@ -1890,20 +1670,11 @@ func (_c *MilvusServiceServer_GetImportState_Call) Return(_a0 *milvuspb.GetImpor
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetImportState_Call) RunAndReturn(run func(context.Context, *milvuspb.GetImportStateRequest) (*milvuspb.GetImportStateResponse, error)) *MilvusServiceServer_GetImportState_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetIndexBuildProgress provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetIndexBuildProgress(_a0 context.Context, _a1 *milvuspb.GetIndexBuildProgressRequest) (*milvuspb.GetIndexBuildProgressResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.GetIndexBuildProgressResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetIndexBuildProgressRequest) (*milvuspb.GetIndexBuildProgressResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetIndexBuildProgressRequest) *milvuspb.GetIndexBuildProgressResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1912,6 +1683,7 @@ func (_m *MilvusServiceServer) GetIndexBuildProgress(_a0 context.Context, _a1 *m
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetIndexBuildProgressRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1927,8 +1699,8 @@ type MilvusServiceServer_GetIndexBuildProgress_Call struct {
 }
 
 // GetIndexBuildProgress is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetIndexBuildProgressRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetIndexBuildProgressRequest
 func (_e *MilvusServiceServer_Expecter) GetIndexBuildProgress(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetIndexBuildProgress_Call {
 	return &MilvusServiceServer_GetIndexBuildProgress_Call{Call: _e.mock.On("GetIndexBuildProgress", _a0, _a1)}
 }
@@ -1945,20 +1717,11 @@ func (_c *MilvusServiceServer_GetIndexBuildProgress_Call) Return(_a0 *milvuspb.G
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetIndexBuildProgress_Call) RunAndReturn(run func(context.Context, *milvuspb.GetIndexBuildProgressRequest) (*milvuspb.GetIndexBuildProgressResponse, error)) *MilvusServiceServer_GetIndexBuildProgress_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetIndexState provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetIndexState(_a0 context.Context, _a1 *milvuspb.GetIndexStateRequest) (*milvuspb.GetIndexStateResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.GetIndexStateResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetIndexStateRequest) (*milvuspb.GetIndexStateResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetIndexStateRequest) *milvuspb.GetIndexStateResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -1967,6 +1730,7 @@ func (_m *MilvusServiceServer) GetIndexState(_a0 context.Context, _a1 *milvuspb.
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetIndexStateRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -1982,8 +1746,8 @@ type MilvusServiceServer_GetIndexState_Call struct {
 }
 
 // GetIndexState is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetIndexStateRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetIndexStateRequest
 func (_e *MilvusServiceServer_Expecter) GetIndexState(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetIndexState_Call {
 	return &MilvusServiceServer_GetIndexState_Call{Call: _e.mock.On("GetIndexState", _a0, _a1)}
 }
@@ -2000,20 +1764,11 @@ func (_c *MilvusServiceServer_GetIndexState_Call) Return(_a0 *milvuspb.GetIndexS
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetIndexState_Call) RunAndReturn(run func(context.Context, *milvuspb.GetIndexStateRequest) (*milvuspb.GetIndexStateResponse, error)) *MilvusServiceServer_GetIndexState_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetLoadState provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetLoadState(_a0 context.Context, _a1 *milvuspb.GetLoadStateRequest) (*milvuspb.GetLoadStateResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.GetLoadStateResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetLoadStateRequest) (*milvuspb.GetLoadStateResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetLoadStateRequest) *milvuspb.GetLoadStateResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2022,6 +1777,7 @@ func (_m *MilvusServiceServer) GetLoadState(_a0 context.Context, _a1 *milvuspb.G
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetLoadStateRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2037,8 +1793,8 @@ type MilvusServiceServer_GetLoadState_Call struct {
 }
 
 // GetLoadState is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetLoadStateRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetLoadStateRequest
 func (_e *MilvusServiceServer_Expecter) GetLoadState(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetLoadState_Call {
 	return &MilvusServiceServer_GetLoadState_Call{Call: _e.mock.On("GetLoadState", _a0, _a1)}
 }
@@ -2055,20 +1811,11 @@ func (_c *MilvusServiceServer_GetLoadState_Call) Return(_a0 *milvuspb.GetLoadSta
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetLoadState_Call) RunAndReturn(run func(context.Context, *milvuspb.GetLoadStateRequest) (*milvuspb.GetLoadStateResponse, error)) *MilvusServiceServer_GetLoadState_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetLoadingProgress provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetLoadingProgress(_a0 context.Context, _a1 *milvuspb.GetLoadingProgressRequest) (*milvuspb.GetLoadingProgressResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.GetLoadingProgressResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetLoadingProgressRequest) (*milvuspb.GetLoadingProgressResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetLoadingProgressRequest) *milvuspb.GetLoadingProgressResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2077,6 +1824,7 @@ func (_m *MilvusServiceServer) GetLoadingProgress(_a0 context.Context, _a1 *milv
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetLoadingProgressRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2092,8 +1840,8 @@ type MilvusServiceServer_GetLoadingProgress_Call struct {
 }
 
 // GetLoadingProgress is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetLoadingProgressRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetLoadingProgressRequest
 func (_e *MilvusServiceServer_Expecter) GetLoadingProgress(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetLoadingProgress_Call {
 	return &MilvusServiceServer_GetLoadingProgress_Call{Call: _e.mock.On("GetLoadingProgress", _a0, _a1)}
 }
@@ -2110,20 +1858,11 @@ func (_c *MilvusServiceServer_GetLoadingProgress_Call) Return(_a0 *milvuspb.GetL
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetLoadingProgress_Call) RunAndReturn(run func(context.Context, *milvuspb.GetLoadingProgressRequest) (*milvuspb.GetLoadingProgressResponse, error)) *MilvusServiceServer_GetLoadingProgress_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetMetrics provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetMetrics(_a0 context.Context, _a1 *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.GetMetricsResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) *milvuspb.GetMetricsResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2132,6 +1871,7 @@ func (_m *MilvusServiceServer) GetMetrics(_a0 context.Context, _a1 *milvuspb.Get
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetMetricsRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2147,8 +1887,8 @@ type MilvusServiceServer_GetMetrics_Call struct {
 }
 
 // GetMetrics is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetMetricsRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetMetricsRequest
 func (_e *MilvusServiceServer_Expecter) GetMetrics(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetMetrics_Call {
 	return &MilvusServiceServer_GetMetrics_Call{Call: _e.mock.On("GetMetrics", _a0, _a1)}
 }
@@ -2165,20 +1905,11 @@ func (_c *MilvusServiceServer_GetMetrics_Call) Return(_a0 *milvuspb.GetMetricsRe
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetMetrics_Call) RunAndReturn(run func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)) *MilvusServiceServer_GetMetrics_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetPartitionStatistics provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetPartitionStatistics(_a0 context.Context, _a1 *milvuspb.GetPartitionStatisticsRequest) (*milvuspb.GetPartitionStatisticsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.GetPartitionStatisticsResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetPartitionStatisticsRequest) (*milvuspb.GetPartitionStatisticsResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetPartitionStatisticsRequest) *milvuspb.GetPartitionStatisticsResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2187,6 +1918,7 @@ func (_m *MilvusServiceServer) GetPartitionStatistics(_a0 context.Context, _a1 *
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetPartitionStatisticsRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2202,8 +1934,8 @@ type MilvusServiceServer_GetPartitionStatistics_Call struct {
 }
 
 // GetPartitionStatistics is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetPartitionStatisticsRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetPartitionStatisticsRequest
 func (_e *MilvusServiceServer_Expecter) GetPartitionStatistics(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetPartitionStatistics_Call {
 	return &MilvusServiceServer_GetPartitionStatistics_Call{Call: _e.mock.On("GetPartitionStatistics", _a0, _a1)}
 }
@@ -2220,20 +1952,11 @@ func (_c *MilvusServiceServer_GetPartitionStatistics_Call) Return(_a0 *milvuspb.
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetPartitionStatistics_Call) RunAndReturn(run func(context.Context, *milvuspb.GetPartitionStatisticsRequest) (*milvuspb.GetPartitionStatisticsResponse, error)) *MilvusServiceServer_GetPartitionStatistics_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetPersistentSegmentInfo provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetPersistentSegmentInfo(_a0 context.Context, _a1 *milvuspb.GetPersistentSegmentInfoRequest) (*milvuspb.GetPersistentSegmentInfoResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.GetPersistentSegmentInfoResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetPersistentSegmentInfoRequest) (*milvuspb.GetPersistentSegmentInfoResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetPersistentSegmentInfoRequest) *milvuspb.GetPersistentSegmentInfoResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2242,6 +1965,7 @@ func (_m *MilvusServiceServer) GetPersistentSegmentInfo(_a0 context.Context, _a1
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetPersistentSegmentInfoRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2257,8 +1981,8 @@ type MilvusServiceServer_GetPersistentSegmentInfo_Call struct {
 }
 
 // GetPersistentSegmentInfo is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetPersistentSegmentInfoRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetPersistentSegmentInfoRequest
 func (_e *MilvusServiceServer_Expecter) GetPersistentSegmentInfo(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetPersistentSegmentInfo_Call {
 	return &MilvusServiceServer_GetPersistentSegmentInfo_Call{Call: _e.mock.On("GetPersistentSegmentInfo", _a0, _a1)}
 }
@@ -2275,20 +1999,11 @@ func (_c *MilvusServiceServer_GetPersistentSegmentInfo_Call) Return(_a0 *milvusp
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetPersistentSegmentInfo_Call) RunAndReturn(run func(context.Context, *milvuspb.GetPersistentSegmentInfoRequest) (*milvuspb.GetPersistentSegmentInfoResponse, error)) *MilvusServiceServer_GetPersistentSegmentInfo_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetQuerySegmentInfo provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetQuerySegmentInfo(_a0 context.Context, _a1 *milvuspb.GetQuerySegmentInfoRequest) (*milvuspb.GetQuerySegmentInfoResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.GetQuerySegmentInfoResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetQuerySegmentInfoRequest) (*milvuspb.GetQuerySegmentInfoResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetQuerySegmentInfoRequest) *milvuspb.GetQuerySegmentInfoResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2297,6 +2012,7 @@ func (_m *MilvusServiceServer) GetQuerySegmentInfo(_a0 context.Context, _a1 *mil
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetQuerySegmentInfoRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2312,8 +2028,8 @@ type MilvusServiceServer_GetQuerySegmentInfo_Call struct {
 }
 
 // GetQuerySegmentInfo is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetQuerySegmentInfoRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetQuerySegmentInfoRequest
 func (_e *MilvusServiceServer_Expecter) GetQuerySegmentInfo(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetQuerySegmentInfo_Call {
 	return &MilvusServiceServer_GetQuerySegmentInfo_Call{Call: _e.mock.On("GetQuerySegmentInfo", _a0, _a1)}
 }
@@ -2330,20 +2046,11 @@ func (_c *MilvusServiceServer_GetQuerySegmentInfo_Call) Return(_a0 *milvuspb.Get
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetQuerySegmentInfo_Call) RunAndReturn(run func(context.Context, *milvuspb.GetQuerySegmentInfoRequest) (*milvuspb.GetQuerySegmentInfoResponse, error)) *MilvusServiceServer_GetQuerySegmentInfo_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetReplicas provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetReplicas(_a0 context.Context, _a1 *milvuspb.GetReplicasRequest) (*milvuspb.GetReplicasResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.GetReplicasResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetReplicasRequest) (*milvuspb.GetReplicasResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetReplicasRequest) *milvuspb.GetReplicasResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2352,6 +2059,7 @@ func (_m *MilvusServiceServer) GetReplicas(_a0 context.Context, _a1 *milvuspb.Ge
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetReplicasRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2367,8 +2075,8 @@ type MilvusServiceServer_GetReplicas_Call struct {
 }
 
 // GetReplicas is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetReplicasRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetReplicasRequest
 func (_e *MilvusServiceServer_Expecter) GetReplicas(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetReplicas_Call {
 	return &MilvusServiceServer_GetReplicas_Call{Call: _e.mock.On("GetReplicas", _a0, _a1)}
 }
@@ -2385,20 +2093,11 @@ func (_c *MilvusServiceServer_GetReplicas_Call) Return(_a0 *milvuspb.GetReplicas
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetReplicas_Call) RunAndReturn(run func(context.Context, *milvuspb.GetReplicasRequest) (*milvuspb.GetReplicasResponse, error)) *MilvusServiceServer_GetReplicas_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // GetVersion provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) GetVersion(_a0 context.Context, _a1 *milvuspb.GetVersionRequest) (*milvuspb.GetVersionResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.GetVersionResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetVersionRequest) (*milvuspb.GetVersionResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetVersionRequest) *milvuspb.GetVersionResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2407,6 +2106,7 @@ func (_m *MilvusServiceServer) GetVersion(_a0 context.Context, _a1 *milvuspb.Get
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetVersionRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2422,8 +2122,8 @@ type MilvusServiceServer_GetVersion_Call struct {
 }
 
 // GetVersion is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.GetVersionRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.GetVersionRequest
 func (_e *MilvusServiceServer_Expecter) GetVersion(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetVersion_Call {
 	return &MilvusServiceServer_GetVersion_Call{Call: _e.mock.On("GetVersion", _a0, _a1)}
 }
@@ -2440,20 +2140,11 @@ func (_c *MilvusServiceServer_GetVersion_Call) Return(_a0 *milvuspb.GetVersionRe
 	return _c
 }
 
-func (_c *MilvusServiceServer_GetVersion_Call) RunAndReturn(run func(context.Context, *milvuspb.GetVersionRequest) (*milvuspb.GetVersionResponse, error)) *MilvusServiceServer_GetVersion_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // HasCollection provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) HasCollection(_a0 context.Context, _a1 *milvuspb.HasCollectionRequest) (*milvuspb.BoolResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.BoolResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasCollectionRequest) (*milvuspb.BoolResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasCollectionRequest) *milvuspb.BoolResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2462,6 +2153,7 @@ func (_m *MilvusServiceServer) HasCollection(_a0 context.Context, _a1 *milvuspb.
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.HasCollectionRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2477,8 +2169,8 @@ type MilvusServiceServer_HasCollection_Call struct {
 }
 
 // HasCollection is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.HasCollectionRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.HasCollectionRequest
 func (_e *MilvusServiceServer_Expecter) HasCollection(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_HasCollection_Call {
 	return &MilvusServiceServer_HasCollection_Call{Call: _e.mock.On("HasCollection", _a0, _a1)}
 }
@@ -2495,20 +2187,11 @@ func (_c *MilvusServiceServer_HasCollection_Call) Return(_a0 *milvuspb.BoolRespo
 	return _c
 }
 
-func (_c *MilvusServiceServer_HasCollection_Call) RunAndReturn(run func(context.Context, *milvuspb.HasCollectionRequest) (*milvuspb.BoolResponse, error)) *MilvusServiceServer_HasCollection_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // HasPartition provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) HasPartition(_a0 context.Context, _a1 *milvuspb.HasPartitionRequest) (*milvuspb.BoolResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.BoolResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasPartitionRequest) (*milvuspb.BoolResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasPartitionRequest) *milvuspb.BoolResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2517,6 +2200,7 @@ func (_m *MilvusServiceServer) HasPartition(_a0 context.Context, _a1 *milvuspb.H
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.HasPartitionRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2532,8 +2216,8 @@ type MilvusServiceServer_HasPartition_Call struct {
 }
 
 // HasPartition is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.HasPartitionRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.HasPartitionRequest
 func (_e *MilvusServiceServer_Expecter) HasPartition(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_HasPartition_Call {
 	return &MilvusServiceServer_HasPartition_Call{Call: _e.mock.On("HasPartition", _a0, _a1)}
 }
@@ -2550,20 +2234,11 @@ func (_c *MilvusServiceServer_HasPartition_Call) Return(_a0 *milvuspb.BoolRespon
 	return _c
 }
 
-func (_c *MilvusServiceServer_HasPartition_Call) RunAndReturn(run func(context.Context, *milvuspb.HasPartitionRequest) (*milvuspb.BoolResponse, error)) *MilvusServiceServer_HasPartition_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // Import provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) Import(_a0 context.Context, _a1 *milvuspb.ImportRequest) (*milvuspb.ImportResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.ImportResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ImportRequest) (*milvuspb.ImportResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ImportRequest) *milvuspb.ImportResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2572,6 +2247,7 @@ func (_m *MilvusServiceServer) Import(_a0 context.Context, _a1 *milvuspb.ImportR
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ImportRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2587,8 +2263,8 @@ type MilvusServiceServer_Import_Call struct {
 }
 
 // Import is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.ImportRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.ImportRequest
 func (_e *MilvusServiceServer_Expecter) Import(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_Import_Call {
 	return &MilvusServiceServer_Import_Call{Call: _e.mock.On("Import", _a0, _a1)}
 }
@@ -2605,20 +2281,11 @@ func (_c *MilvusServiceServer_Import_Call) Return(_a0 *milvuspb.ImportResponse,
 	return _c
 }
 
-func (_c *MilvusServiceServer_Import_Call) RunAndReturn(run func(context.Context, *milvuspb.ImportRequest) (*milvuspb.ImportResponse, error)) *MilvusServiceServer_Import_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // Insert provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) Insert(_a0 context.Context, _a1 *milvuspb.InsertRequest) (*milvuspb.MutationResult, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.MutationResult
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.InsertRequest) (*milvuspb.MutationResult, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.InsertRequest) *milvuspb.MutationResult); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2627,6 +2294,7 @@ func (_m *MilvusServiceServer) Insert(_a0 context.Context, _a1 *milvuspb.InsertR
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.InsertRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2642,8 +2310,8 @@ type MilvusServiceServer_Insert_Call struct {
 }
 
 // Insert is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.InsertRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.InsertRequest
 func (_e *MilvusServiceServer_Expecter) Insert(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_Insert_Call {
 	return &MilvusServiceServer_Insert_Call{Call: _e.mock.On("Insert", _a0, _a1)}
 }
@@ -2660,20 +2328,11 @@ func (_c *MilvusServiceServer_Insert_Call) Return(_a0 *milvuspb.MutationResult,
 	return _c
 }
 
-func (_c *MilvusServiceServer_Insert_Call) RunAndReturn(run func(context.Context, *milvuspb.InsertRequest) (*milvuspb.MutationResult, error)) *MilvusServiceServer_Insert_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // ListCredUsers provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) ListCredUsers(_a0 context.Context, _a1 *milvuspb.ListCredUsersRequest) (*milvuspb.ListCredUsersResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.ListCredUsersResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListCredUsersRequest) (*milvuspb.ListCredUsersResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListCredUsersRequest) *milvuspb.ListCredUsersResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2682,6 +2341,7 @@ func (_m *MilvusServiceServer) ListCredUsers(_a0 context.Context, _a1 *milvuspb.
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ListCredUsersRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2697,8 +2357,8 @@ type MilvusServiceServer_ListCredUsers_Call struct {
 }
 
 // ListCredUsers is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.ListCredUsersRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.ListCredUsersRequest
 func (_e *MilvusServiceServer_Expecter) ListCredUsers(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_ListCredUsers_Call {
 	return &MilvusServiceServer_ListCredUsers_Call{Call: _e.mock.On("ListCredUsers", _a0, _a1)}
 }
@@ -2715,20 +2375,11 @@ func (_c *MilvusServiceServer_ListCredUsers_Call) Return(_a0 *milvuspb.ListCredU
 	return _c
 }
 
-func (_c *MilvusServiceServer_ListCredUsers_Call) RunAndReturn(run func(context.Context, *milvuspb.ListCredUsersRequest) (*milvuspb.ListCredUsersResponse, error)) *MilvusServiceServer_ListCredUsers_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // ListDatabases provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) ListDatabases(_a0 context.Context, _a1 *milvuspb.ListDatabasesRequest) (*milvuspb.ListDatabasesResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.ListDatabasesResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListDatabasesRequest) (*milvuspb.ListDatabasesResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListDatabasesRequest) *milvuspb.ListDatabasesResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2737,6 +2388,7 @@ func (_m *MilvusServiceServer) ListDatabases(_a0 context.Context, _a1 *milvuspb.
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ListDatabasesRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2752,8 +2404,8 @@ type MilvusServiceServer_ListDatabases_Call struct {
 }
 
 // ListDatabases is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.ListDatabasesRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.ListDatabasesRequest
 func (_e *MilvusServiceServer_Expecter) ListDatabases(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_ListDatabases_Call {
 	return &MilvusServiceServer_ListDatabases_Call{Call: _e.mock.On("ListDatabases", _a0, _a1)}
 }
@@ -2770,20 +2422,11 @@ func (_c *MilvusServiceServer_ListDatabases_Call) Return(_a0 *milvuspb.ListDatab
 	return _c
 }
 
-func (_c *MilvusServiceServer_ListDatabases_Call) RunAndReturn(run func(context.Context, *milvuspb.ListDatabasesRequest) (*milvuspb.ListDatabasesResponse, error)) *MilvusServiceServer_ListDatabases_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // ListImportTasks provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) ListImportTasks(_a0 context.Context, _a1 *milvuspb.ListImportTasksRequest) (*milvuspb.ListImportTasksResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.ListImportTasksResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListImportTasksRequest) (*milvuspb.ListImportTasksResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListImportTasksRequest) *milvuspb.ListImportTasksResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2792,6 +2435,7 @@ func (_m *MilvusServiceServer) ListImportTasks(_a0 context.Context, _a1 *milvusp
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ListImportTasksRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2807,8 +2451,8 @@ type MilvusServiceServer_ListImportTasks_Call struct {
 }
 
 // ListImportTasks is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.ListImportTasksRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.ListImportTasksRequest
 func (_e *MilvusServiceServer_Expecter) ListImportTasks(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_ListImportTasks_Call {
 	return &MilvusServiceServer_ListImportTasks_Call{Call: _e.mock.On("ListImportTasks", _a0, _a1)}
 }
@@ -2825,20 +2469,11 @@ func (_c *MilvusServiceServer_ListImportTasks_Call) Return(_a0 *milvuspb.ListImp
 	return _c
 }
 
-func (_c *MilvusServiceServer_ListImportTasks_Call) RunAndReturn(run func(context.Context, *milvuspb.ListImportTasksRequest) (*milvuspb.ListImportTasksResponse, error)) *MilvusServiceServer_ListImportTasks_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // ListResourceGroups provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) ListResourceGroups(_a0 context.Context, _a1 *milvuspb.ListResourceGroupsRequest) (*milvuspb.ListResourceGroupsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.ListResourceGroupsResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListResourceGroupsRequest) (*milvuspb.ListResourceGroupsResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListResourceGroupsRequest) *milvuspb.ListResourceGroupsResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2847,6 +2482,7 @@ func (_m *MilvusServiceServer) ListResourceGroups(_a0 context.Context, _a1 *milv
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ListResourceGroupsRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2862,8 +2498,8 @@ type MilvusServiceServer_ListResourceGroups_Call struct {
 }
 
 // ListResourceGroups is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.ListResourceGroupsRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.ListResourceGroupsRequest
 func (_e *MilvusServiceServer_Expecter) ListResourceGroups(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_ListResourceGroups_Call {
 	return &MilvusServiceServer_ListResourceGroups_Call{Call: _e.mock.On("ListResourceGroups", _a0, _a1)}
 }
@@ -2880,20 +2516,11 @@ func (_c *MilvusServiceServer_ListResourceGroups_Call) Return(_a0 *milvuspb.List
 	return _c
 }
 
-func (_c *MilvusServiceServer_ListResourceGroups_Call) RunAndReturn(run func(context.Context, *milvuspb.ListResourceGroupsRequest) (*milvuspb.ListResourceGroupsResponse, error)) *MilvusServiceServer_ListResourceGroups_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // LoadBalance provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) LoadBalance(_a0 context.Context, _a1 *milvuspb.LoadBalanceRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadBalanceRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadBalanceRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2902,6 +2529,7 @@ func (_m *MilvusServiceServer) LoadBalance(_a0 context.Context, _a1 *milvuspb.Lo
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.LoadBalanceRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2917,8 +2545,8 @@ type MilvusServiceServer_LoadBalance_Call struct {
 }
 
 // LoadBalance is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.LoadBalanceRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.LoadBalanceRequest
 func (_e *MilvusServiceServer_Expecter) LoadBalance(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_LoadBalance_Call {
 	return &MilvusServiceServer_LoadBalance_Call{Call: _e.mock.On("LoadBalance", _a0, _a1)}
 }
@@ -2935,20 +2563,11 @@ func (_c *MilvusServiceServer_LoadBalance_Call) Return(_a0 *commonpb.Status, _a1
 	return _c
 }
 
-func (_c *MilvusServiceServer_LoadBalance_Call) RunAndReturn(run func(context.Context, *milvuspb.LoadBalanceRequest) (*commonpb.Status, error)) *MilvusServiceServer_LoadBalance_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // LoadCollection provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) LoadCollection(_a0 context.Context, _a1 *milvuspb.LoadCollectionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadCollectionRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadCollectionRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -2957,6 +2576,7 @@ func (_m *MilvusServiceServer) LoadCollection(_a0 context.Context, _a1 *milvuspb
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.LoadCollectionRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -2972,8 +2592,8 @@ type MilvusServiceServer_LoadCollection_Call struct {
 }
 
 // LoadCollection is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.LoadCollectionRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.LoadCollectionRequest
 func (_e *MilvusServiceServer_Expecter) LoadCollection(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_LoadCollection_Call {
 	return &MilvusServiceServer_LoadCollection_Call{Call: _e.mock.On("LoadCollection", _a0, _a1)}
 }
@@ -2990,20 +2610,11 @@ func (_c *MilvusServiceServer_LoadCollection_Call) Return(_a0 *commonpb.Status,
 	return _c
 }
 
-func (_c *MilvusServiceServer_LoadCollection_Call) RunAndReturn(run func(context.Context, *milvuspb.LoadCollectionRequest) (*commonpb.Status, error)) *MilvusServiceServer_LoadCollection_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // LoadPartitions provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) LoadPartitions(_a0 context.Context, _a1 *milvuspb.LoadPartitionsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadPartitionsRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadPartitionsRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3012,6 +2623,7 @@ func (_m *MilvusServiceServer) LoadPartitions(_a0 context.Context, _a1 *milvuspb
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.LoadPartitionsRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3027,8 +2639,8 @@ type MilvusServiceServer_LoadPartitions_Call struct {
 }
 
 // LoadPartitions is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.LoadPartitionsRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.LoadPartitionsRequest
 func (_e *MilvusServiceServer_Expecter) LoadPartitions(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_LoadPartitions_Call {
 	return &MilvusServiceServer_LoadPartitions_Call{Call: _e.mock.On("LoadPartitions", _a0, _a1)}
 }
@@ -3045,20 +2657,11 @@ func (_c *MilvusServiceServer_LoadPartitions_Call) Return(_a0 *commonpb.Status,
 	return _c
 }
 
-func (_c *MilvusServiceServer_LoadPartitions_Call) RunAndReturn(run func(context.Context, *milvuspb.LoadPartitionsRequest) (*commonpb.Status, error)) *MilvusServiceServer_LoadPartitions_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // ManualCompaction provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) ManualCompaction(_a0 context.Context, _a1 *milvuspb.ManualCompactionRequest) (*milvuspb.ManualCompactionResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.ManualCompactionResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ManualCompactionRequest) (*milvuspb.ManualCompactionResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ManualCompactionRequest) *milvuspb.ManualCompactionResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3067,6 +2670,7 @@ func (_m *MilvusServiceServer) ManualCompaction(_a0 context.Context, _a1 *milvus
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ManualCompactionRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3082,8 +2686,8 @@ type MilvusServiceServer_ManualCompaction_Call struct {
 }
 
 // ManualCompaction is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.ManualCompactionRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.ManualCompactionRequest
 func (_e *MilvusServiceServer_Expecter) ManualCompaction(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_ManualCompaction_Call {
 	return &MilvusServiceServer_ManualCompaction_Call{Call: _e.mock.On("ManualCompaction", _a0, _a1)}
 }
@@ -3100,20 +2704,11 @@ func (_c *MilvusServiceServer_ManualCompaction_Call) Return(_a0 *milvuspb.Manual
 	return _c
 }
 
-func (_c *MilvusServiceServer_ManualCompaction_Call) RunAndReturn(run func(context.Context, *milvuspb.ManualCompactionRequest) (*milvuspb.ManualCompactionResponse, error)) *MilvusServiceServer_ManualCompaction_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // OperatePrivilege provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) OperatePrivilege(_a0 context.Context, _a1 *milvuspb.OperatePrivilegeRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperatePrivilegeRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperatePrivilegeRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3122,6 +2717,7 @@ func (_m *MilvusServiceServer) OperatePrivilege(_a0 context.Context, _a1 *milvus
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.OperatePrivilegeRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3137,8 +2733,8 @@ type MilvusServiceServer_OperatePrivilege_Call struct {
 }
 
 // OperatePrivilege is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.OperatePrivilegeRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.OperatePrivilegeRequest
 func (_e *MilvusServiceServer_Expecter) OperatePrivilege(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_OperatePrivilege_Call {
 	return &MilvusServiceServer_OperatePrivilege_Call{Call: _e.mock.On("OperatePrivilege", _a0, _a1)}
 }
@@ -3155,20 +2751,11 @@ func (_c *MilvusServiceServer_OperatePrivilege_Call) Return(_a0 *commonpb.Status
 	return _c
 }
 
-func (_c *MilvusServiceServer_OperatePrivilege_Call) RunAndReturn(run func(context.Context, *milvuspb.OperatePrivilegeRequest) (*commonpb.Status, error)) *MilvusServiceServer_OperatePrivilege_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // OperateUserRole provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) OperateUserRole(_a0 context.Context, _a1 *milvuspb.OperateUserRoleRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperateUserRoleRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperateUserRoleRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3177,6 +2764,7 @@ func (_m *MilvusServiceServer) OperateUserRole(_a0 context.Context, _a1 *milvusp
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.OperateUserRoleRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3192,8 +2780,8 @@ type MilvusServiceServer_OperateUserRole_Call struct {
 }
 
 // OperateUserRole is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.OperateUserRoleRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.OperateUserRoleRequest
 func (_e *MilvusServiceServer_Expecter) OperateUserRole(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_OperateUserRole_Call {
 	return &MilvusServiceServer_OperateUserRole_Call{Call: _e.mock.On("OperateUserRole", _a0, _a1)}
 }
@@ -3210,20 +2798,11 @@ func (_c *MilvusServiceServer_OperateUserRole_Call) Return(_a0 *commonpb.Status,
 	return _c
 }
 
-func (_c *MilvusServiceServer_OperateUserRole_Call) RunAndReturn(run func(context.Context, *milvuspb.OperateUserRoleRequest) (*commonpb.Status, error)) *MilvusServiceServer_OperateUserRole_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // Query provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) Query(_a0 context.Context, _a1 *milvuspb.QueryRequest) (*milvuspb.QueryResults, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.QueryResults
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.QueryRequest) (*milvuspb.QueryResults, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.QueryRequest) *milvuspb.QueryResults); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3232,6 +2811,7 @@ func (_m *MilvusServiceServer) Query(_a0 context.Context, _a1 *milvuspb.QueryReq
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.QueryRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3247,8 +2827,8 @@ type MilvusServiceServer_Query_Call struct {
 }
 
 // Query is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.QueryRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.QueryRequest
 func (_e *MilvusServiceServer_Expecter) Query(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_Query_Call {
 	return &MilvusServiceServer_Query_Call{Call: _e.mock.On("Query", _a0, _a1)}
 }
@@ -3265,20 +2845,11 @@ func (_c *MilvusServiceServer_Query_Call) Return(_a0 *milvuspb.QueryResults, _a1
 	return _c
 }
 
-func (_c *MilvusServiceServer_Query_Call) RunAndReturn(run func(context.Context, *milvuspb.QueryRequest) (*milvuspb.QueryResults, error)) *MilvusServiceServer_Query_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // RegisterLink provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) RegisterLink(_a0 context.Context, _a1 *milvuspb.RegisterLinkRequest) (*milvuspb.RegisterLinkResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.RegisterLinkResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RegisterLinkRequest) (*milvuspb.RegisterLinkResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RegisterLinkRequest) *milvuspb.RegisterLinkResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3287,6 +2858,7 @@ func (_m *MilvusServiceServer) RegisterLink(_a0 context.Context, _a1 *milvuspb.R
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.RegisterLinkRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3302,8 +2874,8 @@ type MilvusServiceServer_RegisterLink_Call struct {
 }
 
 // RegisterLink is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.RegisterLinkRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.RegisterLinkRequest
 func (_e *MilvusServiceServer_Expecter) RegisterLink(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_RegisterLink_Call {
 	return &MilvusServiceServer_RegisterLink_Call{Call: _e.mock.On("RegisterLink", _a0, _a1)}
 }
@@ -3320,20 +2892,11 @@ func (_c *MilvusServiceServer_RegisterLink_Call) Return(_a0 *milvuspb.RegisterLi
 	return _c
 }
 
-func (_c *MilvusServiceServer_RegisterLink_Call) RunAndReturn(run func(context.Context, *milvuspb.RegisterLinkRequest) (*milvuspb.RegisterLinkResponse, error)) *MilvusServiceServer_RegisterLink_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // ReleaseCollection provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) ReleaseCollection(_a0 context.Context, _a1 *milvuspb.ReleaseCollectionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ReleaseCollectionRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ReleaseCollectionRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3342,6 +2905,7 @@ func (_m *MilvusServiceServer) ReleaseCollection(_a0 context.Context, _a1 *milvu
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ReleaseCollectionRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3357,8 +2921,8 @@ type MilvusServiceServer_ReleaseCollection_Call struct {
 }
 
 // ReleaseCollection is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.ReleaseCollectionRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.ReleaseCollectionRequest
 func (_e *MilvusServiceServer_Expecter) ReleaseCollection(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_ReleaseCollection_Call {
 	return &MilvusServiceServer_ReleaseCollection_Call{Call: _e.mock.On("ReleaseCollection", _a0, _a1)}
 }
@@ -3375,20 +2939,11 @@ func (_c *MilvusServiceServer_ReleaseCollection_Call) Return(_a0 *commonpb.Statu
 	return _c
 }
 
-func (_c *MilvusServiceServer_ReleaseCollection_Call) RunAndReturn(run func(context.Context, *milvuspb.ReleaseCollectionRequest) (*commonpb.Status, error)) *MilvusServiceServer_ReleaseCollection_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // ReleasePartitions provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) ReleasePartitions(_a0 context.Context, _a1 *milvuspb.ReleasePartitionsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ReleasePartitionsRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ReleasePartitionsRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3397,6 +2952,7 @@ func (_m *MilvusServiceServer) ReleasePartitions(_a0 context.Context, _a1 *milvu
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ReleasePartitionsRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3412,8 +2968,8 @@ type MilvusServiceServer_ReleasePartitions_Call struct {
 }
 
 // ReleasePartitions is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.ReleasePartitionsRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.ReleasePartitionsRequest
 func (_e *MilvusServiceServer_Expecter) ReleasePartitions(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_ReleasePartitions_Call {
 	return &MilvusServiceServer_ReleasePartitions_Call{Call: _e.mock.On("ReleasePartitions", _a0, _a1)}
 }
@@ -3430,20 +2986,11 @@ func (_c *MilvusServiceServer_ReleasePartitions_Call) Return(_a0 *commonpb.Statu
 	return _c
 }
 
-func (_c *MilvusServiceServer_ReleasePartitions_Call) RunAndReturn(run func(context.Context, *milvuspb.ReleasePartitionsRequest) (*commonpb.Status, error)) *MilvusServiceServer_ReleasePartitions_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // RenameCollection provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) RenameCollection(_a0 context.Context, _a1 *milvuspb.RenameCollectionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RenameCollectionRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RenameCollectionRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3452,6 +2999,7 @@ func (_m *MilvusServiceServer) RenameCollection(_a0 context.Context, _a1 *milvus
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.RenameCollectionRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3467,8 +3015,8 @@ type MilvusServiceServer_RenameCollection_Call struct {
 }
 
 // RenameCollection is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.RenameCollectionRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.RenameCollectionRequest
 func (_e *MilvusServiceServer_Expecter) RenameCollection(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_RenameCollection_Call {
 	return &MilvusServiceServer_RenameCollection_Call{Call: _e.mock.On("RenameCollection", _a0, _a1)}
 }
@@ -3485,20 +3033,11 @@ func (_c *MilvusServiceServer_RenameCollection_Call) Return(_a0 *commonpb.Status
 	return _c
 }
 
-func (_c *MilvusServiceServer_RenameCollection_Call) RunAndReturn(run func(context.Context, *milvuspb.RenameCollectionRequest) (*commonpb.Status, error)) *MilvusServiceServer_RenameCollection_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // Search provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) Search(_a0 context.Context, _a1 *milvuspb.SearchRequest) (*milvuspb.SearchResults, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.SearchResults
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SearchRequest) (*milvuspb.SearchResults, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SearchRequest) *milvuspb.SearchResults); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3507,6 +3046,7 @@ func (_m *MilvusServiceServer) Search(_a0 context.Context, _a1 *milvuspb.SearchR
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.SearchRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3522,8 +3062,8 @@ type MilvusServiceServer_Search_Call struct {
 }
 
 // Search is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.SearchRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.SearchRequest
 func (_e *MilvusServiceServer_Expecter) Search(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_Search_Call {
 	return &MilvusServiceServer_Search_Call{Call: _e.mock.On("Search", _a0, _a1)}
 }
@@ -3540,20 +3080,11 @@ func (_c *MilvusServiceServer_Search_Call) Return(_a0 *milvuspb.SearchResults, _
 	return _c
 }
 
-func (_c *MilvusServiceServer_Search_Call) RunAndReturn(run func(context.Context, *milvuspb.SearchRequest) (*milvuspb.SearchResults, error)) *MilvusServiceServer_Search_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // SelectGrant provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) SelectGrant(_a0 context.Context, _a1 *milvuspb.SelectGrantRequest) (*milvuspb.SelectGrantResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.SelectGrantResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectGrantRequest) (*milvuspb.SelectGrantResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectGrantRequest) *milvuspb.SelectGrantResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3562,6 +3093,7 @@ func (_m *MilvusServiceServer) SelectGrant(_a0 context.Context, _a1 *milvuspb.Se
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.SelectGrantRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3577,8 +3109,8 @@ type MilvusServiceServer_SelectGrant_Call struct {
 }
 
 // SelectGrant is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.SelectGrantRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.SelectGrantRequest
 func (_e *MilvusServiceServer_Expecter) SelectGrant(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_SelectGrant_Call {
 	return &MilvusServiceServer_SelectGrant_Call{Call: _e.mock.On("SelectGrant", _a0, _a1)}
 }
@@ -3595,20 +3127,11 @@ func (_c *MilvusServiceServer_SelectGrant_Call) Return(_a0 *milvuspb.SelectGrant
 	return _c
 }
 
-func (_c *MilvusServiceServer_SelectGrant_Call) RunAndReturn(run func(context.Context, *milvuspb.SelectGrantRequest) (*milvuspb.SelectGrantResponse, error)) *MilvusServiceServer_SelectGrant_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // SelectRole provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) SelectRole(_a0 context.Context, _a1 *milvuspb.SelectRoleRequest) (*milvuspb.SelectRoleResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.SelectRoleResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectRoleRequest) (*milvuspb.SelectRoleResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectRoleRequest) *milvuspb.SelectRoleResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3617,6 +3140,7 @@ func (_m *MilvusServiceServer) SelectRole(_a0 context.Context, _a1 *milvuspb.Sel
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.SelectRoleRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3632,8 +3156,8 @@ type MilvusServiceServer_SelectRole_Call struct {
 }
 
 // SelectRole is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.SelectRoleRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.SelectRoleRequest
 func (_e *MilvusServiceServer_Expecter) SelectRole(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_SelectRole_Call {
 	return &MilvusServiceServer_SelectRole_Call{Call: _e.mock.On("SelectRole", _a0, _a1)}
 }
@@ -3650,20 +3174,11 @@ func (_c *MilvusServiceServer_SelectRole_Call) Return(_a0 *milvuspb.SelectRoleRe
 	return _c
 }
 
-func (_c *MilvusServiceServer_SelectRole_Call) RunAndReturn(run func(context.Context, *milvuspb.SelectRoleRequest) (*milvuspb.SelectRoleResponse, error)) *MilvusServiceServer_SelectRole_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // SelectUser provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) SelectUser(_a0 context.Context, _a1 *milvuspb.SelectUserRequest) (*milvuspb.SelectUserResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.SelectUserResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectUserRequest) (*milvuspb.SelectUserResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectUserRequest) *milvuspb.SelectUserResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3672,6 +3187,7 @@ func (_m *MilvusServiceServer) SelectUser(_a0 context.Context, _a1 *milvuspb.Sel
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.SelectUserRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3687,8 +3203,8 @@ type MilvusServiceServer_SelectUser_Call struct {
 }
 
 // SelectUser is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.SelectUserRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.SelectUserRequest
 func (_e *MilvusServiceServer_Expecter) SelectUser(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_SelectUser_Call {
 	return &MilvusServiceServer_SelectUser_Call{Call: _e.mock.On("SelectUser", _a0, _a1)}
 }
@@ -3705,20 +3221,11 @@ func (_c *MilvusServiceServer_SelectUser_Call) Return(_a0 *milvuspb.SelectUserRe
 	return _c
 }
 
-func (_c *MilvusServiceServer_SelectUser_Call) RunAndReturn(run func(context.Context, *milvuspb.SelectUserRequest) (*milvuspb.SelectUserResponse, error)) *MilvusServiceServer_SelectUser_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // ShowCollections provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) ShowCollections(_a0 context.Context, _a1 *milvuspb.ShowCollectionsRequest) (*milvuspb.ShowCollectionsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.ShowCollectionsResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowCollectionsRequest) (*milvuspb.ShowCollectionsResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowCollectionsRequest) *milvuspb.ShowCollectionsResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3727,6 +3234,7 @@ func (_m *MilvusServiceServer) ShowCollections(_a0 context.Context, _a1 *milvusp
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ShowCollectionsRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3742,8 +3250,8 @@ type MilvusServiceServer_ShowCollections_Call struct {
 }
 
 // ShowCollections is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.ShowCollectionsRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.ShowCollectionsRequest
 func (_e *MilvusServiceServer_Expecter) ShowCollections(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_ShowCollections_Call {
 	return &MilvusServiceServer_ShowCollections_Call{Call: _e.mock.On("ShowCollections", _a0, _a1)}
 }
@@ -3760,20 +3268,11 @@ func (_c *MilvusServiceServer_ShowCollections_Call) Return(_a0 *milvuspb.ShowCol
 	return _c
 }
 
-func (_c *MilvusServiceServer_ShowCollections_Call) RunAndReturn(run func(context.Context, *milvuspb.ShowCollectionsRequest) (*milvuspb.ShowCollectionsResponse, error)) *MilvusServiceServer_ShowCollections_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // ShowPartitions provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) ShowPartitions(_a0 context.Context, _a1 *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *milvuspb.ShowPartitionsResponse
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowPartitionsRequest) *milvuspb.ShowPartitionsResponse); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3782,6 +3281,7 @@ func (_m *MilvusServiceServer) ShowPartitions(_a0 context.Context, _a1 *milvuspb
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ShowPartitionsRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3797,8 +3297,8 @@ type MilvusServiceServer_ShowPartitions_Call struct {
 }
 
 // ShowPartitions is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.ShowPartitionsRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.ShowPartitionsRequest
 func (_e *MilvusServiceServer_Expecter) ShowPartitions(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_ShowPartitions_Call {
 	return &MilvusServiceServer_ShowPartitions_Call{Call: _e.mock.On("ShowPartitions", _a0, _a1)}
 }
@@ -3815,20 +3315,11 @@ func (_c *MilvusServiceServer_ShowPartitions_Call) Return(_a0 *milvuspb.ShowPart
 	return _c
 }
 
-func (_c *MilvusServiceServer_ShowPartitions_Call) RunAndReturn(run func(context.Context, *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error)) *MilvusServiceServer_ShowPartitions_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // TransferNode provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) TransferNode(_a0 context.Context, _a1 *milvuspb.TransferNodeRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.TransferNodeRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.TransferNodeRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3837,6 +3328,7 @@ func (_m *MilvusServiceServer) TransferNode(_a0 context.Context, _a1 *milvuspb.T
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.TransferNodeRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3852,8 +3344,8 @@ type MilvusServiceServer_TransferNode_Call struct {
 }
 
 // TransferNode is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.TransferNodeRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.TransferNodeRequest
 func (_e *MilvusServiceServer_Expecter) TransferNode(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_TransferNode_Call {
 	return &MilvusServiceServer_TransferNode_Call{Call: _e.mock.On("TransferNode", _a0, _a1)}
 }
@@ -3870,20 +3362,11 @@ func (_c *MilvusServiceServer_TransferNode_Call) Return(_a0 *commonpb.Status, _a
 	return _c
 }
 
-func (_c *MilvusServiceServer_TransferNode_Call) RunAndReturn(run func(context.Context, *milvuspb.TransferNodeRequest) (*commonpb.Status, error)) *MilvusServiceServer_TransferNode_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // TransferReplica provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) TransferReplica(_a0 context.Context, _a1 *milvuspb.TransferReplicaRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.TransferReplicaRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.TransferReplicaRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3892,6 +3375,7 @@ func (_m *MilvusServiceServer) TransferReplica(_a0 context.Context, _a1 *milvusp
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.TransferReplicaRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3907,8 +3391,8 @@ type MilvusServiceServer_TransferReplica_Call struct {
 }
 
 // TransferReplica is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.TransferReplicaRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.TransferReplicaRequest
 func (_e *MilvusServiceServer_Expecter) TransferReplica(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_TransferReplica_Call {
 	return &MilvusServiceServer_TransferReplica_Call{Call: _e.mock.On("TransferReplica", _a0, _a1)}
 }
@@ -3925,20 +3409,11 @@ func (_c *MilvusServiceServer_TransferReplica_Call) Return(_a0 *commonpb.Status,
 	return _c
 }
 
-func (_c *MilvusServiceServer_TransferReplica_Call) RunAndReturn(run func(context.Context, *milvuspb.TransferReplicaRequest) (*commonpb.Status, error)) *MilvusServiceServer_TransferReplica_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 // UpdateCredential provides a mock function with given fields: _a0, _a1
 func (_m *MilvusServiceServer) UpdateCredential(_a0 context.Context, _a1 *milvuspb.UpdateCredentialRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
 	var r0 *commonpb.Status
-	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.UpdateCredentialRequest) (*commonpb.Status, error)); ok {
-		return rf(_a0, _a1)
-	}
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.UpdateCredentialRequest) *commonpb.Status); ok {
 		r0 = rf(_a0, _a1)
 	} else {
@@ -3947,6 +3422,7 @@ func (_m *MilvusServiceServer) UpdateCredential(_a0 context.Context, _a1 *milvus
 		}
 	}
 
+	var r1 error
 	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.UpdateCredentialRequest) error); ok {
 		r1 = rf(_a0, _a1)
 	} else {
@@ -3962,8 +3438,8 @@ type MilvusServiceServer_UpdateCredential_Call struct {
 }
 
 // UpdateCredential is a helper method to define mock.On call
-//   - _a0 context.Context
-//   - _a1 *milvuspb.UpdateCredentialRequest
+//  - _a0 context.Context
+//  - _a1 *milvuspb.UpdateCredentialRequest
 func (_e *MilvusServiceServer_Expecter) UpdateCredential(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_UpdateCredential_Call {
 	return &MilvusServiceServer_UpdateCredential_Call{Call: _e.mock.On("UpdateCredential", _a0, _a1)}
 }
@@ -3980,11 +3456,6 @@ func (_c *MilvusServiceServer_UpdateCredential_Call) Return(_a0 *commonpb.Status
 	return _c
 }
 
-func (_c *MilvusServiceServer_UpdateCredential_Call) RunAndReturn(run func(context.Context, *milvuspb.UpdateCredentialRequest) (*commonpb.Status, error)) *MilvusServiceServer_UpdateCredential_Call {
-	_c.Call.Return(run)
-	return _c
-}
-
 type mockConstructorTestingTNewMilvusServiceServer interface {
 	mock.TestingT
 	Cleanup(func())


Test patch:
diff --git a/client/client_grpc_admin.go b/client/admin.go
similarity index 100%
rename from client/client_grpc_admin.go
rename to client/admin.go
diff --git a/client/client_grpc_alias.go b/client/alias.go
similarity index 100%
rename from client/client_grpc_alias.go
rename to client/alias.go
diff --git a/client/client_grpc_alias_test.go b/client/alias_test.go
similarity index 100%
rename from client/client_grpc_alias_test.go
rename to client/alias_test.go
diff --git a/client/client_grpc_authentication.go b/client/authentication.go
similarity index 100%
rename from client/client_grpc_authentication.go
rename to client/authentication.go
diff --git a/client/client_grpc_authentication_test.go b/client/authentication_test.go
similarity index 100%
rename from client/client_grpc_authentication_test.go
rename to client/authentication_test.go
diff --git a/client/client_mock_test.go b/client/client_mock_test.go
index ac5b2d4..8208f3e 100644
--- a/client/client_mock_test.go
+++ b/client/client_mock_test.go
@@ -9,6 +9,8 @@ import (
 	"github.com/golang/protobuf/proto"
 	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
 	server "github.com/milvus-io/milvus-proto/go-api/milvuspb"
+	schema "github.com/milvus-io/milvus-proto/go-api/schemapb"
+	"github.com/milvus-io/milvus-sdk-go/v2/entity"
 	"github.com/milvus-io/milvus-sdk-go/v2/mocks"
 	"github.com/stretchr/testify/mock"
 	"github.com/stretchr/testify/suite"
@@ -77,17 +79,132 @@ func (s *MockSuiteBase) resetMock() {
 	}
 }
 
-func (s *MockSuiteBase) setupHasCollection(collName string) {
+func (s *MockSuiteBase) setupHasCollection(collNames ...string) {
 	s.mock.EXPECT().HasCollection(mock.Anything, mock.AnythingOfType("*milvuspb.HasCollectionRequest")).
 		Call.Return(func(ctx context.Context, req *server.HasCollectionRequest) *server.BoolResponse {
 		resp := &server.BoolResponse{Status: &common.Status{}}
+		for _, collName := range collNames {
+			if req.GetCollectionName() == collName {
+				resp.Value = true
+				break
+			}
+		}
+		return resp
+	}, nil)
+}
+
+func (s *MockSuiteBase) setupHasCollectionError(errorCode common.ErrorCode, err error) {
+	s.mock.EXPECT().HasCollection(mock.Anything, mock.AnythingOfType("*milvuspb.HasCollectionRequest")).
+		Return(&server.BoolResponse{
+			Status: &common.Status{ErrorCode: errorCode},
+		}, err)
+}
+
+func (s *MockSuiteBase) setupHasPartition(collName string, partNames ...string) {
+	s.mock.EXPECT().HasPartition(mock.Anything, mock.AnythingOfType("*milvuspb.HasPartitionRequest")).
+		Call.Return(func(ctx context.Context, req *server.HasPartitionRequest) *server.BoolResponse {
+		resp := &server.BoolResponse{Status: &common.Status{}}
 		if req.GetCollectionName() == collName {
-			resp.Value = true
+			for _, partName := range partNames {
+				if req.GetPartitionName() == partName {
+					resp.Value = true
+					break
+				}
+			}
 		}
 		return resp
 	}, nil)
 }
 
+func (s *MockSuiteBase) setupHasPartitionError(errorCode common.ErrorCode, err error) {
+	s.mock.EXPECT().HasPartition(mock.Anything, mock.AnythingOfType("*milvuspb.HasPartitionRequest")).
+		Return(&server.BoolResponse{
+			Status: &common.Status{ErrorCode: errorCode},
+		}, err)
+}
+
+func (s *MockSuiteBase) setupDescribeCollection(collName string, schema *entity.Schema) {
+	s.mock.EXPECT().DescribeCollection(mock.Anything, mock.AnythingOfType("*milvuspb.DescribeCollectionRequest")).
+		Call.Return(func(ctx context.Context, req *server.DescribeCollectionRequest) *server.DescribeCollectionResponse {
+		return &server.DescribeCollectionResponse{
+			Status: &common.Status{ErrorCode: common.ErrorCode_Success},
+			Schema: schema.ProtoMessage(),
+		}
+	}, nil)
+}
+
+func (s *MockSuiteBase) setupDescribeCollectionError(errorCode common.ErrorCode, err error) {
+	s.mock.EXPECT().DescribeCollection(mock.Anything, mock.AnythingOfType("*milvuspb.DescribeCollectionRequest")).
+		Return(&server.DescribeCollectionResponse{
+			Status: &common.Status{ErrorCode: errorCode},
+		}, err)
+}
+
+func (s *MockSuiteBase) getInt64FieldData(name string, data []int64) *schema.FieldData {
+	return &schema.FieldData{
+		Type:      schema.DataType_Int64,
+		FieldName: name,
+		Field: &schema.FieldData_Scalars{
+			Scalars: &schema.ScalarField{
+				Data: &schema.ScalarField_LongData{
+					LongData: &schema.LongArray{
+						Data: data,
+					},
+				},
+			},
+		},
+	}
+}
+
+func (s *MockSuiteBase) getVarcharFieldData(name string, data []string) *schema.FieldData {
+	return &schema.FieldData{
+		Type:      schema.DataType_VarChar,
+		FieldName: name,
+		Field: &schema.FieldData_Scalars{
+			Scalars: &schema.ScalarField{
+				Data: &schema.ScalarField_StringData{
+					StringData: &schema.StringArray{
+						Data: data,
+					},
+				},
+			},
+		},
+	}
+}
+
+func (s *MockSuiteBase) getJSONBytesFieldData(name string, data [][]byte) *schema.FieldData {
+	return &schema.FieldData{
+		Type:      schema.DataType_JSON,
+		FieldName: name,
+		Field: &schema.FieldData_Scalars{
+			Scalars: &schema.ScalarField{
+				Data: &schema.ScalarField_JsonData{
+					JsonData: &schema.JSONArray{
+						Data: data,
+					},
+				},
+			},
+		},
+	}
+}
+
+func (s *MockSuiteBase) getFloatVectorFieldData(name string, dim int64, data []float32) *schema.FieldData {
+	return &schema.FieldData{
+		Type:      schema.DataType_FloatVector,
+		FieldName: name,
+		Field: &schema.FieldData_Vectors{
+			Vectors: &schema.VectorField{
+				Dim: dim,
+				Data: &schema.VectorField_FloatVector{
+					FloatVector: &schema.FloatArray{
+						Data: data,
+					},
+				},
+			},
+		},
+	}
+}
+
 // ref https://stackoverflow.com/questions/42102496/testing-a-grpc-service
 
 var errNotImplemented = errors.New("not implemented")
@@ -788,6 +905,10 @@ func (m *MockServer) GetFlushAllState(_a0 context.Context, _a1 *server.GetFlushA
 	panic("not implemented")
 }
 
+func (m *MockServer) Connect(_ context.Context, _ *server.ConnectRequest) (*server.ConnectResponse, error) {
+	panic("not implemented") // TODO: Implement
+}
+
 func SuccessStatus() (*common.Status, error) {
 	return &common.Status{ErrorCode: common.ErrorCode_Success}, nil
 }
diff --git a/entity/columns_json_test.go b/entity/columns_json_test.go
new file mode 100644
index 0000000..90da76e
--- /dev/null
+++ b/entity/columns_json_test.go
@@ -0,0 +1,78 @@
+package entity
+
+import (
+	"fmt"
+	"math/rand"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/suite"
+)
+
+type ColumnJSONBytesSuite struct {
+	suite.Suite
+}
+
+func (s *ColumnJSONBytesSuite) SetupSuite() {
+	rand.Seed(time.Now().UnixNano())
+}
+
+func (s *ColumnJSONBytesSuite) TestAttrMethods() {
+	columnName := fmt.Sprintf("column_jsonbs_%d", rand.Int())
+	columnLen := 8 + rand.Intn(10)
+
+	v := make([][]byte, columnLen)
+	column := NewColumnJSONBytes(columnName, v)
+
+	s.Run("test_meta", func() {
+		ft := FieldTypeJSON
+		s.Equal("JSON", ft.Name())
+		s.Equal("JSON", ft.String())
+		pbName, pbType := ft.PbFieldType()
+		s.Equal("JSON", pbName)
+		s.Equal("JSON", pbType)
+	})
+
+	s.Run("test_column_attribute", func() {
+		s.Equal(columnName, column.Name())
+		s.Equal(FieldTypeJSON, column.Type())
+		s.Equal(columnLen, column.Len())
+		s.EqualValues(v, column.Data())
+	})
+
+	s.Run("test_column_field_data", func() {
+		fd := column.FieldData()
+		s.NotNil(fd)
+		s.Equal(fd.GetFieldName(), columnName)
+	})
+
+	s.Run("test_column_valuer_by_idx", func() {
+		_, err := column.ValueByIdx(-1)
+		s.Error(err)
+		_, err = column.ValueByIdx(columnLen)
+		s.Error(err)
+		for i := 0; i < columnLen; i++ {
+			v, err := column.ValueByIdx(i)
+			s.NoError(err)
+			s.Equal(column.values[i], v)
+		}
+	})
+
+	s.Run("test_append_value", func() {
+		item := make([]byte, 10)
+		err := column.AppendValue(item)
+		s.NoError(err)
+		s.Equal(columnLen+1, column.Len())
+		val, err := column.ValueByIdx(columnLen)
+		s.NoError(err)
+		s.Equal(item, val)
+
+		err = column.AppendValue(1)
+		s.Error(err)
+	})
+
+}
+
+func TestColumnJSONBytes(t *testing.T) {
+	suite.Run(t, new(ColumnJSONBytesSuite))
+}