com.github.ji4597056.DefaultWsHandlerRegistration Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-websocket-forward Show documentation

spring websocket forward

The newest version!

package com.github.ji4597056;

import com.github.ji4597056.server.AbstractWsServerHandler;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import org.springframework.web.socket.server.HandshakeInterceptor;

/**
 * default websocket handler registration
 *
 * @author Jeffrey
 * @since 2017/12/04 18:47
 */
public class DefaultWsHandlerRegistration implements WsHandlerRegistration {

    private Map handlerMap = Collections
        .synchronizedMap(new HashMap<>(5));

    private Map interceptorMap = Collections
        .synchronizedMap(new HashMap<>(5));

    @Override
    public void addHandler(AbstractWsServerHandler... handlers) {
        Arrays.stream(handlers)
            .forEach(handler -> handlerMap.putIfAbsent(handler.getClass().getName(), handler));
    }

    @Override
    public void addInterceptor(HandshakeInterceptor... interceptors) {
        Arrays.stream(interceptors)
            .forEach(interceptor -> interceptorMap
                .putIfAbsent(interceptor.getClass().getName(), interceptor));
    }

    @Override
    public AbstractWsServerHandler getHandler(String handlerClassName) {
        return handlerMap.get(handlerClassName);
    }

    @Override
    public HandshakeInterceptor getInterceptor(String interceptorClassName) {
        return interceptorMap.get(interceptorClassName);
    }
}