7979 builder = CosmosClientBuilder .class ,
8080 isAsync = true )
8181public final class CosmosAsyncClient implements Closeable {
82+ private static ImplementationBridgeHelpers .CosmosClientTelemetryConfigHelper .CosmosClientTelemetryConfigAccessor clientTelemetryConfigAccessor () {
83+ return ImplementationBridgeHelpers .CosmosClientTelemetryConfigHelper .getCosmosClientTelemetryConfigAccessor ();
84+ }
85+
86+ private static ImplementationBridgeHelpers .ReadConsistencyStrategyHelper .ReadConsistencyStrategyAccessor readConsistencyStrategyAccessor () {
87+ return ImplementationBridgeHelpers .ReadConsistencyStrategyHelper .getReadConsistencyStrategyAccessor ();
88+ }
89+
90+ private static ImplementationBridgeHelpers .CosmosQueryRequestOptionsHelper .CosmosQueryRequestOptionsAccessor queryOptionsAccessor () {
91+ return ImplementationBridgeHelpers .CosmosQueryRequestOptionsHelper .getCosmosQueryRequestOptionsAccessor ();
92+ }
93+
94+ private static ImplementationBridgeHelpers .FeedResponseHelper .FeedResponseAccessor feedResponseAccessor () {
95+ return ImplementationBridgeHelpers .FeedResponseHelper .getFeedResponseAccessor ();
96+ }
97+
8298 private static final Logger logger = LoggerFactory .getLogger (CosmosAsyncClient .class );
8399
84100 private static final CosmosClientTelemetryConfig DEFAULT_TELEMETRY_CONFIG = new CosmosClientTelemetryConfig ();
85- private static final ImplementationBridgeHelpers .CosmosQueryRequestOptionsHelper .CosmosQueryRequestOptionsAccessor queryOptionsAccessor =
86- ImplementationBridgeHelpers .CosmosQueryRequestOptionsHelper .getCosmosQueryRequestOptionsAccessor ();
87- private static final ImplementationBridgeHelpers .FeedResponseHelper .FeedResponseAccessor feedResponseAccessor =
88- ImplementationBridgeHelpers .FeedResponseHelper .getFeedResponseAccessor ();
89- private static final ImplementationBridgeHelpers .CosmosClientTelemetryConfigHelper .CosmosClientTelemetryConfigAccessor
90- telemetryConfigAccessor = ImplementationBridgeHelpers
91- .CosmosClientTelemetryConfigHelper
92- .getCosmosClientTelemetryConfigAccessor ();
93-
94- private static final ImplementationBridgeHelpers .ReadConsistencyStrategyHelper .ReadConsistencyStrategyAccessor
95- readConsistencyStrategyAccessor = ImplementationBridgeHelpers
96- .ReadConsistencyStrategyHelper
97- .getReadConsistencyStrategyAccessor ();
98101
99102 private final static Function <CosmosAsyncContainer , CosmosAsyncContainer > DEFAULT_CONTAINER_FACTORY =
100103 (originalContainer ) -> originalContainer ;
@@ -148,7 +151,7 @@ public final class CosmosAsyncClient implements Closeable {
148151 this .clientTelemetryConfig = effectiveTelemetryConfig ;
149152 boolean contentResponseOnWriteEnabled = builder .isContentResponseOnWriteEnabled ();
150153 ApiType apiType = builder .apiType ();
151- String clientCorrelationId = telemetryConfigAccessor
154+ String clientCorrelationId = clientTelemetryConfigAccessor ()
152155 .getClientCorrelationId (effectiveTelemetryConfig );
153156
154157 List <Permission > permissionList = new ArrayList <>();
@@ -207,15 +210,14 @@ public final class CosmosAsyncClient implements Closeable {
207210 TagName .ClientCorrelationId .toString (),
208211 ClientTelemetryMetrics .escape (effectiveClientCorrelationId ));
209212
210- this .clientMetricRegistrySnapshot = telemetryConfigAccessor
213+ this .clientMetricRegistrySnapshot = clientTelemetryConfigAccessor ()
211214 .getClientMetricRegistry (effectiveTelemetryConfig );
212215
213- CosmosMeterOptions cpuMeterOptions = telemetryConfigAccessor
216+ CosmosMeterOptions cpuMeterOptions = clientTelemetryConfigAccessor ()
214217 .getMeterOptions (effectiveTelemetryConfig , CosmosMetricName .SYSTEM_CPU );
215- CosmosMeterOptions memoryMeterOptions = telemetryConfigAccessor
218+ CosmosMeterOptions memoryMeterOptions = clientTelemetryConfigAccessor ()
216219 .getMeterOptions (effectiveTelemetryConfig , CosmosMetricName .SYSTEM_MEMORY_FREE );
217220
218-
219221 if (clientMetricRegistrySnapshot != null ) {
220222 ClientTelemetryMetrics .add (clientMetricRegistrySnapshot , cpuMeterOptions , memoryMeterOptions );
221223 }
@@ -224,15 +226,15 @@ public final class CosmosAsyncClient implements Closeable {
224226 );
225227
226228 if (this .clientMetricRegistrySnapshot != null ) {
227- telemetryConfigAccessor .setClientCorrelationTag (
229+ clientTelemetryConfigAccessor () .setClientCorrelationTag (
228230 effectiveTelemetryConfig ,
229231 this .clientCorrelationTag );
230- telemetryConfigAccessor .setAccountName (
232+ clientTelemetryConfigAccessor () .setAccountName (
231233 effectiveTelemetryConfig ,
232234 this .accountTagValue
233235 );
234236
235- telemetryConfigAccessor .addDiagnosticsHandler (
237+ clientTelemetryConfigAccessor () .addDiagnosticsHandler (
236238 effectiveTelemetryConfig ,
237239 new ClientMetricsDiagnosticsHandler (this )
238240 );
@@ -475,7 +477,7 @@ CosmosPagedFlux<CosmosDatabaseProperties> readAllDatabases(CosmosQueryRequestOpt
475477 null ,
476478 ResourceType .Database ,
477479 OperationType .ReadFeed ,
478- queryOptionsAccessor .getQueryNameOrDefault (nonNullOptions , spanName ),
480+ queryOptionsAccessor () .getQueryNameOrDefault (nonNullOptions , spanName ),
479481 nonNullOptions ,
480482 pagedFluxOptions
481483 );
@@ -484,7 +486,7 @@ CosmosPagedFlux<CosmosDatabaseProperties> readAllDatabases(CosmosQueryRequestOpt
484486
485487 return getDocClientWrapper ().readDatabases (state )
486488 .map (response ->
487- feedResponseAccessor .createFeedResponse (
489+ feedResponseAccessor () .createFeedResponse (
488490 ModelBridgeInternal .getCosmosDatabasePropertiesFromV2Results (response .getResults ()),
489491 response .getResponseHeaders (),
490492 response .getCosmosDiagnostics ()));
@@ -504,7 +506,6 @@ public CosmosPagedFlux<CosmosDatabaseProperties> readAllDatabases() {
504506 return readAllDatabases (new CosmosQueryRequestOptions ());
505507 }
506508
507-
508509 /**
509510 * Query for databases.
510511 * <br/>
@@ -662,22 +663,21 @@ private CosmosPagedFlux<CosmosDatabaseProperties> queryDatabasesInternal(
662663 null ,
663664 ResourceType .Database ,
664665 OperationType .Query ,
665- queryOptionsAccessor .getQueryNameOrDefault (nonNullOptions , spanName ),
666+ queryOptionsAccessor () .getQueryNameOrDefault (nonNullOptions , spanName ),
666667 nonNullOptions ,
667668 pagedFluxOptions
668669 );
669670
670671 pagedFluxOptions .setFeedOperationState (state );
671672
672673 return getDocClientWrapper ().queryDatabases (querySpec , state )
673- .map (response -> feedResponseAccessor .createFeedResponse (
674+ .map (response -> feedResponseAccessor () .createFeedResponse (
674675 ModelBridgeInternal .getCosmosDatabasePropertiesFromV2Results (response .getResults ()),
675676 response .getResponseHeaders (),
676677 response .getCosmosDiagnostics ()));
677678 });
678679 }
679680
680-
681681 private Mono <CosmosDatabaseResponse > createDatabaseIfNotExistsInternal (CosmosAsyncDatabase database ,
682682 ThroughputProperties throughputProperties , Context context ) {
683683 String spanName = "createDatabaseIfNotExists." + database .getId ();
@@ -765,7 +765,7 @@ ReadConsistencyStrategy getEffectiveReadConsistencyStrategy(
765765 OperationType operationType ,
766766 ReadConsistencyStrategy desiredReadConsistencyStrategyOfOperation ) {
767767
768- return readConsistencyStrategyAccessor .getEffectiveReadConsistencyStrategy (
768+ return readConsistencyStrategyAccessor () .getEffectiveReadConsistencyStrategy (
769769 resourceType ,
770770 operationType ,
771771 desiredReadConsistencyStrategyOfOperation ,
@@ -780,13 +780,12 @@ CosmosDiagnosticsThresholds getEffectiveDiagnosticsThresholds(
780780 return operationLevelThresholds ;
781781 }
782782
783-
784783 if (this .clientTelemetryConfig == null ) {
785784 return new CosmosDiagnosticsThresholds ();
786785 }
787786
788787 CosmosDiagnosticsThresholds clientLevelThresholds =
789- telemetryConfigAccessor .getDiagnosticsThresholds (this .clientTelemetryConfig );
788+ clientTelemetryConfigAccessor () .getDiagnosticsThresholds (this .clientTelemetryConfig );
790789
791790 return clientLevelThresholds != null ? clientLevelThresholds : new CosmosDiagnosticsThresholds ();
792791 }
@@ -805,15 +804,15 @@ boolean isTransportLevelTracingEnabled() {
805804 this .clientTelemetryConfig
806805 : DEFAULT_TELEMETRY_CONFIG ;
807806
808- if (telemetryConfigAccessor .isLegacyTracingEnabled (effectiveConfig )) {
807+ if (clientTelemetryConfigAccessor () .isLegacyTracingEnabled (effectiveConfig )) {
809808 return false ;
810809 }
811810
812811 if (this .getConnectionPolicy ().getConnectionMode () != ConnectionMode .DIRECT ) {
813812 return false ;
814813 }
815814
816- return telemetryConfigAccessor .isTransportLevelTracingEnabled (effectiveConfig );
815+ return clientTelemetryConfigAccessor () .isTransportLevelTracingEnabled (effectiveConfig );
817816 }
818817
819818 void recordOpenConnectionsAndInitCachesCompleted (List <CosmosContainerIdentity > cosmosContainerIdentities ) {
@@ -859,13 +858,13 @@ public String getAccountTagValue(CosmosAsyncClient client) {
859858
860859 @ Override
861860 public EnumSet <TagName > getMetricTagNames (CosmosAsyncClient client ) {
862- return telemetryConfigAccessor
861+ return clientTelemetryConfigAccessor ()
863862 .getMetricTagNames (client .clientTelemetryConfig );
864863 }
865864
866865 @ Override
867866 public EnumSet <MetricCategory > getMetricCategories (CosmosAsyncClient client ) {
868- return telemetryConfigAccessor
867+ return clientTelemetryConfigAccessor ()
869868 .getMetricCategories (client .clientTelemetryConfig );
870869 }
871870
@@ -896,7 +895,7 @@ public String getUserAgent(CosmosAsyncClient client) {
896895
897896 @ Override
898897 public CosmosMeterOptions getMeterOptions (CosmosAsyncClient client , CosmosMetricName name ) {
899- return telemetryConfigAccessor
898+ return clientTelemetryConfigAccessor ()
900899 .getMeterOptions (client .clientTelemetryConfig , name );
901900 }
902901
0 commit comments