pl.allegro.tech.hermes.common.schema.AvroCompiledSchemaRepositoryFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hermes-common Show documentation

Fast and reliable message broker built on top of Kafka.

The newest version!

package pl.allegro.tech.hermes.common.schema;

import org.apache.avro.Schema;
import pl.allegro.tech.hermes.schema.CachedCompiledSchemaRepository;
import pl.allegro.tech.hermes.schema.CompiledSchemaRepository;
import pl.allegro.tech.hermes.schema.DirectCompiledSchemaRepository;
import pl.allegro.tech.hermes.schema.RawSchemaClient;
import pl.allegro.tech.hermes.schema.SchemaCompilersFactory;

import java.time.Duration;

public class AvroCompiledSchemaRepositoryFactory {

    private final RawSchemaClient rawSchemaClient;
    private final int maximumSize;
    private final Duration expireAfterAccess;
    private final boolean cacheEnabled;

    public AvroCompiledSchemaRepositoryFactory(RawSchemaClient rawSchemaClient,
                                               int maximumSize,
                                               Duration expireAfterAccess,
                                               boolean cacheEnabled) {
        this.rawSchemaClient = rawSchemaClient;
        this.maximumSize = maximumSize;
        this.expireAfterAccess = expireAfterAccess;
        this.cacheEnabled = cacheEnabled;
    }

    public CompiledSchemaRepository provide() {
        CompiledSchemaRepository repository = new DirectCompiledSchemaRepository<>(rawSchemaClient,
                SchemaCompilersFactory.avroSchemaCompiler());

        if (cacheEnabled) {
            return new CachedCompiledSchemaRepository<>(repository,
                    maximumSize,
                    expireAfterAccess);
        } else {
            return repository;
        }
    }
}