Skip to content

Commit 9938385

Browse files
authored
fix: convert rpcerr to ErrorType (#580)
Signed-off-by: jyjiangkai <[email protected]>
1 parent ba347ae commit 9938385

File tree

5 files changed

+23
-12
lines changed

5 files changed

+23
-12
lines changed

pkg/cluster/auth.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ import (
2020
"google.golang.org/protobuf/types/known/wrapperspb"
2121

2222
"github.com/vanus-labs/vanus/pkg/cluster/raw_client"
23+
"github.com/vanus-labs/vanus/pkg/errors"
2324
ctrlpb "github.com/vanus-labs/vanus/proto/pkg/controller"
2425
metapb "github.com/vanus-labs/vanus/proto/pkg/meta"
2526
)
@@ -31,15 +32,15 @@ type authService struct {
3132
func (a *authService) GetUserRole(ctx context.Context, user string) ([]*metapb.UserRole, error) {
3233
resp, err := a.client.GetUserRole(ctx, &ctrlpb.GetUserRoleRequest{UserIdentifier: user})
3334
if err != nil {
34-
return nil, err
35+
return nil, errors.To(err)
3536
}
3637
return resp.GetUserRole(), nil
3738
}
3839

3940
func (a *authService) GetUserByToken(ctx context.Context, token string) (string, error) {
4041
user, err := a.client.GetUserByToken(ctx, wrapperspb.String(token))
4142
if err != nil {
42-
return "", err
43+
return "", errors.To(err)
4344
}
4445
return user.GetValue(), nil
4546
}

pkg/cluster/eventbus.go

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -44,14 +44,14 @@ func (es *eventbusService) GetEventbusByName(ctx context.Context, ns, name strin
4444

4545
pb, err := es.nsSvc.GetNamespaceByName(ctx, ns)
4646
if err != nil {
47-
return nil, err
47+
return nil, errors.To(err)
4848
}
4949
eb, err := es.client.GetEventbusWithHumanFriendly(ctx, &ctrlpb.GetEventbusWithHumanFriendlyRequest{
5050
NamespaceId: pb.Id,
5151
EventbusName: name,
5252
})
5353
if err != nil {
54-
return nil, err
54+
return nil, errors.To(err)
5555
}
5656

5757
// es.cache.Store(key, eb) unmask when dirty cache is resolved
@@ -66,7 +66,7 @@ func (es *eventbusService) GetEventbus(ctx context.Context, id uint64) (*meta.Ev
6666

6767
eb, err := es.client.GetEventbus(ctx, wrapperspb.UInt64(id))
6868
if err != nil {
69-
return nil, err
69+
return nil, errors.To(err)
7070
}
7171

7272
// es.cache.Store(id, eb) unmask when dirty cache is resolved
@@ -93,20 +93,21 @@ func (es *eventbusService) CreateSystemEventbusIfNotExist(ctx context.Context, n
9393

9494
nsPb, err := es.nsSvc.GetSystemNamespace(ctx)
9595
if err != nil {
96-
return nil, err
96+
return nil, errors.To(err)
9797
}
9898

99-
return es.client.CreateSystemEventbus(ctx, &ctrlpb.CreateEventbusRequest{
99+
eventbus, err := es.client.CreateSystemEventbus(ctx, &ctrlpb.CreateEventbusRequest{
100100
Name: name,
101101
LogNumber: int32(defaultSystemEventbusEventlog),
102102
Description: desc,
103103
NamespaceId: nsPb.Id,
104104
})
105+
return eventbus, errors.To(err)
105106
}
106107

107108
func (es *eventbusService) Delete(ctx context.Context, id uint64) error {
108109
_, err := es.client.DeleteEventbus(ctx, wrapperspb.UInt64(id))
109-
return err
110+
return errors.To(err)
110111
}
111112

112113
func (es *eventbusService) RawClient() ctrlpb.EventbusControllerClient {

pkg/cluster/namespace.go

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,12 @@ package cluster
1616

1717
import (
1818
"context"
19-
"google.golang.org/protobuf/types/known/wrapperspb"
2019
"sync"
2120

21+
"google.golang.org/protobuf/types/known/wrapperspb"
22+
2223
"github.com/vanus-labs/vanus/pkg/cluster/raw_client"
24+
"github.com/vanus-labs/vanus/pkg/errors"
2325
ctrlpb "github.com/vanus-labs/vanus/proto/pkg/controller"
2426
metapb "github.com/vanus-labs/vanus/proto/pkg/meta"
2527
)
@@ -41,7 +43,7 @@ func (ns *namespaceService) GetNamespace(ctx context.Context, id uint64) (*metap
4143
}
4244
n, err := ns.client.GetNamespace(ctx, &ctrlpb.GetNamespaceRequest{Id: id})
4345
if err != nil {
44-
return nil, err
46+
return nil, errors.To(err)
4547
}
4648
// ns.cache.Store(id, n) unmask when dirty cache is resolved
4749
return n, nil
@@ -62,7 +64,7 @@ func (ns *namespaceService) GetNamespaceByName(ctx context.Context, name string)
6264
}
6365
n, err := ns.client.GetNamespaceWithHumanFriendly(ctx, wrapperspb.String(name))
6466
if err != nil {
65-
return nil, err
67+
return nil, errors.To(err)
6668
}
6769
// ns.cache.Store(name, n) unmask when dirty cache is resolved
6870
return n, nil

pkg/cluster/trigger.go

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ import (
88
metapb "github.com/vanus-labs/vanus/proto/pkg/meta"
99

1010
"github.com/vanus-labs/vanus/pkg/cluster/raw_client"
11+
"github.com/vanus-labs/vanus/pkg/errors"
1112
)
1213

1314
type triggerService struct {
@@ -27,5 +28,6 @@ func (es *triggerService) RegisterHeartbeat(ctx context.Context, interval time.D
2728
}
2829

2930
func (es *triggerService) GetSubscription(ctx context.Context, id uint64) (*metapb.Subscription, error) {
30-
return es.client.GetSubscription(ctx, &ctrlpb.GetSubscriptionRequest{Id: id})
31+
subscription, err := es.client.GetSubscription(ctx, &ctrlpb.GetSubscriptionRequest{Id: id})
32+
return subscription, errors.To(err)
3133
}

pkg/errors/utils.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,11 @@ func Is(err error, target error) bool {
6666
return errType != nil && errType.Code == targetType.Code
6767
}
6868

69+
func To(err error) *ErrorType {
70+
et, _ := FromError(err)
71+
return et
72+
}
73+
6974
func FromError(err error) (*ErrorType, bool) {
7075
if err == nil {
7176
return nil, true

0 commit comments

Comments
 (0)