Examples with ValidatorProvider tech.pegasys.teku.validator.client.loader.ValidatorSource.ValidatorProvider used on opensource projects

Example 1 with ValidatorProvider

use of tech.pegasys.teku.validator.client.loader.ValidatorSource.ValidatorProvider in project teku by ConsenSys.

the class MultithreadedValidatorLoader method loadValidators.

public static void loadValidators(final OwnedValidators ownedValidators, final Map<BLSPublicKey, ValidatorProvider> providers, final GraffitiProvider graffitiProvider) {
    final int totalValidatorCount = providers.size();
    STATUS_LOG.loadingValidators(totalValidatorCount);
    final ExecutorService executorService = Executors.newFixedThreadPool(Math.min(4, Runtime.getRuntime().availableProcessors()));
    try {
        final AtomicInteger numberOfLoadedKeys = new AtomicInteger(0);
        final List<Future<Validator>> futures = providers.values().stream().map(provider -> executorService.submit(() -> {
            final Validator validator = new Validator(provider.getPublicKey(), new DeletableSigner(provider.createSigner()), graffitiProvider, provider.isReadOnly());
            int loadedValidatorCount = numberOfLoadedKeys.incrementAndGet();
            if (loadedValidatorCount % 10 == 0) {
                STATUS_LOG.atLoadedValidatorNumber(loadedValidatorCount, totalValidatorCount);
            }
            return validator;
        })).collect(toList());
        final List<Validator> addedValidators = new ArrayList<>();
        for (Future<Validator> future : futures) {
            final Validator validator = future.get();
            addedValidators.add(validator);
        }
        // Only start adding validators once we've successfully loaded all keys
        addedValidators.forEach(ownedValidators::addValidator);
        STATUS_LOG.validatorsInitialised(addedValidators.stream().map(validator -> validator.getPublicKey().toAbbreviatedString()).collect(toList()));
    } catch (InterruptedException e) {
        throw new RuntimeException("Interrupted while attempting to load validator key files", e);
    } catch (ExecutionException e) {
        if (e.getCause() instanceof RuntimeException) {
            throw (RuntimeException) e.getCause();
        }
        throw new RuntimeException("Unable to load validator key files", e);
    } finally {
        executorService.shutdownNow();
    }
}

Example 2 with ValidatorProvider

use of tech.pegasys.teku.validator.client.loader.ValidatorSource.ValidatorProvider in project teku by ConsenSys.

the class LocalValidatorSourceTest method shouldLoadKeysFromKeyStores.

@Test
void shouldLoadKeysFromKeyStores(@TempDir final Path tempDir) throws Exception {
    // load keystores from resources
    final Path scryptKeystore = Path.of(Resources.getResource("scryptTestVector.json").toURI());
    final Path pbkdf2Keystore = Path.of(Resources.getResource("pbkdf2TestVector.json").toURI());
    // create password file
    final Path tempPasswordFile = createTempFile(tempDir, "pass", ".txt");
    writeString(tempPasswordFile, EXPECTED_PASSWORD);
    final List<Pair<Path, Path>> keystorePasswordFilePairs = List.of(Pair.of(scryptKeystore, tempPasswordFile), Pair.of(pbkdf2Keystore, tempPasswordFile));
    when(keyStoreFilesLocator.parse()).thenReturn(keystorePasswordFilePairs);
    final List<ValidatorProvider> availableValidators = validatorSource.getAvailableValidators();
    assertThat(availableValidators).hasSize(2);
    // Both keystores encyrpt the same key.
    assertProviderMatchesKey(availableValidators.get(0), EXPECTED_BLS_KEY_PAIR);
    assertProviderMatchesKey(availableValidators.get(1), EXPECTED_BLS_KEY_PAIR);
}