123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566 |
- // Licensed to the LF AI & Data foundation under one
- // or more contributor license agreements. See the NOTICE file
- // distributed with this work for additional information
- // regarding copyright ownership. The ASF licenses this file
- // to you under the Apache License, Version 2.0 (the
- // "License"); you may not use this file except in compliance
- // with the License. You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- package client
- import (
- "context"
- "google.golang.org/grpc"
- "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
- "github.com/milvus-io/milvus/pkg/util/merr"
- )
- func (c *Client) UsingDatabase(ctx context.Context, option UsingDatabaseOption) error {
- dbName := option.DbName()
- c.usingDatabase(dbName)
- return c.connectInternal(ctx)
- }
- func (c *Client) ListDatabase(ctx context.Context, option ListDatabaseOption, callOptions ...grpc.CallOption) (databaseNames []string, err error) {
- req := option.Request()
- err = c.callService(func(milvusService milvuspb.MilvusServiceClient) error {
- resp, err := milvusService.ListDatabases(ctx, req, callOptions...)
- err = merr.CheckRPCCall(resp, err)
- if err != nil {
- return err
- }
- databaseNames = resp.GetDbNames()
- return nil
- })
- return databaseNames, err
- }
- func (c *Client) CreateDatabase(ctx context.Context, option CreateDatabaseOption, callOptions ...grpc.CallOption) error {
- req := option.Request()
- return c.callService(func(milvusService milvuspb.MilvusServiceClient) error {
- resp, err := milvusService.CreateDatabase(ctx, req, callOptions...)
- return merr.CheckRPCCall(resp, err)
- })
- }
- func (c *Client) DropDatabase(ctx context.Context, option DropDatabaseOption, callOptions ...grpc.CallOption) error {
- req := option.Request()
- return c.callService(func(milvusService milvuspb.MilvusServiceClient) error {
- resp, err := milvusService.DropDatabase(ctx, req, callOptions...)
- return merr.CheckRPCCall(resp, err)
- })
- }
|