Skip to content

Commit bd8e118

Browse files
committed
revert formating
1 parent e9e3384 commit bd8e118

File tree

2 files changed

+12
-44
lines changed

2 files changed

+12
-44
lines changed

appender.go

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,6 @@ func New(client elastictransport.Interface, cfg Config) (*Appender, error) {
113113
if err := BulkIndexerConfigFrom(client, cfg).Validate(); err != nil {
114114
return nil, fmt.Errorf("error creating bulk indexer: %w", err)
115115
}
116-
117116
indexer := &Appender{
118117
pool: cfg.BulkIndexerPool,
119118
config: cfg,
@@ -209,11 +208,8 @@ func (a *Appender) Add(ctx context.Context, index string, document io.WriterTo)
209208
Body: document,
210209
}
211210
if len(a.bulkItems) == cap(a.bulkItems) {
212-
a.metrics.blockedAdd.Add(
213-
context.Background(),
214-
1,
215-
metric.WithAttributeSet(a.config.MetricAttributes),
216-
)
211+
attrs := metric.WithAttributeSet(a.config.MetricAttributes)
212+
a.metrics.blockedAdd.Add(context.Background(), 1, attrs)
217213
}
218214

219215
select {

appender_test.go

Lines changed: 10 additions & 38 deletions
Original file line numberDiff line numberDiff line change
@@ -97,8 +97,7 @@ func TestAppender(t *testing.T) {
9797
))
9898

9999
indexerAttrs := attribute.NewSet(
100-
attribute.String("a", "b"),
101-
attribute.String("c", "d"),
100+
attribute.String("a", "b"), attribute.String("c", "d"),
102101
)
103102

104103
indexer, err := docappender.New(client, docappender.Config{
@@ -114,7 +113,7 @@ func TestAppender(t *testing.T) {
114113
for i := 0; i < N; i++ {
115114
addMinimalDoc(t, indexer, "logs-foo-testing")
116115
}
117-
<-time.After(2 * time.Second)
116+
<-time.After(1 * time.Second)
118117

119118
// Appender has not been flushed, there is one active bulk indexer.
120119
// assert.Equal(t, docappender.Stats{Added: N, Active: N, AvailableBulkRequests: 9, IndexersActive: 1}, indexer.Stats())
@@ -228,7 +227,6 @@ func TestAppenderRetry(t *testing.T) {
228227
var bytesTotal int64
229228
var bytesUncompressed int64
230229
var first atomic.Bool
231-
232230
client := docappendertest.NewMockElasticsearchClient(t, func(w http.ResponseWriter, r *http.Request) {
233231
bytesTotal += r.ContentLength
234232
_, result, stat := docappendertest.DecodeBulkRequestWithStats(r)
@@ -265,33 +263,29 @@ func TestAppenderRetry(t *testing.T) {
265263
))
266264

267265
indexerAttrs := attribute.NewSet(
268-
attribute.String("a", "b"),
269-
attribute.String("c", "d"),
266+
attribute.String("a", "b"), attribute.String("c", "d"),
270267
)
271268

272269
indexer, err := docappender.New(client, docappender.Config{
273-
FlushInterval: 2 * time.Minute,
270+
FlushInterval: time.Minute,
274271
FlushBytes: 800, // this is enough to flush after 9 documents
275272
MaxRequests: 1, // to ensure the test is stable
276273
MaxDocumentRetries: 1, // to test the document retry logic
277-
MeterProvider: sdkmetric.NewMeterProvider(
278-
sdkmetric.WithReader(rdr),
279-
),
280-
MetricAttributes: indexerAttrs,
274+
MeterProvider: sdkmetric.NewMeterProvider(sdkmetric.WithReader(rdr)),
275+
MetricAttributes: indexerAttrs,
281276
})
282277

283278
require.NoError(t, err)
284-
// defer indexer.Close(context.Background())
279+
defer indexer.Close(context.Background())
285280

286281
const N = 10
287282
for i := 0; i < N; i++ {
288283
addMinimalDoc(t, indexer, "logs-foo-testing")
289284
}
290-
<-time.After(20 * time.Second)
285+
<-time.After(1 * time.Second)
291286

292287
var rm metricdata.ResourceMetrics
293288
assert.NoError(t, rdr.Collect(context.Background(), &rm))
294-
<-time.After(20 * time.Second)
295289

296290
var asserted atomic.Int64
297291
assertCounter := docappendertest.NewAssertCounter(t, &asserted)
@@ -454,7 +448,6 @@ func TestAppenderAvailableAppenders(t *testing.T) {
454448
for i := 0; i < N; i++ {
455449
addMinimalDoc(t, indexer, "logs-foo-testing")
456450
}
457-
458451
timeout := time.NewTimer(2 * time.Second)
459452
defer timeout.Stop()
460453
for i := 0; i < N; i++ {
@@ -878,7 +871,6 @@ func TestAppenderFlushRequestError(t *testing.T) {
878871

879872
// Closing the indexer flushes enqueued documents.
880873
err = indexer.Close(context.Background())
881-
882874
switch includeSource {
883875
case docappender.False, docappender.True:
884876
// include_source=false is implemented in ES so we just assert we're not
@@ -1774,7 +1766,6 @@ func TestAppenderScaling(t *testing.T) {
17741766
t.Cleanup(func() { indexer.Close(context.Background()) })
17751767
return indexer
17761768
}
1777-
17781769
sendDocuments := func(t *testing.T, indexer *docappender.Appender, docs int) {
17791770
for i := 0; i < docs; i++ {
17801771
err := indexer.Add(context.Background(), "logs-foo-testing", newJSONReader(map[string]any{
@@ -1866,7 +1857,6 @@ func TestAppenderScaling(t *testing.T) {
18661857
}
18671858
}
18681859
}
1869-
18701860
t.Run("DownscaleIdle", func(t *testing.T) {
18711861
rdr := sdkmetric.NewManualReader(sdkmetric.WithTemporalitySelector(
18721862
func(ik sdkmetric.InstrumentKind) metricdata.Temporality {
@@ -1893,7 +1883,6 @@ func TestAppenderScaling(t *testing.T) {
18931883
})
18941884
docs := int64(20)
18951885
sendDocuments(t, indexer, int(docs))
1896-
18971886
waitForScaleUp(t, indexer, 3)
18981887
waitForScaleDown(t, indexer, rdr, 1)
18991888

@@ -1954,7 +1943,6 @@ func TestAppenderScaling(t *testing.T) {
19541943
}
19551944
})
19561945
})
1957-
19581946
t.Run("DownscaleActiveLimit", func(t *testing.T) {
19591947
rdr := sdkmetric.NewManualReader(sdkmetric.WithTemporalitySelector(
19601948
func(ik sdkmetric.InstrumentKind) metricdata.Temporality {
@@ -1982,11 +1970,9 @@ func TestAppenderScaling(t *testing.T) {
19821970
})
19831971
docs := int64(14)
19841972
sendDocuments(t, indexer, int(docs))
1985-
19861973
waitForScaleUp(t, indexer, 3)
19871974
// Set the gomaxprocs to 4, which should result in an activeLimit of 1.
19881975
setGOMAXPROCS(t, 4)
1989-
19901976
// Wait for the indexers to scale down from 3 to 1. The downscale cool
19911977
// down of `1m` isn't respected, since the active limit is breached with
19921978
// the gomaxprocs change.
@@ -2051,7 +2037,6 @@ func TestAppenderScaling(t *testing.T) {
20512037
}
20522038
})
20532039
})
2054-
20552040
t.Run("UpscaleCooldown", func(t *testing.T) {
20562041
rdr := sdkmetric.NewManualReader(sdkmetric.WithTemporalitySelector(
20572042
func(ik sdkmetric.InstrumentKind) metricdata.Temporality {
@@ -2079,12 +2064,9 @@ func TestAppenderScaling(t *testing.T) {
20792064
})
20802065
docs := int64(50)
20812066
sendDocuments(t, indexer, int(docs))
2082-
20832067
waitForScaleUp(t, indexer, 2)
20842068
waitForBulkRequests(t, indexer, rdr, docs)
2085-
20862069
assert.Equal(t, int64(2), indexer.IndexersActive())
2087-
20882070
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
20892071
defer cancel()
20902072
assert.NoError(t, indexer.Close(ctx))
@@ -2144,7 +2126,6 @@ func TestAppenderScaling(t *testing.T) {
21442126
}
21452127
})
21462128
})
2147-
21482129
t.Run("Downscale429Rate", func(t *testing.T) {
21492130
rdr := sdkmetric.NewManualReader(sdkmetric.WithTemporalitySelector(
21502131
func(ik sdkmetric.InstrumentKind) metricdata.Temporality {
@@ -2177,23 +2158,19 @@ func TestAppenderScaling(t *testing.T) {
21772158
FlushBytes: 1,
21782159
Scaling: docappender.ScalingConfig{
21792160
ScaleUp: docappender.ScaleActionConfig{
2180-
Threshold: 5,
2181-
CoolDown: 1,
2161+
Threshold: 5, CoolDown: 1,
21822162
},
21832163
ScaleDown: docappender.ScaleActionConfig{
2184-
Threshold: 100,
2185-
CoolDown: 100 * time.Millisecond,
2164+
Threshold: 100, CoolDown: 100 * time.Millisecond,
21862165
},
21872166
IdleInterval: 100 * time.Millisecond,
21882167
},
21892168
MeterProvider: sdkmetric.NewMeterProvider(sdkmetric.WithReader(rdr)),
21902169
})
21912170
require.NoError(t, err)
21922171
t.Cleanup(func() { indexer.Close(context.Background()) })
2193-
21942172
docs := int64(20)
21952173
sendDocuments(t, indexer, int(docs))
2196-
21972174
waitForScaleUp(t, indexer, 3)
21982175
waitForBulkRequests(t, indexer, rdr, docs)
21992176

@@ -2204,7 +2181,6 @@ func TestAppenderScaling(t *testing.T) {
22042181
mu.Unlock()
22052182
docs += 5
22062183
sendDocuments(t, indexer, 5)
2207-
22082184
waitForScaleDown(t, indexer, rdr, 1)
22092185
waitForBulkRequests(t, indexer, rdr, docs)
22102186

@@ -2215,15 +2191,11 @@ func TestAppenderScaling(t *testing.T) {
22152191
mu.Unlock()
22162192
docs += 600
22172193
sendDocuments(t, indexer, 600)
2218-
22192194
waitForScaleUp(t, indexer, 3)
22202195
waitForBulkRequests(t, indexer, rdr, docs)
2221-
22222196
assert.Equal(t, int64(3), indexer.IndexersActive())
2223-
22242197
var rm metricdata.ResourceMetrics
22252198
assert.NoError(t, rdr.Collect(context.Background(), &rm))
2226-
22272199
docappendertest.AssertOTelMetrics(t, rm.ScopeMetrics[0].Metrics, func(m metricdata.Metrics) {
22282200
switch n := m.Name; n {
22292201
case "elasticsearch.indexer.created":

0 commit comments

Comments
 (0)