Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

etcdserver: parse context error for better message

Browse Source
This commit is contained in:
Yicheng Qin 2014-10-26 11:35:59 -07:00
parent 91a4aa151a
commit e77f8e311c
2 changed files with 29 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
etcdserver/server.go
View File

@ -64,6 +64,8 @@ var (
ErrIDRemoved = errors.New("etcdserver: ID removed") ErrIDRemoved = errors.New("etcdserver: ID removed")
ErrIDExists = errors.New("etcdserver: ID exists") ErrIDExists = errors.New("etcdserver: ID exists")
ErrIDNotFound = errors.New("etcdserver: ID not found") ErrIDNotFound = errors.New("etcdserver: ID not found")
ErrCanceled = errors.New("etcdserver: request cancelled")
ErrTimeout = errors.New("etcdserver: request timed out")
storeMembersPrefix = path.Join(StoreAdminPrefix, "members") storeMembersPrefix = path.Join(StoreAdminPrefix, "members")
storeRemovedMembersPrefix = path.Join(StoreAdminPrefix, "removed_members") storeRemovedMembersPrefix = path.Join(StoreAdminPrefix, "removed_members")
@ -383,7 +385,7 @@ func (s *EtcdServer) Do(ctx context.Context, r pb.Request) (Response, error) {
return resp, resp.err return resp, resp.err
case <-ctx.Done(): case <-ctx.Done():
s.w.Trigger(r.ID, nil) // GC wait s.w.Trigger(r.ID, nil) // GC wait
return Response{}, ctx.Err() return Response{}, parseCtxErr(ctx.Err())
case <-s.done: case <-s.done:
return Response{}, ErrStopped return Response{}, ErrStopped
} }
@ -477,7 +479,7 @@ func (s *EtcdServer) configure(ctx context.Context, cc raftpb.ConfChange) error
return nil return nil
case <-ctx.Done(): case <-ctx.Done():
s.w.Trigger(cc.ID, nil) // GC wait s.w.Trigger(cc.ID, nil) // GC wait
return ctx.Err() return parseCtxErr(ctx.Err())
case <-s.done: case <-s.done:
return ErrStopped return ErrStopped
} }
@ -752,6 +754,17 @@ func GenID() (n uint64) {
return return
} }
func parseCtxErr(err error) error {
switch err {
case context.Canceled:
return ErrCanceled
case context.DeadlineExceeded:
return ErrTimeout
default:
return err
}
}
func getBool(v *bool) (vv bool, set bool) { func getBool(v *bool) (vv bool, set bool) {
if v == nil { if v == nil {
return false, false return false, false

16
etcdserver/server_test.go
View File

@ -614,8 +614,8 @@ func TestDoProposalCancelled(t *testing.T) {
if len(gaction) != 0 { if len(gaction) != 0 {
t.Errorf("len(action) = %v, want 0", len(gaction)) t.Errorf("len(action) = %v, want 0", len(gaction))
} }
if err != context.Canceled { if err != ErrCanceled {
t.Fatalf("err = %v, want %v", err, context.Canceled) t.Fatalf("err = %v, want %v", err, ErrCanceled)
} }
w := []action{action{name: "Register1"}, action{name: "Trigger1"}} w := []action{action{name: "Register1"}, action{name: "Trigger1"}}
if !reflect.DeepEqual(wait.action, w) { if !reflect.DeepEqual(wait.action, w) {
@ -623,6 +623,18 @@ func TestDoProposalCancelled(t *testing.T) {
} }
} }
func TestDoProposalTimeout(t *testing.T) {
ctx, _ := context.WithTimeout(context.Background(), 0)
srv := &EtcdServer{
node: &nodeRecorder{},
w: &waitRecorder{},
}
_, err := srv.Do(ctx, pb.Request{Method: "PUT", ID: 1})
if err != ErrTimeout {
t.Fatalf("err = %v, want %v", err, ErrTimeout)
}
}
func TestDoProposalStopped(t *testing.T) { func TestDoProposalStopped(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()