Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement MsgCount for Tigris #928

Merged
merged 22 commits into from
Jul 26, 2022
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix dealing with errors
  • Loading branch information
Elena Grahovac committed Jul 21, 2022
commit 231fb803f655de3e204ce6f562d4edb53569c53d
16 changes: 8 additions & 8 deletions internal/handlers/tigris/fetch.go
Original file line number Diff line number Diff line change
Expand Up @@ -16,9 +16,8 @@ package tigris

import (
"context"
"errors"

tigris "github.com/tigrisdata/tigris-client-go/api/server/v1"
api "github.com/tigrisdata/tigris-client-go/api/server/v1"
"github.com/tigrisdata/tigris-client-go/driver"
"go.uber.org/zap"

Expand All @@ -40,20 +39,21 @@ func (h *Handler) fetch(ctx context.Context, param fetchParam) ([]*types.Documen
db := h.driver.UseDatabase(param.db)

collection, err := db.DescribeCollection(ctx, param.collection)
if err != nil {
var tigErr *tigris.TigrisError
if !errors.As(err, &tigErr) {
return nil, lazyerrors.Error(err)
}
switch err := err.(type) {
case nil:
// do nothing
case *driver.Error:
//nolint:nosnakecase // Tigris named their const that way
if tigErr.Code == tigris.Code_NOT_FOUND {
if err.Code == api.Code_NOT_FOUND {
h.L.Debug(
"Collection doesn't exist, handling a case to deal with a non-existing collection (return empty list)",
zap.String("db", param.db), zap.String("collection", param.collection),
)
return []*types.Document{}, nil
}
return nil, lazyerrors.Error(err)
default:
return nil, lazyerrors.Error(err)
}

var schema tjson.Schema
Expand Down
18 changes: 10 additions & 8 deletions internal/handlers/tigris/msg_drop.go
Original file line number Diff line number Diff line change
Expand Up @@ -16,9 +16,9 @@ package tigris

import (
"context"
"errors"

tigris "github.com/tigrisdata/tigris-client-go/api/server/v1"
api "github.com/tigrisdata/tigris-client-go/api/server/v1"
"github.com/tigrisdata/tigris-client-go/driver"

"github.com/FerretDB/FerretDB/internal/handlers/common"
"github.com/FerretDB/FerretDB/internal/types"
Expand Down Expand Up @@ -46,16 +46,18 @@ func (h *Handler) MsgDrop(ctx context.Context, msg *wire.OpMsg) (*wire.OpMsg, er
return nil, err
}

if err = h.driver.UseDatabase(db).DropCollection(ctx, collection); err != nil {
var tigErr *tigris.TigrisError
if !errors.As(err, &tigErr) {
return nil, lazyerrors.Error(err)
}
err = h.driver.UseDatabase(db).DropCollection(ctx, collection)
switch err := err.(type) {
case nil:
// do nothing
case *driver.Error:
//nolint:nosnakecase // Tigris named their const that way
if tigErr.Code == tigris.Code_NOT_FOUND {
if err.Code == api.Code_NOT_FOUND {
AlekSi marked this conversation as resolved.
Show resolved Hide resolved
return nil, common.NewErrorMsg(common.ErrNamespaceNotFound, "ns not found")
}
return nil, lazyerrors.Error(err)
default:
return nil, lazyerrors.Error(err)
}

var reply wire.OpMsg
Expand Down