Skip to content

Commit e42e853

Browse files
author
Bhumika Sharma
committed
Fix: updates Counter to AtomicInteger
Signed-off-by: Bhumika Sharma <[email protected]>
1 parent 5cd283f commit e42e853

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

server/src/test/java/org/opensearch/telemetry/TestInMemoryCounter.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313

1414
import java.util.HashMap;
1515
import java.util.concurrent.ConcurrentHashMap;
16-
import java.util.concurrent.atomic.AtomicReference;
16+
import java.util.concurrent.atomic.AtomicInteger;
1717

1818
/**
1919
* This is a simple implementation of Counter which is utilized by TestInMemoryMetricsRegistry for
@@ -23,11 +23,11 @@
2323
*/
2424
public class TestInMemoryCounter implements Counter {
2525

26-
private AtomicReference<Double> counterValue = new AtomicReference<Double>(0.0);
26+
private AtomicInteger counterValue = new AtomicInteger(0);
2727
private ConcurrentHashMap<HashMap<String, ?>, Double> counterValueForTags = new ConcurrentHashMap<>();
2828

2929
public Double getCounterValue() {
30-
return this.counterValue.get();
30+
return (double) this.counterValue.get();
3131
}
3232

3333
public ConcurrentHashMap<HashMap<String, ?>, Double> getCounterValueForTags() {
@@ -37,7 +37,7 @@ public Double getCounterValue() {
3737
@Override
3838
public void add(double value) {
3939
double oldValue = counterValue.get();
40-
counterValue.set(oldValue + value);
40+
counterValue.set((int) (oldValue + value));
4141
}
4242

4343
@Override

0 commit comments

Comments
 (0)