org.graylog2.shared.bindings.providers.MessagePackProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of graylog2-shared Show documentation

Graylog shared code

There is a newer version: 1.3.4

/**
 * This file is part of Graylog.
 *
 * Graylog 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.
 *
 * Graylog 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 Graylog.  If not, see .
 */

package org.graylog2.shared.bindings.providers;

import org.graylog2.plugin.RadioMessage;
import org.msgpack.MessagePack;

import javax.inject.Provider;
import java.io.ByteArrayOutputStream;
import java.io.IOException;

/**
 * MessagePack generates classes for each new instance and message type which can leak into PermGen.
 *
 * Must be a singleton provider!
 */
public class MessagePackProvider implements Provider {
    @Override
    public MessagePack get() {
        final MessagePack messagePack = new MessagePack();

        // Use a separate class loader for msgpack to avoid generation of duplicate class names.
        // The JavaassistTemplateBuilder used by MessagePack uses a sequence number for class naming
        // and is not thread-safe.
        messagePack.setClassLoader(new ClassLoader(Thread.currentThread().getContextClassLoader()) {});

        try {
            // Eagerly generate RadioMessage classes in the MessagePack object to avoid doing it on runtime.
            // The generated code is thread-safe, but generating it is not.
            final RadioMessage radioMessage = new RadioMessage();
            final ByteArrayOutputStream stream = new ByteArrayOutputStream();
            messagePack.write(stream, radioMessage);
            final byte[] bytes = stream.toByteArray();
            messagePack.read(bytes, RadioMessage.class);
        } catch (IOException ignore) {
        }

        return messagePack;
    }
}