-
Notifications
You must be signed in to change notification settings - Fork 1.9k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Anshu Agarwal
committed
Dec 6, 2023
1 parent
69cc2a1
commit 783f3c6
Showing
12 changed files
with
211 additions
and
9 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
78 changes: 78 additions & 0 deletions
78
server/src/main/java/org/opensearch/transport/TransportConnectionFailureStats.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,78 @@ | ||
| /* | ||
| * SPDX-License-Identifier: Apache-2.0 | ||
| * | ||
| * The OpenSearch Contributors require contributions made to | ||
| * this file be licensed under the Apache-2.0 license or a | ||
| * compatible open source license. | ||
| */ | ||
|
|
||
| package org.opensearch.transport; | ||
|
|
||
| import org.opensearch.common.util.concurrent.ConcurrentCollections; | ||
| import org.opensearch.core.common.io.stream.StreamInput; | ||
| import org.opensearch.core.common.io.stream.StreamOutput; | ||
| import org.opensearch.core.common.io.stream.Writeable; | ||
| import org.opensearch.core.xcontent.ToXContentFragment; | ||
| import org.opensearch.core.xcontent.XContentBuilder; | ||
|
|
||
| import java.io.IOException; | ||
| import java.util.HashMap; | ||
| import java.util.Map; | ||
| import java.util.Objects; | ||
| import java.util.concurrent.ConcurrentHashMap; | ||
|
|
||
| public class TransportConnectionFailureStats implements ToXContentFragment, Writeable { | ||
| // Node as key with failure count as value | ||
| private Map<String, Integer> connectionFailureMap = ConcurrentCollections.newConcurrentMap(); | ||
|
|
||
| private static final TransportConnectionFailureStats INSTANCE = new TransportConnectionFailureStats(); | ||
|
|
||
| public TransportConnectionFailureStats() { | ||
|
|
||
| } | ||
|
|
||
| public TransportConnectionFailureStats(StreamInput in) throws IOException { | ||
| connectionFailureMap = in.readMap(StreamInput::readString, StreamInput::readInt); | ||
| } | ||
|
|
||
| public static TransportConnectionFailureStats getInstance() { | ||
| return INSTANCE; | ||
| } | ||
|
|
||
| public void updateConnectionFailureCount(String node) { | ||
| connectionFailureMap.compute(node, (k, v) -> v == null ? 1 : v + 1); | ||
| } | ||
|
|
||
| public int getConnectionFailureCount(String node) { | ||
| if (connectionFailureMap.containsKey(node)) { | ||
| Integer value = connectionFailureMap.get(node); | ||
| return value; | ||
| } | ||
| return 0; | ||
| } | ||
|
|
||
| public Map<String, Integer> getConnectionFailure() { | ||
| return new ConcurrentHashMap(connectionFailureMap); | ||
| } | ||
|
|
||
| @Override | ||
| public void writeTo(StreamOutput out) throws IOException { | ||
| Map<String, Integer> map = connectionFailureMap; | ||
| out.writeMap(map, StreamOutput::writeString, StreamOutput::writeInt); | ||
| } | ||
|
|
||
| @Override | ||
| public XContentBuilder toXContent(XContentBuilder builder, Params params) throws IOException { | ||
| builder.startObject("transport_connection_failure"); | ||
| for (var entry : connectionFailureMap.entrySet()) { | ||
| builder.field(entry.getKey(), entry.getValue()); | ||
| } | ||
| builder.endObject(); | ||
| return builder; | ||
| } | ||
|
|
||
| @Override | ||
| public int hashCode() { | ||
| return Objects.hash(connectionFailureMap); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.