Examples with AssignPartitions com.hazelcast.internal.partition.operation.AssignPartitions used on opensource projects

Example 1 with AssignPartitions

use of com.hazelcast.internal.partition.operation.AssignPartitions in project hazelcast by hazelcast.

the class PartitionDataSerializerHook method createFactory.

@Override
public DataSerializableFactory createFactory() {
    ConstructorFunction<Integer, IdentifiedDataSerializable>[] constructors = new ConstructorFunction[LEN];
    constructors[PARTITION_RUNTIME_STATE] = arg -> new PartitionRuntimeState();
    constructors[ASSIGN_PARTITIONS] = arg -> new AssignPartitions();
    constructors[PARTITION_BACKUP_REPLICA_ANTI_ENTROPY] = arg -> new PartitionBackupReplicaAntiEntropyOperation();
    constructors[FETCH_PARTITION_STATE] = arg -> new FetchPartitionStateOperation();
    constructors[HAS_ONGOING_MIGRATION] = arg -> new HasOngoingMigration();
    constructors[MIGRATION_COMMIT] = arg -> new MigrationCommitOperation();
    constructors[PARTITION_STATE_OP] = arg -> new PartitionStateOperation();
    constructors[PROMOTION_COMMIT] = arg -> new PromotionCommitOperation();
    constructors[REPLICA_SYNC_REQUEST] = arg -> new PartitionReplicaSyncRequest();
    constructors[REPLICA_SYNC_RESPONSE] = arg -> new PartitionReplicaSyncResponse();
    constructors[REPLICA_SYNC_RETRY_RESPONSE] = arg -> new PartitionReplicaSyncRetryResponse();
    constructors[SAFE_STATE_CHECK] = arg -> new SafeStateCheckOperation();
    constructors[SHUTDOWN_REQUEST] = arg -> new ShutdownRequestOperation();
    constructors[SHUTDOWN_RESPONSE] = arg -> new ShutdownResponseOperation();
    constructors[REPLICA_FRAGMENT_MIGRATION_STATE] = arg -> new ReplicaFragmentMigrationState();
    constructors[MIGRATION] = arg -> new MigrationOperation();
    constructors[MIGRATION_REQUEST] = arg -> new MigrationRequestOperation();
    constructors[NON_FRAGMENTED_SERVICE_NAMESPACE] = arg -> NonFragmentedServiceNamespace.INSTANCE;
    constructors[PARTITION_REPLICA] = arg -> new PartitionReplica();
    constructors[PUBLISH_COMPLETED_MIGRATIONS] = arg -> new PublishCompletedMigrationsOperation();
    constructors[PARTITION_STATE_CHECK_OP] = arg -> new PartitionStateCheckOperation();
    constructors[REPLICA_MIGRATION_EVENT] = arg -> new ReplicaMigrationEventImpl();
    constructors[MIGRATION_EVENT] = arg -> new MigrationStateImpl();
    constructors[PARTITION_LOST_EVENT] = arg -> new PartitionLostEventImpl();
    constructors[REPLICA_SYNC_REQUEST_OFFLOADABLE] = arg -> new PartitionReplicaSyncRequestOffloadable();
    return new ArrayDataSerializableFactory(constructors);
}

Example 2 with AssignPartitions

use of com.hazelcast.internal.partition.operation.AssignPartitions in project hazelcast by hazelcast.

the class InternalPartitionServiceImpl method triggerMasterToAssignPartitions.

/**
 * Sends a {@link AssignPartitions} to the master to assign partitions.
 */
private void triggerMasterToAssignPartitions() {
    if (!shouldTriggerMasterToAssignPartitions()) {
        return;
    }
    ClusterServiceImpl clusterService = node.getClusterService();
    ClusterState clusterState = clusterService.getClusterState();
    if (!clusterState.isMigrationAllowed()) {
        logger.warning("Partitions can't be assigned since cluster-state=" + clusterState);
        return;
    }
    final Address masterAddress = latestMaster;
    if (masterAddress == null || masterAddress.equals(node.getThisAddress())) {
        return;
    }
    if (masterTriggered.compareAndSet(false, true)) {
        OperationServiceImpl operationService = nodeEngine.getOperationService();
        InvocationFuture<PartitionRuntimeState> future = operationService.invokeOnTarget(SERVICE_NAME, new AssignPartitions(), masterAddress);
        future.whenCompleteAsync((partitionState, throwable) -> {
            if (throwable == null) {
                resetMasterTriggeredFlag();
                if (partitionState != null) {
                    partitionState.setMaster(masterAddress);
                    processPartitionRuntimeState(partitionState);
                }
            } else {
                resetMasterTriggeredFlag();
                logger.severe(throwable);
            }
        });
        masterTrigger.executeWithDelay();
    }
}