update message validator to provide rejection signal

This commit is contained in:
vyzo 2020-05-14 13:31:45 +03:00
parent e714294273
commit 432c69c616

View File

@ -377,10 +377,12 @@ func (mv *MessageValidator) Validate(ctx context.Context, pid peer.ID, msg *pubs
tag.Insert(metrics.FailureType, "add"), tag.Insert(metrics.FailureType, "add"),
) )
stats.Record(ctx, metrics.MessageValidationFailure.M(1)) stats.Record(ctx, metrics.MessageValidationFailure.M(1))
if xerrors.Is(err, messagepool.ErrBroadcastAnyway) { switch {
return pubsub.ValidationAccept case xerrors.Is(err, messagepool.ErrBroadcastAnyway):
return pubsub.ValidationIgnore
default:
return pubsub.ValidationReject
} }
return pubsub.ValidationIgnore
} }
stats.Record(ctx, metrics.MessageValidationSuccess.M(1)) stats.Record(ctx, metrics.MessageValidationSuccess.M(1))
return pubsub.ValidationAccept return pubsub.ValidationAccept