Repository: ignite Updated Branches: refs/heads/ignite-zk 276152a99 -> 186e66127
zk Project: http://git-wip-us.apache.org/repos/asf/ignite/repo Commit: http://git-wip-us.apache.org/repos/asf/ignite/commit/186e6612 Tree: http://git-wip-us.apache.org/repos/asf/ignite/tree/186e6612 Diff: http://git-wip-us.apache.org/repos/asf/ignite/diff/186e6612 Branch: refs/heads/ignite-zk Commit: 186e66127ec5764ba1f819cb4b0db13ff3c6ed32 Parents: 276152a Author: sboikov <[email protected]> Authored: Fri Dec 29 14:33:22 2017 +0300 Committer: sboikov <[email protected]> Committed: Fri Dec 29 14:33:22 2017 +0300 ---------------------------------------------------------------------- .../src/main/scala/org/apache/ignite/spark/IgniteRDD.scala | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/ignite/blob/186e6612/modules/spark/src/main/scala/org/apache/ignite/spark/IgniteRDD.scala ---------------------------------------------------------------------- diff --git a/modules/spark/src/main/scala/org/apache/ignite/spark/IgniteRDD.scala b/modules/spark/src/main/scala/org/apache/ignite/spark/IgniteRDD.scala index 78e2223..8a69acc 100644 --- a/modules/spark/src/main/scala/org/apache/ignite/spark/IgniteRDD.scala +++ b/modules/spark/src/main/scala/org/apache/ignite/spark/IgniteRDD.scala @@ -25,6 +25,7 @@ import org.apache.ignite.internal.processors.cache.query.QueryCursorEx import org.apache.ignite.internal.processors.query.GridQueryFieldMetadata import org.apache.ignite.lang.IgniteUuid import org.apache.ignite.spark.impl._ +import org.apache.ignite.spi.discovery.tcp.TcpDiscoverySpi import org.apache.ignite.spi.discovery.tcp.internal.TcpDiscoveryNode import org.apache.spark.rdd.RDD import org.apache.spark.sql.types._ @@ -91,8 +92,14 @@ class IgniteRDD[K, V] ( override protected[spark] def getPreferredLocations(split: Partition): Seq[String] = { ensureCache() - ic.ignite().affinity(cacheName).mapPartitionToPrimaryAndBackups(split.index) + if (ic.ignite().configuration().getDiscoverySpi().isInstanceOf[TcpDiscoverySpi]) { + ic.ignite().affinity(cacheName).mapPartitionToPrimaryAndBackups(split.index) .map(_.asInstanceOf[TcpDiscoveryNode].socketAddresses()).flatten.map(_.getHostName).toList + } + else { + ic.ignite().affinity(cacheName).mapPartitionToPrimaryAndBackups(split.index) + .flatten(_.hostNames).toSeq + } } /**
