raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 10

External sharding client handle empty DData key (#28668)

Browse source
This commit is contained in:
Christopher Batey 2020-03-04 16:16:06 +00:00 • committed by GitHub
parent e487088a27
commit f1dbb79b71
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
akka-cluster-sharding/src/main/scala/akka/cluster/sharding/external/internal/ExternalShardAllocationClientImpl.scala vendored
View file

@ -17,6 +17,7 @@ import akka.cluster.ddata.LWWMap
import akka.cluster.ddata.Replicator.Get
import akka.cluster.ddata.Replicator.GetFailure
import akka.cluster.ddata.Replicator.GetSuccess
import akka.cluster.ddata.Replicator.NotFound
import akka.cluster.ddata.Replicator.ReadMajority
import akka.cluster.ddata.Replicator.Update
import akka.cluster.ddata.Replicator.UpdateSuccess
@ -79,6 +80,8 @@ final private[external] class ExternalShardAllocationClientImpl(system: ActorSys
case success @ GetSuccess(`Key`, _) =>
Future.successful(
success.get(Key).entries.transform((_, asStr) => ShardLocation(AddressFromURIString(asStr))))
case NotFound(_, _) =>
Future.successful(Map.empty[ShardId, ShardLocation])
case GetFailure(_, _) =>
Future.failed((new ClientTimeoutException(s"Unable to get shard locations after ${timeout.duration.pretty}")))
}

10
akka-cluster-sharding/src/test/scala/akka/cluster/sharding/external/ExternalShardAllocationStrategySpec.scala vendored
View file

@ -20,6 +20,16 @@ class ExternalShardAllocationStrategySpec extends AkkaSpec("""
val requester = TestProbe()
"ExternalShardAllocationClient" must {
"default to no locations if sharding never started" in {
ExternalShardAllocation(system)
.clientFor("not found")
.shardLocations()
.futureValue
.locations shouldEqual Map.empty
}
}
"ExternalShardAllocation allocate" must {
"default to requester if query times out" in {
val (strat, _) = createStrategy()