iNG-1507: Add handling for guardrail errors 0x36 to 0x38 for XDCR

Author: chvck

gateway/dataimpl/server_v1/errorhandler.go

@@ -1124,6 +1124,39 @@ func (e ErrorHandler) NewVbUuidDivergenceStatus(ctx context.Context, baseErr err
 	return st
 }
 
+func (e ErrorHandler) NewRateLimitBucketResidentRatioTooLowStatus(ctx context.Context, baseErr error, bucketName, scopeName, collectionName, docId string) *status.Status {
+	st := e.newStatus(ctx, codes.ResourceExhausted,
+		fmt.Sprintf("The bucket resident ratio is too low to write '%s' to '%s/%s/%s'.",
+			docId, bucketName, scopeName, collectionName))
+	st = e.tryAttachStatusDetails(st, &epb.ErrorInfo{
+		Reason: "RATE_LIMIT_BUCKET_RESIDENT_RATIO_TOO_LOW",
+	})
+	st = e.tryAttachExtraContext(st, baseErr)
+	return st
+}
+
+func (e ErrorHandler) NewRateLimitBucketDataSizeTooBigStatus(ctx context.Context, baseErr error, bucketName, scopeName, collectionName, docId string) *status.Status {
+	st := e.newStatus(ctx, codes.ResourceExhausted,
+		fmt.Sprintf("The bucket data size is too big to write '%s' to '%s/%s/%s'.",
+			docId, bucketName, scopeName, collectionName))
+	st = e.tryAttachStatusDetails(st, &epb.ErrorInfo{
+		Reason: "RATE_LIMIT_BUCKET_DATA_SIZE_TOO_BIG",
+	})
+	st = e.tryAttachExtraContext(st, baseErr)
+	return st
+}
+
+func (e ErrorHandler) NewRateLimitBucketDiskSpaceTooLowStatus(ctx context.Context, baseErr error, bucketName, scopeName, collectionName, docId string) *status.Status {
+	st := e.newStatus(ctx, codes.ResourceExhausted,
+		fmt.Sprintf("Not enough bucket disk space to write '%s' to '%s/%s/%s'.",
+			docId, bucketName, scopeName, collectionName))
+	st = e.tryAttachStatusDetails(st, &epb.ErrorInfo{
+		Reason: "RATE_LIMIT_BUCKET_DISK_SPACE_TOO_LOW",
+	})
+	st = e.tryAttachExtraContext(st, baseErr)
+	return st
+}
+
 func (e ErrorHandler) NewUnimplementedServerVersionStatus(ctx context.Context) *status.Status {
 	st := e.newStatus(ctx,
 		codes.Unimplemented,

gateway/dataimpl/server_v1/xdcrserver.go

@@ -807,6 +807,12 @@ func (s *XdcrServer) PushDocument(
 				return nil, s.errorHandler.NewValueTooLargeStatus(ctx, err, in.BucketName, in.ScopeName, in.CollectionName, in.Key, false).Err()
 			} else if errors.Is(err, gocbcorex.ErrVbucketUUIDMismatch) {
 				return nil, s.errorHandler.NewVbUuidDivergenceStatus(ctx, err, in.BucketName, in.ScopeName, in.CollectionName, in.Key).Err()
+			} else if errors.Is(err, memdx.ErrRateLimitedBucketResidentRatioTooLow) {
+				return nil, s.errorHandler.NewRateLimitBucketResidentRatioTooLowStatus(ctx, err, in.BucketName, in.ScopeName, in.CollectionName, in.Key).Err()
+			} else if errors.Is(err, memdx.ErrRateLimitedBucketDataSizeTooBig) {
+				return nil, s.errorHandler.NewRateLimitBucketDataSizeTooBigStatus(ctx, err, in.BucketName, in.ScopeName, in.CollectionName, in.Key).Err()
+			} else if errors.Is(err, memdx.ErrRateLimitedBucketDiskSpaceTooLow) {
+				return nil, s.errorHandler.NewRateLimitBucketDiskSpaceTooLowStatus(ctx, err, in.BucketName, in.ScopeName, in.CollectionName, in.Key).Err()
 			}
 			return nil, s.errorHandler.NewGenericStatus(ctx, err).Err()
 		}
@@ -874,6 +880,12 @@ func (s *XdcrServer) PushDocument(
 				return nil, s.errorHandler.NewValueTooLargeStatus(ctx, err, in.BucketName, in.ScopeName, in.CollectionName, in.Key, false).Err()
 			} else if errors.Is(err, gocbcorex.ErrVbucketUUIDMismatch) {
 				return nil, s.errorHandler.NewVbUuidDivergenceStatus(ctx, err, in.BucketName, in.ScopeName, in.CollectionName, in.Key).Err()
+			} else if errors.Is(err, memdx.ErrRateLimitedBucketResidentRatioTooLow) {
+				return nil, s.errorHandler.NewRateLimitBucketResidentRatioTooLowStatus(ctx, err, in.BucketName, in.ScopeName, in.CollectionName, in.Key).Err()
+			} else if errors.Is(err, memdx.ErrRateLimitedBucketDataSizeTooBig) {
+				return nil, s.errorHandler.NewRateLimitBucketDataSizeTooBigStatus(ctx, err, in.BucketName, in.ScopeName, in.CollectionName, in.Key).Err()
+			} else if errors.Is(err, memdx.ErrRateLimitedBucketDiskSpaceTooLow) {
+				return nil, s.errorHandler.NewRateLimitBucketDiskSpaceTooLowStatus(ctx, err, in.BucketName, in.ScopeName, in.CollectionName, in.Key).Err()
 			}
 			return nil, s.errorHandler.NewGenericStatus(ctx, err).Err()
 		}