Skip to content

Commit fc01ed8

Browse files
programmatixmikereiche
authored andcommitted
Add private support for setting the analytics endpoint,
to support the Spark Connector Change-Id: I20b161ccfe91784a445b15d1b30ae6b4e5c297f4 Reviewed-on: https://review.couchbase.org/c/couchbase-jvm-clients/+/213653 Tested-by: Build Bot <[email protected]> Reviewed-by: David Nault <[email protected]>
1 parent eeadf25 commit fc01ed8

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

scala-client/src/main/scala/com/couchbase/client/scala/analytics/AnalyticsOptions.scala

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ import scala.concurrent.duration.Duration
3030
* @author Graham Pople
3131
* @since 1.0.0
3232
*/
33-
case class AnalyticsOptions(
33+
case class AnalyticsOptions private (
3434
private[scala] val parameters: Option[AnalyticsParameters] = None,
3535
private[scala] val clientContextId: Option[String] = None,
3636
private[scala] val retryStrategy: Option[RetryStrategy] = None,
@@ -39,7 +39,8 @@ case class AnalyticsOptions(
3939
@SinceCouchbase("6.5") private[scala] val readonly: Option[Boolean] = None,
4040
private[scala] val scanConsistency: Option[AnalyticsScanConsistency] = None,
4141
private[scala] val parentSpan: Option[RequestSpan] = None,
42-
private[scala] val raw: Option[Map[String, Any]] = None
42+
private[scala] val raw: Option[Map[String, Any]] = None,
43+
private[scala] val _endpointIdx: Int = 0
4344
) {
4445

4546
/** Sets the parent `RequestSpan`.
@@ -132,6 +133,9 @@ case class AnalyticsOptions(
132133
def scanConsistency(scanConsistency: AnalyticsScanConsistency): AnalyticsOptions =
133134
copy(scanConsistency = Some(scanConsistency))
134135

136+
private[couchbase] def endpointIdx(idx: Int): AnalyticsOptions =
137+
copy(_endpointIdx = idx)
138+
135139
private[scala] def encode() = {
136140
val out = JsonObject.create
137141

scala-client/src/main/scala/com/couchbase/client/scala/query/handlers/AnalyticsHandler.scala

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,9 @@ private[scala] class AnalyticsHandler(hp: HandlerBasicParams) {
104104
hp.tracer
105105
.requestSpan(TracingIdentifiers.SPAN_REQUEST_ANALYTICS, options.parentSpan.orNull),
106106
bucket.orNull,
107-
scope.orNull
107+
scope.orNull,
108+
true,
109+
options._endpointIdx
108110
)
109111
})
110112
}

0 commit comments

Comments
 (0)