io.lightflame.websocket.FlameWsExceptionStore Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of lightflame-core Show documentation

Full web framework

The newest version!

package io.lightflame.websocket;

import java.util.HashMap;
import java.util.Map;


/**
 * DefaultExceptionStore
 */
public class FlameWsExceptionStore {
    static private Map functionMap = new HashMap<>();

    public ExceptionWsFunction getFunction(Throwable e){
        ExceptionWsFunction f = functionMap.get(e);
        if (f == null){
            f = defaultFunction();
        }
        return f;
    }

    public void add(Exception e, ExceptionWsFunction function){
        functionMap.put(e, function);
    }

    private ExceptionWsFunction defaultFunction(){
        return (e) -> {
            e.printStackTrace();
        };
    }
    
}