uk.co.real_logic.sbe.generation.java.JavaOutputManager Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sbe-tool Show documentation

FIX/SBE - OSI layer 6 presentation for encoding and decoding application messages in binary format for low-latency applications.

The newest version!

/*
 * Copyright 2013-2024 Real Logic Limited.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package uk.co.real_logic.sbe.generation.java;

import org.agrona.collections.Object2NullableObjectHashMap;
import org.agrona.collections.Object2ObjectHashMap;
import org.agrona.generation.DynamicPackageOutputManager;
import org.agrona.generation.PackageOutputManager;

import java.io.FilterWriter;
import java.io.IOException;
import java.io.Writer;

/**
 * Implementation of {@link DynamicPackageOutputManager} for Java.
 */
public class JavaOutputManager implements DynamicPackageOutputManager
{
    private final String baseDirName;
    private final PackageOutputManager initialPackageOutputManager;
    private PackageOutputManager actingPackageOutputManager;
    private final Object2ObjectHashMap outputManagerByPackageName
        = new Object2NullableObjectHashMap<>();

    /**
     * Constructor.
     *
     * @param baseDirName the target directory
     * @param packageName the initial package name
     */
    public JavaOutputManager(final String baseDirName, final String packageName)
    {
        initialPackageOutputManager = new PackageOutputManager(baseDirName, packageName);
        actingPackageOutputManager = initialPackageOutputManager;
        this.baseDirName = baseDirName;
    }

    /**
     * {@inheritDoc}
     */
    public void setPackageName(final String packageName)
    {
        actingPackageOutputManager = outputManagerByPackageName.get(packageName);
        if (actingPackageOutputManager == null)
        {
            actingPackageOutputManager = new PackageOutputManager(baseDirName, packageName);
            outputManagerByPackageName.put(packageName, actingPackageOutputManager);
        }
    }

    private void resetPackage()
    {
        actingPackageOutputManager = initialPackageOutputManager;
    }

    /**
     * {@inheritDoc}
     */
    public Writer createOutput(final String name) throws IOException
    {
        return new FilterWriter(actingPackageOutputManager.createOutput(name))
        {
            public void close() throws IOException
            {
                super.close();
                resetPackage();
            }
        };
    }
}