io.vlingo.lattice.router.PartitioningCommandRouter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vlingo-lattice Show documentation

Tooling for reactive Domain-Driven Design projects that are highly concurrent. Includes compute grid, actor caching, spaces, cross-node cluster messaging, CQRS, and Event Sourcing support.

There is a newer version: 1.7.5

Show newest version

// Copyright © 2012-2020 VLINGO LABS. All rights reserved.
//
// This Source Code Form is subject to the terms of the
// Mozilla Public License, v. 2.0. If a copy of the MPL
// was not distributed with this file, You can obtain
// one at https://mozilla.org/MPL/2.0/.

package io.vlingo.lattice.router;

import io.vlingo.actors.ContentBasedRouter;
import io.vlingo.actors.Definition;
import io.vlingo.actors.Routee;
import io.vlingo.actors.RouterSpecification;
import io.vlingo.actors.Routing;
import io.vlingo.lattice.model.Command;

/**
 * The {@code CommandRouter} implementation for partitioning on the {@code RoutableCommand}.
 */
public class PartitioningCommandRouter extends ContentBasedRouter implements CommandRouter {
  private Routee currentRoutee;
  private final int totalRoutees;

  /**
   * Constructs my default state.
   * @param totalRoutees the int number or routees to create
   */
  public PartitioningCommandRouter(final int totalRoutees) {
    super(new RouterSpecification(
            totalRoutees,
            Definition.has(CommandRouterWorkerActor.class, Definition.NoParameters),
            CommandRouter.class));

    this.totalRoutees = totalRoutees;
  }

  /**
   * Route to routee of calculated partition.
   * @see io.vlingo.lattice.router.CommandRouter#route(io.vlingo.lattice.router.RoutableCommand)
   */
  @Override
  public  void route(final RoutableCommand command) {
    final int partition = command.hashCode() % totalRoutees;
    currentRoutee = routeeAt(partition);
    dispatchCommand(CommandRouter::route, command);
  }

  /**
   * @see io.vlingo.actors.ContentBasedRouter#computeRouting()
   */
  @Override
  protected Routing computeRouting() {
    return Routing.with(currentRoutee);
  }
}