com.github.veithen.visualwas.connector.mapped.MappedObjectOutputStream Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of connector Show documentation

The newest version!

/*
 * #%L
 * VisualWAS
 * %%
 * Copyright (C) 2013 - 2020 Andreas Veithen
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as
 * published by the Free Software Foundation, either version 3 of the 
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public 
 * License along with this program.  If not, see
 * .
 * #L%
 */
package com.github.veithen.visualwas.connector.mapped;

import java.io.IOException;
import java.io.ObjectOutputStream;
import java.io.ObjectStreamClass;

import com.github.veithen.visualwas.connector.feature.InvocationContext;

/**
 * The {@link ObjectOutputStream} implementation used when the {@link ClassMappingFeature} is
 * enabled. It gives the mapped classes access to the {@link InvocationContext}.
 */
public final class MappedObjectOutputStream extends ObjectOutputStream {
    private final ClassDescriptorRewritingOutputStream out;
    private final InvocationContext context;

    MappedObjectOutputStream(ClassDescriptorRewritingOutputStream out, InvocationContext context) throws IOException {
        super(out);
        this.out = out;
        this.context = context;
    }

    public InvocationContext getInvocationContext() {
        return context;
    }

    @Override
    protected void writeClassDescriptor(ObjectStreamClass desc) throws IOException {
        flush();
        out.startClassDescriptor();
        super.writeClassDescriptor(desc);
        flush();
        out.endClassDescriptor();
    }
}