Examples with BlockEncoding io.trino.spi.block.BlockEncoding used on opensource projects

Example 1 with BlockEncoding

use of io.trino.spi.block.BlockEncoding in project trino by trinodb.

the class BlockEncodingManager method getBlockEncoding.

public BlockEncoding getBlockEncoding(String encodingName) {
    BlockEncoding blockEncoding = blockEncodings.get(encodingName);
    checkArgument(blockEncoding != null, "Unknown block encoding: %s", encodingName);
    return blockEncoding;
}

Example 2 with BlockEncoding

use of io.trino.spi.block.BlockEncoding in project trino by trinodb.

the class InternalBlockEncodingSerde method writeBlock.

@Override
public void writeBlock(SliceOutput output, Block block) {
    while (true) {
        // get the encoding name
        String encodingName = block.getEncodingName();
        // look up the BlockEncoding
        BlockEncoding blockEncoding = blockEncodings.apply(encodingName);
        // see if a replacement block should be written instead
        Optional<Block> replacementBlock = blockEncoding.replacementBlockForWrite(block);
        if (replacementBlock.isPresent()) {
            block = replacementBlock.get();
            continue;
        }
        // write the name to the output
        writeLengthPrefixedString(output, encodingName);
        // write the block to the output
        blockEncoding.writeBlock(this, output, block);
        break;
    }
}

Example 3 with BlockEncoding

use of io.trino.spi.block.BlockEncoding in project trino by trinodb.

the class PluginManager method installPluginInternal.

private void installPluginInternal(Plugin plugin, Function<CatalogName, ClassLoader> duplicatePluginClassLoaderFactory) {
    for (BlockEncoding blockEncoding : plugin.getBlockEncodings()) {
        log.info("Registering block encoding %s", blockEncoding.getName());
        blockEncodingManager.addBlockEncoding(blockEncoding);
    }
    for (Type type : plugin.getTypes()) {
        log.info("Registering type %s", type.getTypeSignature());
        typeRegistry.addType(type);
    }
    for (ParametricType parametricType : plugin.getParametricTypes()) {
        log.info("Registering parametric type %s", parametricType.getName());
        typeRegistry.addParametricType(parametricType);
    }
    for (ConnectorFactory connectorFactory : plugin.getConnectorFactories()) {
        log.info("Registering connector %s", connectorFactory.getName());
        connectorManager.addConnectorFactory(connectorFactory, duplicatePluginClassLoaderFactory);
    }
    Set<Class<?>> functions = plugin.getFunctions();
    if (!functions.isEmpty()) {
        log.info("Registering functions from %s", plugin.getClass().getSimpleName());
        InternalFunctionBundleBuilder builder = InternalFunctionBundle.builder();
        functions.forEach(builder::functions);
        globalFunctionCatalog.addFunctions(builder.build());
    }
    for (SessionPropertyConfigurationManagerFactory sessionConfigFactory : plugin.getSessionPropertyConfigurationManagerFactories()) {
        log.info("Registering session property configuration manager %s", sessionConfigFactory.getName());
        sessionPropertyDefaults.addConfigurationManagerFactory(sessionConfigFactory);
    }
    for (ResourceGroupConfigurationManagerFactory configurationManagerFactory : plugin.getResourceGroupConfigurationManagerFactories()) {
        log.info("Registering resource group configuration manager %s", configurationManagerFactory.getName());
        resourceGroupManager.addConfigurationManagerFactory(configurationManagerFactory);
    }
    for (SystemAccessControlFactory accessControlFactory : plugin.getSystemAccessControlFactories()) {
        log.info("Registering system access control %s", accessControlFactory.getName());
        accessControlManager.addSystemAccessControlFactory(accessControlFactory);
    }
    passwordAuthenticatorManager.ifPresent(authenticationManager -> {
        for (PasswordAuthenticatorFactory authenticatorFactory : plugin.getPasswordAuthenticatorFactories()) {
            log.info("Registering password authenticator %s", authenticatorFactory.getName());
            authenticationManager.addPasswordAuthenticatorFactory(authenticatorFactory);
        }
    });
    for (CertificateAuthenticatorFactory authenticatorFactory : plugin.getCertificateAuthenticatorFactories()) {
        log.info("Registering certificate authenticator %s", authenticatorFactory.getName());
        certificateAuthenticatorManager.addCertificateAuthenticatorFactory(authenticatorFactory);
    }
    headerAuthenticatorManager.ifPresent(authenticationManager -> {
        for (HeaderAuthenticatorFactory authenticatorFactory : plugin.getHeaderAuthenticatorFactories()) {
            log.info("Registering header authenticator %s", authenticatorFactory.getName());
            authenticationManager.addHeaderAuthenticatorFactory(authenticatorFactory);
        }
    });
    for (EventListenerFactory eventListenerFactory : plugin.getEventListenerFactories()) {
        log.info("Registering event listener %s", eventListenerFactory.getName());
        eventListenerManager.addEventListenerFactory(eventListenerFactory);
    }
    for (GroupProviderFactory groupProviderFactory : plugin.getGroupProviderFactories()) {
        log.info("Registering group provider %s", groupProviderFactory.getName());
        groupProviderManager.addGroupProviderFactory(groupProviderFactory);
    }
    for (ExchangeManagerFactory exchangeManagerFactory : plugin.getExchangeManagerFactories()) {
        log.info("Registering exchange manager %s", exchangeManagerFactory.getName());
        exchangeManagerRegistry.addExchangeManagerFactory(exchangeManagerFactory);
    }
}

Example 4 with BlockEncoding

use of io.trino.spi.block.BlockEncoding in project trino by trinodb.

the class HiveBlockEncodingSerde method writeBlock.

@Override
public void writeBlock(SliceOutput output, Block block) {
    while (true) {
        // get the encoding name
        String encodingName = block.getEncodingName();
        // look up the encoding factory
        BlockEncoding blockEncoding = blockEncodings.get(encodingName);
        // see if a replacement block should be written instead
        Optional<Block> replacementBlock = blockEncoding.replacementBlockForWrite(block);
        if (replacementBlock.isPresent()) {
            block = replacementBlock.get();
            continue;
        }
        // write the name to the output
        writeLengthPrefixedString(output, encodingName);
        // write the block to the output
        blockEncoding.writeBlock(this, output, block);
        break;
    }
}

Example 5 with BlockEncoding

use of io.trino.spi.block.BlockEncoding in project trino by trinodb.

the class HiveBlockEncodingSerde method readBlock.

@Override
public Block readBlock(SliceInput input) {
    // read the encoding name
    String encodingName = readLengthPrefixedString(input);
    // look up the encoding factory
    BlockEncoding blockEncoding = blockEncodings.get(encodingName);
    checkArgument(blockEncoding != null, "Unknown block encoding %s", encodingName);
    // load read the encoding factory from the output stream
    return blockEncoding.readBlock(this, input);
}