Read the clock fewer times during message routing (#408)
Previously, we read the system clock twice for each event unless we
switched partitions early due to reaching `maxBatchingMessages` or
`maxBatchingSize`.
Now, we read the clock once every `maxBatchingMessages / 100` messages.
This improves performance (especially for larger batch sizes) as the
router function is called for every message produced.
A bench test of the default router was added; results are below:
```
name old time/op new time/op delta
DefaultRouter 106ns ± 0% 16ns ± 1% -84.94% (p=0.016 n=4+5)
DefaultRouter-4 106ns ± 0% 16ns ± 0% ~ (p=0.079 n=4+5)
```
Signed-off-by: Daniel Ferstay <dferstay@splunk.com>
Co-authored-by: Daniel Ferstay <dferstay@splunk.com>
diff --git a/pulsar/default_router.go b/pulsar/default_router.go
index 82be982..0e1a354 100644
--- a/pulsar/default_router.go
+++ b/pulsar/default_router.go
@@ -48,6 +48,7 @@
lastChangeTimestamp: math.MinInt64,
}
+ readClockAfterNumMessages := uint32(maxBatchingMessages / 10)
return func(message *ProducerMessage, numPartitions uint32) int {
if numPartitions == 1 {
// When there are no partitions, don't even bother
@@ -72,23 +73,34 @@
// Note that it is possible that we skip more than one partition if multiple goroutines increment
// currentPartitionCursor at the same time. If that happens it shouldn't be a problem because we only want to
// spread the data on different partitions but not necessarily in a specific sequence.
+ var now int64
size := uint32(len(message.Payload))
previousMessageCount := atomic.LoadUint32(&state.msgCounter)
previousBatchingMaxSize := atomic.LoadUint32(&state.cumulativeBatchSize)
previousLastChange := atomic.LoadInt64(&state.lastChangeTimestamp)
- if (previousMessageCount >= uint32(maxBatchingMessages-1)) ||
- (size >= uint32(maxBatchingSize)-previousBatchingMaxSize) ||
- (time.Now().UnixNano()-previousLastChange >= maxBatchingDelay.Nanoseconds()) {
+
+ messageCountReached := previousMessageCount >= uint32(maxBatchingMessages-1)
+ sizeReached := (size >= uint32(maxBatchingSize)-previousBatchingMaxSize)
+ durationReached := false
+ if readClockAfterNumMessages == 0 || previousMessageCount%readClockAfterNumMessages == 0 {
+ now = time.Now().UnixNano()
+ durationReached = now-previousLastChange >= maxBatchingDelay.Nanoseconds()
+ }
+ if messageCountReached || sizeReached || durationReached {
atomic.AddUint32(&state.currentPartitionCursor, 1)
- atomic.StoreInt64(&state.lastChangeTimestamp, time.Now().UnixNano())
- atomic.StoreUint32(&state.cumulativeBatchSize, 0)
atomic.StoreUint32(&state.msgCounter, 0)
+ atomic.StoreUint32(&state.cumulativeBatchSize, 0)
+ if now != 0 {
+ atomic.StoreInt64(&state.lastChangeTimestamp, now)
+ }
return int(state.currentPartitionCursor % numPartitions)
}
- atomic.StoreInt64(&state.lastChangeTimestamp, time.Now().UnixNano())
atomic.AddUint32(&state.msgCounter, 1)
atomic.AddUint32(&state.cumulativeBatchSize, size)
+ if now != 0 {
+ atomic.StoreInt64(&state.lastChangeTimestamp, now)
+ }
return int(state.currentPartitionCursor % numPartitions)
}
}
diff --git a/pulsar/default_router_bench_test.go b/pulsar/default_router_bench_test.go
new file mode 100644
index 0000000..d7ec175
--- /dev/null
+++ b/pulsar/default_router_bench_test.go
@@ -0,0 +1,45 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements. See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership. The ASF licenses this file
+// to you 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 pulsar
+
+import (
+ "testing"
+ "time"
+
+ "github.com/apache/pulsar-client-go/pulsar/internal"
+)
+
+var (
+ targetPartition int
+)
+
+func BenchmarkDefaultRouter(b *testing.B) {
+ const (
+ numPartitions = uint32(200)
+ maxBatchingMessages = 2000
+ maxBatchingSize = 524288
+ maxBatchingDelay = 100 * time.Millisecond
+ )
+ msg := &ProducerMessage{
+ Payload: []byte("message 1"),
+ }
+ router := NewDefaultRouter(internal.JavaStringHash, maxBatchingMessages, maxBatchingSize, maxBatchingDelay, false)
+ for i := 0; i < b.N; i++ {
+ targetPartition = router(msg, numPartitions)
+ }
+}
diff --git a/pulsar/default_router_test.go b/pulsar/default_router_test.go
index 60e10c7..31b27af 100644
--- a/pulsar/default_router_test.go
+++ b/pulsar/default_router_test.go
@@ -47,7 +47,7 @@
func TestDefaultRouterRoutingBecauseMaxPublishDelayReached(t *testing.T) {
maxPublishDelay := time.Nanosecond * 10
- router := NewDefaultRouter(internal.JavaStringHash, 20, 100, maxPublishDelay, false)
+ router := NewDefaultRouter(internal.JavaStringHash, 10, 100, maxPublishDelay, false)
const numPartitions = uint32(3)
p1 := router(&ProducerMessage{
Payload: []byte("message 1"),