SimpleRouter.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.apache.tomee.catalina.routing;

import org.apache.openejb.monitoring.DynamicMBeanWrapper;
import org.apache.openejb.monitoring.LocalMBeanServer;
import org.apache.openejb.monitoring.ObjectNameBuilder;

import javax.management.ManagedAttribute;
import javax.management.ManagedOperation;
import javax.management.ObjectName;
import javax.management.openmbean.CompositeData;
import javax.management.openmbean.CompositeDataSupport;
import javax.management.openmbean.CompositeType;
import javax.management.openmbean.OpenDataException;
import javax.management.openmbean.OpenType;
import javax.management.openmbean.SimpleType;
import javax.management.openmbean.TabularData;
import javax.management.openmbean.TabularDataSupport;
import javax.management.openmbean.TabularType;
import javax.servlet.ServletException;
import java.io.BufferedInputStream;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.URL;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class SimpleRouter {
    private static final Pattern PATTERN = Pattern.compile("(.*)->(.*)");

    private String prefix = "";
    private ObjectName objectName;
    private Route[] routes = new Route[0];
    private final Map<String, Route> cache = new ConcurrentHashMap<String, Route>();

    public SimpleRouter readConfiguration(final URL url) {
        if (url == null) {
            return this;
        }

        try {
            final InputStream is = new BufferedInputStream(url.openStream());
            final BufferedReader reader = new BufferedReader(new InputStreamReader(is));

            String line;
            while ((line = reader.readLine()) != null) {
                line = line.trim();
                if (!line.isEmpty() && !line.startsWith("#")) {
                   parseRoute(line);
                }
            }
        } catch (final IOException e) {
            throw new RouterException("can't read " + url.toExternalForm());
        }
        return this;
    }

    private void parseRoute(final String line) {
        final Matcher matcher = PATTERN.matcher(line);
        if (matcher.matches()) {
            final String from = prefix(matcher.group(1).trim());
            final String to = prefix(matcher.group(2).trim());
            addRoute(new Route().from(from).to(to));
        }
    }

    public String route(final String uri) throws IOException, ServletException {
        if (uri == null) {
            return null;
        }

        final Route cachedRoute = cache.get(uri);
        if (cachedRoute != null) {
            cachedRoute.matches(uri);
            return cachedRoute.cleanDestination(prefix);
        }

        for (final Route route : routes) {
            if (route.matches(uri)) {
                if (route.getOrigin().equals(uri)) {
                    cache.put(uri, route);
                }
                return route.cleanDestination(prefix);
            }
        }

        return null;
    }

    public synchronized void addRoute(final Route route) {
        final Route[] newRoutes = new Route[routes.length + 1];
        System.arraycopy(routes, 0, newRoutes, 0, routes.length);
        newRoutes[routes.length] = route;
        routes = newRoutes;
    }

    public void cleanUp() {
        JMXOff();
        routes = null;
        cache.clear();
    }

    public void setPrefix(final String prefix) {
        if (prefix == null || prefix.isEmpty() || prefix.equals("/")) {
            this.prefix = "";
        } else {
            this.prefix = prefix;
        }
    }

    private String prefix(final String value) {
        if (prefix != null) {
            return prefix + value;
        }
        return value;
    }

    public void JMXOn(final String name) {
        final ObjectNameBuilder jmxName = new ObjectNameBuilder("openejb.management");
        jmxName.set("J2EEServer", "Router");
        jmxName.set("J2EEApplication", name);
        jmxName.set("Type", "SimpleRouter");

        objectName = jmxName.build();
        try {
            LocalMBeanServer.get().registerMBean(new DynamicMBeanWrapper(this), objectName);
        } catch (final Exception e) {
            objectName = null;
        }
    }

    public void JMXOff() {
        if (objectName != null) {
            try {
                LocalMBeanServer.get().unregisterMBean(objectName);
            } catch (final Exception e) {
                // no-op
            }
        }
    }

    @ManagedAttribute
    public TabularData getActiveRoutes() {
        if (routes.length == 0) {
            return null;
        }

        final OpenType<?>[] types = new OpenType<?>[routes.length];
        final String[] keys = new String[types.length];
        final String[] values = new String[types.length];

        for (int i = 0; i < types.length; i++) {
            types[i] = SimpleType.STRING;
            keys[i] = routes[i].getOrigin().substring(prefix.length());
            values[i] = routes[i].getRawDestination().substring(prefix.length());
        }

        try {
            final CompositeType ct = new CompositeType("routes", "routes", keys, keys, types);
            final TabularType type = new TabularType("router", "routes", ct, keys);
            final TabularDataSupport data = new TabularDataSupport(type);

            final CompositeData line = new CompositeDataSupport(ct, keys, values);
            data.put(line);
            return data;
        } catch (final OpenDataException e) {
            return null;
        }
    }

    @ManagedOperation
    public void addRoute(final String from, final String to) {
        addRoute(new Route().from(prefix(from)).to(prefix(to)));
    }

    @ManagedOperation
    public void removeRoute(final String from, final String to) {
        if (routes.length == 0) {
            return;
        }

        for (int i = 0; i < routes.length; i++) {
            if (routes[i].getOrigin().equals(from) && routes[i].getRawDestination().endsWith(to)) {
                final Route[] newRoutes = new Route[routes.length - 1];
                System.arraycopy(routes, 0, newRoutes, 0, i);
                System.arraycopy(routes, i + 1, newRoutes, i, routes.length - i - 1);
                routes = newRoutes;
            }
        }
    }

    public boolean hasPrefix() {
        return prefix != null && !prefix.isEmpty();
    }

    public String getPrefix() {
        return prefix;
    }
}