mirror of
https://github.com/status-im/consul.git
synced 2025-01-10 05:45:46 +00:00
5fb9df1640
* Adding explicit MPL license for sub-package This directory and its subdirectories (packages) contain files licensed with the MPLv2 `LICENSE` file in this directory and are intentionally licensed separately from the BSL `LICENSE` file at the root of this repository. * Adding explicit MPL license for sub-package This directory and its subdirectories (packages) contain files licensed with the MPLv2 `LICENSE` file in this directory and are intentionally licensed separately from the BSL `LICENSE` file at the root of this repository. * Updating the license from MPL to Business Source License Going forward, this project will be licensed under the Business Source License v1.1. Please see our blog post for more details at <Blog URL>, FAQ at www.hashicorp.com/licensing-faq, and details of the license at www.hashicorp.com/bsl. * add missing license headers * Update copyright file headers to BUSL-1.1 * Update copyright file headers to BUSL-1.1 * Update copyright file headers to BUSL-1.1 * Update copyright file headers to BUSL-1.1 * Update copyright file headers to BUSL-1.1 * Update copyright file headers to BUSL-1.1 * Update copyright file headers to BUSL-1.1 * Update copyright file headers to BUSL-1.1 * Update copyright file headers to BUSL-1.1 * Update copyright file headers to BUSL-1.1 * Update copyright file headers to BUSL-1.1 * Update copyright file headers to BUSL-1.1 * Update copyright file headers to BUSL-1.1 * Update copyright file headers to BUSL-1.1 * Update copyright file headers to BUSL-1.1 --------- Co-authored-by: hashicorp-copywrite[bot] <110428419+hashicorp-copywrite[bot]@users.noreply.github.com>
101 lines
2.6 KiB
Go
101 lines
2.6 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package subscribe
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/hashicorp/go-uuid"
|
|
|
|
"github.com/hashicorp/consul/agent/consul/stream"
|
|
"github.com/hashicorp/consul/proto/private/pbsubscribe"
|
|
)
|
|
|
|
// streamID is used in logs as a unique identifier for a subscription. The value
|
|
// is created lazily on the first call to String() so that we do not call it
|
|
// if trace logging is disabled.
|
|
// If a random UUID can not be created, defaults to the current time formatted
|
|
// as RFC3339Nano.
|
|
//
|
|
// TODO(banks) it might be nice one day to replace this with OpenTracing ID
|
|
// if one is set etc. but probably pointless until we support that properly
|
|
// in other places so it's actually propagated properly. For now this just
|
|
// makes lifetime of a stream more traceable in our regular server logs for
|
|
// debugging/dev.
|
|
type streamID struct {
|
|
once sync.Once
|
|
id string
|
|
}
|
|
|
|
func (s *streamID) String() string {
|
|
s.once.Do(func() {
|
|
var err error
|
|
s.id, err = uuid.GenerateUUID()
|
|
if err != nil {
|
|
s.id = time.Now().Format(time.RFC3339Nano)
|
|
}
|
|
})
|
|
return s.id
|
|
}
|
|
|
|
func newLoggerForRequest(l Logger, req *pbsubscribe.SubscribeRequest) Logger {
|
|
l = l.With(
|
|
"topic", req.Topic.String(),
|
|
"dc", req.Datacenter,
|
|
"request_index", req.Index,
|
|
"stream_id", &streamID{},
|
|
)
|
|
|
|
if req.GetWildcardSubject() {
|
|
return l.With("wildcard_subject", true)
|
|
}
|
|
|
|
named := req.GetNamedSubject()
|
|
if named == nil {
|
|
named = &pbsubscribe.NamedSubject{
|
|
Key: req.Key, // nolint:staticcheck // SA1019 intentional use of deprecated field
|
|
Partition: req.Partition, // nolint:staticcheck // SA1019 intentional use of deprecated field
|
|
Namespace: req.Namespace, // nolint:staticcheck // SA1019 intentional use of deprecated field
|
|
PeerName: req.PeerName, // nolint:staticcheck // SA1019 intentional use of deprecated field
|
|
}
|
|
}
|
|
|
|
return l.With(
|
|
"peer", named.PeerName,
|
|
"key", named.Key,
|
|
"namespace", named.Namespace,
|
|
"partition", named.Partition,
|
|
)
|
|
}
|
|
|
|
type eventLogger struct {
|
|
logger Logger
|
|
snapshotDone bool
|
|
count uint64
|
|
}
|
|
|
|
func (l *eventLogger) Trace(e stream.Event) {
|
|
switch {
|
|
case e.IsEndOfSnapshot():
|
|
l.snapshotDone = true
|
|
l.logger.Trace("snapshot complete", "index", e.Index, "sent", l.count)
|
|
return
|
|
case e.IsNewSnapshotToFollow():
|
|
l.logger.Trace("starting new snapshot", "sent", l.count)
|
|
return
|
|
}
|
|
|
|
size := 1
|
|
if l, ok := e.Payload.(length); ok {
|
|
size = l.Len()
|
|
}
|
|
l.logger.Trace("sending events", "index", e.Index, "sent", l.count, "batch_size", size)
|
|
l.count += uint64(size)
|
|
}
|
|
|
|
type length interface {
|
|
Len() int
|
|
}
|