mirror of https://github.com/tidwall/tile38.git
101 lines
2.9 KiB
Go
101 lines
2.9 KiB
Go
|
// Copyright 2016 Google LLC
|
||
|
|
//
|
||
|
|
// Licensed 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 pubsub
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"math"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
gax "github.com/googleapis/gax-go/v2"
|
||
|
|
pb "google.golang.org/genproto/googleapis/pubsub/v1"
|
||
|
|
"google.golang.org/grpc/codes"
|
||
|
|
"google.golang.org/grpc/status"
|
||
|
|
)
|
||
|
|
|
||
|
|
// maxPayload is the maximum number of bytes to devote to the
|
||
|
|
// encoded AcknowledgementRequest / ModifyAckDeadline proto message.
|
||
|
|
//
|
||
|
|
// With gRPC there is no way for the client to know the server's max message size (it is
|
||
|
|
// configurable on the server). We know from experience that it
|
||
|
|
// it 512K.
|
||
|
|
const (
|
||
|
|
maxPayload = 512 * 1024
|
||
|
|
maxSendRecvBytes = 20 * 1024 * 1024 // 20M
|
||
|
|
)
|
||
|
|
|
||
|
|
func convertMessages(rms []*pb.ReceivedMessage) ([]*Message, error) {
|
||
|
|
msgs := make([]*Message, 0, len(rms))
|
||
|
|
for i, m := range rms {
|
||
|
|
msg, err := toMessage(m)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("pubsub: cannot decode the retrieved message at index: %d, message: %+v", i, m)
|
||
|
|
}
|
||
|
|
msgs = append(msgs, msg)
|
||
|
|
}
|
||
|
|
return msgs, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func trunc32(i int64) int32 {
|
||
|
|
if i > math.MaxInt32 {
|
||
|
|
i = math.MaxInt32
|
||
|
|
}
|
||
|
|
return int32(i)
|
||
|
|
}
|
||
|
|
|
||
|
|
type defaultRetryer struct {
|
||
|
|
bo gax.Backoff
|
||
|
|
}
|
||
|
|
|
||
|
|
// Logic originally from
|
||
|
|
// https://github.com/GoogleCloudPlatform/google-cloud-java/blob/master/google-cloud-clients/google-cloud-pubsub/src/main/java/com/google/cloud/pubsub/v1/StatusUtil.java
|
||
|
|
func (r *defaultRetryer) Retry(err error) (pause time.Duration, shouldRetry bool) {
|
||
|
|
s, ok := status.FromError(err)
|
||
|
|
if !ok { // includes io.EOF, normal stream close, which causes us to reopen
|
||
|
|
return r.bo.Pause(), true
|
||
|
|
}
|
||
|
|
switch s.Code() {
|
||
|
|
case codes.DeadlineExceeded, codes.Internal, codes.ResourceExhausted, codes.Aborted:
|
||
|
|
return r.bo.Pause(), true
|
||
|
|
case codes.Unavailable:
|
||
|
|
c := strings.Contains(s.Message(), "Server shutdownNow invoked")
|
||
|
|
if !c {
|
||
|
|
return r.bo.Pause(), true
|
||
|
|
}
|
||
|
|
return 0, false
|
||
|
|
default:
|
||
|
|
return 0, false
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type streamingPullRetryer struct {
|
||
|
|
defaultRetryer gax.Retryer
|
||
|
|
}
|
||
|
|
|
||
|
|
// Does not retry ResourceExhausted. See: https://github.com/GoogleCloudPlatform/google-cloud-go/issues/1166#issuecomment-443744705
|
||
|
|
func (r *streamingPullRetryer) Retry(err error) (pause time.Duration, shouldRetry bool) {
|
||
|
|
s, ok := status.FromError(err)
|
||
|
|
if !ok { // call defaultRetryer so that its backoff can be used
|
||
|
|
return r.defaultRetryer.Retry(err)
|
||
|
|
}
|
||
|
|
switch s.Code() {
|
||
|
|
case codes.ResourceExhausted:
|
||
|
|
return 0, false
|
||
|
|
default:
|
||
|
|
return r.defaultRetryer.Retry(err)
|
||
|
|
}
|
||
|
|
}
|