chore: reorg imports, format code, make MaxRetires default to 0 (#1165)

* chore: reverse the order of stopping services

* chore: reverse the order of stopping services

* chore: reorg imports, format code

* chore: format code, and refactor

* feat: change MaxRetries default to 0, disable retry

core/retry/backoff/backoff_test.go

@@ -1,9 +1,10 @@
 package backoff
 
 import (
-	"github.com/stretchr/testify/assert"
 	"testing"
 	"time"
+
+	"github.com/stretchr/testify/assert"
 )
 
 func TestWaitBetween(t *testing.T) {

core/retry/options.go

@@ -1,14 +1,14 @@
 package retry
 
 import (
+	"time"
+
 	"github.com/tal-tech/go-zero/core/retry/backoff"
 	"google.golang.org/grpc/codes"
-	"time"
 )
 
 // WithDisable disables the retry behaviour on this call, or this interceptor.
-//
-// Its semantically the same to `WithMax`
+// It's semantically the same to `WithMax(0)`
 func WithDisable() *CallOption {
 	return WithMax(0)
 }

core/retry/options_test.go

@@ -3,14 +3,15 @@ package retry
 import (
 	"context"
 	"errors"
+	"testing"
+	"time"
+
 	"github.com/stretchr/testify/assert"
 	"github.com/tal-tech/go-zero/core/logx"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/metadata"
 	"google.golang.org/grpc/status"
-	"testing"
-	"time"
 )
 
 func TestRetryWithDisable(t *testing.T) {
@@ -33,7 +34,6 @@ func TestRetryWithBackoff(t *testing.T) {
 		return time.Millisecond
 	}))
 	assert.EqualValues(t, time.Millisecond, retryCallOptions.backoffFunc(1))
-
 }
 
 func TestRetryWithCodes(t *testing.T) {
@@ -50,7 +50,6 @@ func TestRetryWithPerRetryTimeout(t *testing.T) {
 }
 
 func Test_waitRetryBackoff(t *testing.T) {
-
 	opt := &options{perCallTimeout: time.Second, backoffFunc: func(attempt int) time.Duration {
 		return time.Second
 	}}
@@ -76,7 +75,6 @@ func Test_perCallContext(t *testing.T) {
 	md, ok := metadata.FromOutgoingContext(callContext)
 	assert.True(t, ok)
 	assert.EqualValues(t, metadata.MD{"1": {"1"}, AttemptMetadataKey: {"1"}}, md)
-
 }
 
 func Test_filterCallOptions(t *testing.T) {
@@ -88,5 +86,4 @@ func Test_filterCallOptions(t *testing.T) {
 	})
 	assert.EqualValues(t, []grpc.CallOption{grpcEmptyCallOpt}, options)
 	assert.EqualValues(t, []*CallOption{retryCallOpt}, retryCallOptions)
-
 }

core/retry/retryinterceptor.go

@@ -2,16 +2,15 @@ package retry
 
 import (
 	"context"
+	"strconv"
+	"time"
+
 	"github.com/tal-tech/go-zero/core/logx"
 	"github.com/tal-tech/go-zero/core/retry/backoff"
-
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/metadata"
 	"google.golang.org/grpc/status"
-
-	"strconv"
-	"time"
 )
 
 const AttemptMetadataKey = "x-retry-attempt"
@@ -38,6 +37,7 @@ type (
 		codes              []codes.Code
 		backoffFunc        backoff.Func
 	}
+
 	// CallOption is a grpc.CallOption that is local to grpc retry.
 	CallOption struct {
 		grpc.EmptyCallOption // make sure we implement private after() and before() fields so we don't panic.
@@ -65,6 +65,7 @@ func waitRetryBackoff(logger logx.Logger, attempt int, ctx context.Context, retr
 			}
 		}
 	}
+
 	return nil
 }
 
@@ -73,11 +74,13 @@ func isRetriable(err error, retryOptions *options) bool {
 	if isContextError(err) {
 		return false
 	}
+
 	for _, code := range retryOptions.codes {
 		if code == errCode {
 			return true
 		}
 	}
+
 	return false
 }
 
@@ -90,6 +93,7 @@ func reuseOrNewWithCallOptions(opt *options, retryCallOptions []*CallOption) *op
 	if len(retryCallOptions) == 0 {
 		return opt
 	}
+
 	return parseRetryCallOptions(opt, retryCallOptions...)
 }
 
@@ -97,6 +101,7 @@ func parseRetryCallOptions(opt *options, opts ...*CallOption) *options {
 	for _, option := range opts {
 		option.apply(opt)
 	}
+
 	return opt
 }
 
@@ -122,7 +127,7 @@ func extractIncomingAndClone(ctx context.Context) metadata.MD {
 	if !ok {
 		return metadata.MD{}
 	}
-	// clone
+
 	return md.Copy()
 }
 
@@ -134,6 +139,7 @@ func filterCallOptions(callOptions []grpc.CallOption) (grpcOptions []grpc.CallOp
 			grpcOptions = append(grpcOptions, opt)
 		}
 	}
+
 	return grpcOptions, retryOptions
 }
 
@@ -145,6 +151,7 @@ func Do(ctx context.Context, call func(ctx context.Context, opts ...grpc.CallOpt
 	if callOpts.max == 0 {
 		return call(ctx, opts...)
 	}
+
 	var lastErr error
 	for attempt := 0; attempt <= callOpts.max; attempt++ {
 		if err := waitRetryBackoff(logger, attempt, ctx, callOpts); err != nil {
@@ -157,6 +164,7 @@ func Do(ctx context.Context, call func(ctx context.Context, opts ...grpc.CallOpt
 		if lastErr == nil {
 			return nil
 		}
+
 		if attempt == 0 {
 			logger.Errorf("grpc call failed, got err: %v", lastErr)
 		} else {
@@ -175,5 +183,6 @@ func Do(ctx context.Context, call func(ctx context.Context, opts ...grpc.CallOpt
 			return lastErr
 		}
 	}
+
 	return lastErr
 }

core/retry/retryinterceptor_test.go

@@ -2,11 +2,12 @@ package retry
 
 import (
 	"context"
+	"testing"
+
 	"github.com/stretchr/testify/assert"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/status"
-	"testing"
 )
 
 func TestDo(t *testing.T) {
@@ -16,10 +17,8 @@ func TestDo(t *testing.T) {
 		err := Do(context.Background(), func(ctx context.Context, opts ...grpc.CallOption) error {
 			count++
 			return status.Error(codes.ResourceExhausted, "ResourceExhausted")
-
 		}, WithMax(i))
 		assert.Error(t, err)
 		assert.Equal(t, i+1, count)
 	}
-
 }