org.redkalex.socks.NodeSocksServer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of redkale-plugins Show documentation
Redkale-Plugins -- java framework
There is a newer version: 2.7.7
/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package org.redkalex.socks;

import org.redkale.net.Server;
import org.redkale.util.AnyValue;
import org.redkale.boot.Application;
import org.redkale.net.Servlet;
import org.redkale.boot.ClassFilter;
import org.redkale.boot.NodeServer;
import org.redkale.boot.NodeProtocol;
import org.redkale.boot.ClassFilter.FilterEntry;
import static org.redkale.boot.NodeServer.LINE_SEPARATOR;
import org.redkale.util.AnyValue.DefaultAnyValue;
import java.lang.reflect.*;
import java.net.*;
import java.util.logging.*;

/**
 * < server protocol="SOCKS" host="0.0.0.0" port="1080" bindaddr="外网IP"> < /server>
 *
 * 详情见: https://redkale.org
 *
 * @author zhangjx
 */
@NodeProtocol("SOCKS")
public class NodeSocksServer extends NodeServer {

    private final SocksServer socksServer;

    public NodeSocksServer(Application application, AnyValue serconf) {
        super(application, createServer(application, serconf));
        this.socksServer = (SocksServer) server;
    }

    private static Server createServer(Application application, AnyValue serconf) {
        return new SocksServer(application.getStartTime(), application.getResourceFactory().createChild());
    }

    @Override
    public InetSocketAddress getSocketAddress() {
        return socksServer == null ? null : socksServer.getSocketAddress();
    }

    @Override
    protected ClassFilter createServletClassFilter() {
        return createClassFilter(null, null, SocksServlet.class, null, null, "servlets", "servlet");
    }

    @Override
    protected void loadServlet(ClassFilter servletFilter, ClassFilter otherFilter) throws Exception {
        if (socksServer != null) loadSocksServlet(this.serverConf.getAnyValue("servlets"), servletFilter);
    }

    protected void loadSocksServlet(final AnyValue conf, ClassFilter filter) throws Exception {
        final StringBuilder sb = logger.isLoggable(Level.FINE) ? new StringBuilder() : null;
        final String threadName = "[" + Thread.currentThread().getName() + "] ";
        for (FilterEntry en : filter.getFilterEntrys()) {
            Class clazz = (Class) en.getType();
            if (Modifier.isAbstract(clazz.getModifiers())) continue;
            final SocksServlet servlet = clazz.getDeclaredConstructor().newInstance();
            resourceFactory.inject(servlet);
            DefaultAnyValue servletConf = (DefaultAnyValue) en.getProperty();
            this.socksServer.addServlet(servlet, null, servletConf);
            if (sb != null) sb.append(threadName).append(" Loaded ").append(clazz.getName()).append(" --> ").append(servletConf).append(LINE_SEPARATOR);
        }
        if (sb != null && sb.length() > 0) logger.log(Level.FINE, sb.toString());
    }

    @Override
    protected void loadFilter(ClassFilter filterFilter, ClassFilter otherFilter) throws Exception {
    }

    @Override
    protected ClassFilter createFilterClassFilter() {
        return null;
    }

}