org.wildfly.httpclient.ejb.HttpDiscoveryHandler Maven / Gradle / Ivy
/*
* JBoss, Home of Professional Open Source.
* Copyright 2020 Red Hat, Inc., and individual contributors
* as indicated by the @author tags.
*
* 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
*
* http://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 org.wildfly.httpclient.ejb;
import io.undertow.server.HttpServerExchange;
import io.undertow.util.Headers;
import org.jboss.ejb.client.EJBModuleIdentifier;
import org.jboss.ejb.server.Association;
import org.jboss.ejb.server.ModuleAvailabilityListener;
import org.jboss.marshalling.Marshaller;
import org.jboss.marshalling.Marshalling;
import org.jboss.marshalling.MarshallingConfiguration;
import org.jboss.marshalling.river.RiverMarshallerFactory;
import org.wildfly.httpclient.common.NoFlushByteOutput;
import java.io.ByteArrayOutputStream;
import java.nio.ByteBuffer;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.concurrent.ExecutorService;
/**
* @author Tomasz Adamski
*/
public class HttpDiscoveryHandler extends RemoteHTTPHandler {
private final Set availableModules = new HashSet<>();
public HttpDiscoveryHandler(ExecutorService executorService, Association association) {
super(executorService);
association.registerModuleAvailabilityListener(new ModuleAvailabilityListener() {
@Override
public void moduleAvailable(List modules) {
availableModules.addAll(modules);
}
@Override
public void moduleUnavailable(List modules) {
availableModules.removeAll(modules);
}
});
}
@Override
protected void handleInternal(HttpServerExchange exchange) throws Exception {
exchange.getResponseHeaders().put(Headers.CONTENT_TYPE, EjbHeaders.EJB_DISCOVERY_RESPONSE_VERSION_ONE.toString());
final ByteArrayOutputStream out = new ByteArrayOutputStream();
Marshaller marshaller = new RiverMarshallerFactory().createMarshaller(createMarshallingConfig());
marshaller.start(new NoFlushByteOutput(Marshalling.createByteOutput(out)));
marshaller.writeInt(availableModules.size());
for (EJBModuleIdentifier ejbModuleIdentifier : availableModules) {
marshaller.writeObject(ejbModuleIdentifier);
}
marshaller.finish();
marshaller.flush();
exchange.getResponseSender().send(ByteBuffer.wrap(out.toByteArray()));
}
private MarshallingConfiguration createMarshallingConfig() {
final MarshallingConfiguration marshallingConfiguration = new MarshallingConfiguration();
marshallingConfiguration.setObjectTable(HttpProtocolV1ObjectTable.INSTANCE);
marshallingConfiguration.setVersion(2);
return marshallingConfiguration;
}
}