-
Notifications
You must be signed in to change notification settings - Fork 12
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
First form, thread per task iterator
- Loading branch information
1 parent
07a0fb8
commit f160699
Showing
3 changed files
with
267 additions
and
0 deletions.
There are no files selected for viewing
22 changes: 22 additions & 0 deletions
22
memberships/src/main/java/com/salesforce/apollo/ring/ResultConsumer.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,22 @@ | ||
/* | ||
* Copyright (c) 2021, salesforce.com, inc. | ||
* All rights reserved. | ||
* SPDX-License-Identifier: BSD-3-Clause | ||
* For full license text, see the LICENSE file in the repo root or https://opensource.org/licenses/BSD-3-Clause | ||
*/ | ||
package com.salesforce.apollo.ring; | ||
|
||
import com.google.common.util.concurrent.ListenableFuture; | ||
import com.salesforce.apollo.ring.RingCommunications.Destination; | ||
|
||
import java.util.Optional; | ||
import java.util.concurrent.atomic.AtomicInteger; | ||
|
||
/** | ||
* @author hal.hildebrand | ||
* | ||
*/ | ||
@FunctionalInterface | ||
public interface ResultConsumer<M, T, Comm> { | ||
boolean handle(AtomicInteger tally, Optional<T> result, Destination<M, Comm> destination); | ||
} |
214 changes: 214 additions & 0 deletions
214
memberships/src/main/java/com/salesforce/apollo/ring/RingIteratorSync.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,214 @@ | ||
/* | ||
* Copyright (c) 2021, salesforce.com, inc. | ||
* All rights reserved. | ||
* SPDX-License-Identifier: BSD-3-Clause | ||
* For full license text, see the LICENSE file in the repo root or https://opensource.org/licenses/BSD-3-Clause | ||
*/ | ||
package com.salesforce.apollo.ring; | ||
|
||
import com.salesforce.apollo.archipelago.Link; | ||
import com.salesforce.apollo.archipelago.RouterImpl.CommonCommunications; | ||
import com.salesforce.apollo.crypto.Digest; | ||
import com.salesforce.apollo.membership.Context; | ||
import com.salesforce.apollo.membership.Member; | ||
import com.salesforce.apollo.membership.SigningMember; | ||
import com.salesforce.apollo.utils.Utils; | ||
import org.slf4j.Logger; | ||
import org.slf4j.LoggerFactory; | ||
|
||
import java.io.IOException; | ||
import java.time.Duration; | ||
import java.util.Optional; | ||
import java.util.Set; | ||
import java.util.concurrent.ConcurrentSkipListSet; | ||
import java.util.concurrent.Executor; | ||
import java.util.concurrent.ScheduledExecutorService; | ||
import java.util.concurrent.TimeUnit; | ||
import java.util.concurrent.atomic.AtomicInteger; | ||
import java.util.function.BiFunction; | ||
import java.util.function.Consumer; | ||
|
||
/** | ||
* @author hal.hildebrand | ||
*/ | ||
public class RingIteratorSync<T extends Member, Comm extends Link> extends RingCommunications<T, Comm> { | ||
private static final Logger log = LoggerFactory.getLogger(RingIteratorSync.class); | ||
|
||
private final Duration frequency; | ||
private final ScheduledExecutorService scheduler; | ||
private volatile boolean majorityFailed = false; | ||
private volatile boolean majoritySucceed = false; | ||
|
||
public RingIteratorSync(Duration frequency, Context<T> context, SigningMember member, | ||
CommonCommunications<Comm, ?> comm, Executor exec, boolean ignoreSelf, | ||
ScheduledExecutorService scheduler) { | ||
super(context, member, comm, exec, ignoreSelf); | ||
this.scheduler = scheduler; | ||
this.frequency = frequency; | ||
} | ||
|
||
public RingIteratorSync(Duration frequency, Context<T> context, SigningMember member, | ||
ScheduledExecutorService scheduler, CommonCommunications<Comm, ?> comm, Executor exec) { | ||
this(frequency, context, member, comm, exec, false, scheduler); | ||
} | ||
|
||
public RingIteratorSync(Duration frequency, Direction direction, Context<T> context, SigningMember member, | ||
CommonCommunications<Comm, ?> comm, Executor exec, boolean ignoreSelf, | ||
ScheduledExecutorService scheduler) { | ||
super(direction, context, member, comm, exec, ignoreSelf); | ||
this.scheduler = scheduler; | ||
this.frequency = frequency; | ||
} | ||
|
||
public RingIteratorSync(Duration frequency, Direction direction, Context<T> context, SigningMember member, | ||
ScheduledExecutorService scheduler, CommonCommunications<Comm, ?> comm, Executor exec) { | ||
this(frequency, direction, context, member, comm, exec, false, scheduler); | ||
} | ||
|
||
public <Q> void iterate(Digest digest, BiFunction<Comm, Integer, Q> round, | ||
ResultConsumer<T, Q, Comm> handler) { | ||
iterate(digest, null, round, null, handler, null); | ||
} | ||
|
||
public <Q> void iterate(Digest digest, BiFunction<Comm, Integer, Q> round, | ||
ResultConsumer<T, Q, Comm> handler, Consumer<Integer> onComplete) { | ||
iterate(digest, null, round, null, handler, onComplete); | ||
} | ||
|
||
public <Q> void iterate(Digest digest, Runnable onMajority, BiFunction<Comm, Integer, Q> round, | ||
Runnable failedMajority, ResultConsumer<T, Q, Comm> handler, | ||
Consumer<Integer> onComplete) { | ||
AtomicInteger tally = new AtomicInteger(0); | ||
var traversed = new ConcurrentSkipListSet<Member>(); | ||
exec.execute(() -> internalIterate(digest, onMajority, round, failedMajority, handler, onComplete, tally, | ||
traversed)); | ||
|
||
} | ||
|
||
public int iteration() { | ||
return currentIndex + 1; | ||
} | ||
|
||
@Override | ||
public RingIteratorSync<T, Comm> noDuplicates() { | ||
super.noDuplicates(); | ||
return this; | ||
} | ||
|
||
@Override | ||
protected Logger getLog() { | ||
return log; | ||
} | ||
|
||
private <Q> void internalIterate(Digest digest, Runnable onMajority, | ||
BiFunction<Comm, Integer, Q> round, Runnable failedMajority, | ||
ResultConsumer<T, Q, Comm> handler, Consumer<Integer> onComplete, | ||
AtomicInteger tally, Set<Member> traversed) { | ||
|
||
Runnable proceed = () -> internalIterate(digest, onMajority, round, failedMajority, handler, onComplete, tally, | ||
traversed); | ||
boolean completed = currentIndex == context.getRingCount() - 1; | ||
|
||
Consumer<Boolean> allowed = allow -> proceed(digest, allow, onMajority, failedMajority, tally, completed, | ||
onComplete); | ||
if (completed) { | ||
allowed.accept(true); | ||
return; | ||
} | ||
|
||
var next = next(digest); | ||
log.trace("Iteration: {} tally: {} for: {} on: {} ring: {} complete: false on: {}", iteration(), tally.get(), | ||
digest, context.getId(), next.ring(), member.getId()); | ||
if (next.link() == null) { | ||
log.trace("No successor found of: {} on: {} iteration: {} traversed: {} ring: {} on: {}", digest, | ||
context.getId(), iteration(), traversed, context.ring(currentIndex).stream().toList(), | ||
member.getId()); | ||
final boolean allow = handler.handle(tally, Optional.empty(), next); | ||
allowed.accept(allow); | ||
if (allow) { | ||
log.trace("Finished on iteration: {} proceeding on: {} for: {} tally: {} on: {}", iteration(), digest, | ||
context.getId(), tally.get(), member.getId()); | ||
schedule(proceed); | ||
} else { | ||
log.trace("Completed on iteration: {} on: {} for: {} for: {} tally: {} on: {}", iteration(), digest, | ||
context.getId(), tally.get(), member.getId()); | ||
} | ||
return; | ||
} | ||
try (Comm link = next.link()) { | ||
log.trace("Continuation on iteration: {} tally: {} for: {} on: {} ring: {} to: {} on: {}", iteration(), | ||
tally.get(), digest, context.getId(), next.ring(), | ||
link.getMember() == null ? null : link.getMember().getId(), member.getId()); | ||
var result = round.apply(link, next.ring()); | ||
if (result == null) { | ||
log.trace("No asynchronous response for: {} on: {} iteration: {} from: {} on: {}", digest, | ||
context.getId(), iteration(), link.getMember() == null ? null : link.getMember().getId(), | ||
member.getId()); | ||
final boolean allow = handler.handle(tally, Optional.empty(), next); | ||
allowed.accept(allow); | ||
if (allow) { | ||
log.trace("Proceeding on iteration: {} on: {} for: {} tally: {} on: {}", iteration(), digest, | ||
context.getId(), tally.get(), member.getId()); | ||
schedule(proceed); | ||
} else { | ||
log.trace("Completed on iteration: {} on: {} for: {} tally: {} on: {}", iteration(), digest, | ||
context.getId(), tally.get(), member.getId()); | ||
} | ||
return; | ||
} | ||
final var allow = handler.handle(tally, Optional.of(result), next); | ||
allowed.accept(allow); | ||
if (allow) { | ||
log.trace("Scheduling next iteration: {} on: {} for: {} tally: {} on: {}", iteration(), digest, | ||
context.getId(), tally.get(), member.getId()); | ||
schedule(proceed); | ||
} else { | ||
log.trace("Finished on iteration: {} on: {} for: {} tally: {} on: {}", iteration(), digest, | ||
context.getId(), tally.get(), member.getId()); | ||
} | ||
} catch (IOException e) { | ||
log.debug("Error closing", e); | ||
} | ||
} | ||
|
||
private void proceed(Digest key, final boolean allow, Runnable onMajority, Runnable failedMajority, | ||
AtomicInteger tally, boolean finalIteration, Consumer<Integer> onComplete) { | ||
final var current = currentIndex; | ||
if (!finalIteration) { | ||
log.trace("Determining: {} continuation of: {} for: {} tally: {} majority: {} final itr: {} allow: {} on: {}", | ||
current, key, context.getId(), tally.get(), context.majority(), finalIteration, allow, | ||
member.getId()); | ||
} | ||
if (finalIteration && allow) { | ||
log.trace("Completing iteration: {} of: {} for: {} tally: {} on: {}", iteration(), key, context.getId(), | ||
tally.get(), member.getId()); | ||
if (failedMajority != null && !majorityFailed) { | ||
if (tally.get() < context.majority()) { | ||
majorityFailed = true; | ||
log.debug("Failed to obtain majority of: {} for: {} tally: {} required: {} on: {}", key, | ||
context.getId(), tally.get(), context.majority(), member.getId()); | ||
failedMajority.run(); | ||
} | ||
} | ||
if (onComplete != null) { | ||
onComplete.accept(tally.get()); | ||
} | ||
} else if (!allow) { | ||
log.trace("Termination of: {} for: {} tally: {} on: {}", key, context.getId(), tally.get(), member.getId()); | ||
} else { | ||
if (onMajority != null && !majoritySucceed) { | ||
if (tally.get() >= context.majority()) { | ||
majoritySucceed = true; | ||
log.debug("Obtained: {} majority of: {} for: {} tally: {} on: {}", current, key, context.getId(), | ||
tally.get(), member.getId()); | ||
onMajority.run(); | ||
} | ||
} | ||
} | ||
} | ||
|
||
private void schedule(Runnable proceed) { | ||
scheduler.schedule(Utils.wrapped(() -> exec.execute(Utils.wrapped(proceed, log)), log), frequency.toNanos(), | ||
TimeUnit.NANOSECONDS); | ||
} | ||
} |
31 changes: 31 additions & 0 deletions
31
memberships/src/test/java/com/salesforce/apollo/ring/RingIteratorSyncTest.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,31 @@ | ||
package com.salesforce.apollo.ring; | ||
|
||
import com.salesforce.apollo.archipelago.Link; | ||
import com.salesforce.apollo.archipelago.RouterImpl; | ||
import com.salesforce.apollo.crypto.Digest; | ||
import com.salesforce.apollo.membership.Context; | ||
import com.salesforce.apollo.membership.Member; | ||
import com.salesforce.apollo.membership.SigningMember; | ||
import com.salesforce.apollo.utils.Utils; | ||
|
||
import java.time.Duration; | ||
import java.util.concurrent.Executor; | ||
import java.util.concurrent.Executors; | ||
import java.util.concurrent.ScheduledExecutorService; | ||
|
||
/** | ||
* @author hal.hildebrand | ||
**/ | ||
public class RingIteratorSyncTest { | ||
public void smokin() throws Exception { | ||
Duration frequency = Duration.ofMillis(100); | ||
Context<Member> context = Context.newBuilder().build(); | ||
SigningMember member = (SigningMember) Utils.getMember(0); | ||
ScheduledExecutorService scheduler = Executors.newSingleThreadScheduledExecutor(); | ||
RouterImpl.CommonCommunications<Link, ?> comm = null; | ||
Executor exec = Executors.newVirtualThreadPerTaskExecutor(); | ||
var sync = new RingIteratorSync<>(frequency, context, member, scheduler, comm, exec); | ||
|
||
sync.iterate(Digest.NONE, (link, round) -> null, (m, result, link) -> false); | ||
} | ||
} |