me.schiz.jmeter.protocol.SessionStorage.java Source code

Introduction

Here is the source code for me.schiz.jmeter.protocol.SessionStorage.java

Source

/*
 * 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 me.schiz.jmeter.protocol;

import org.apache.commons.net.PrintCommandListener;
import org.apache.commons.net.SocketClient;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.log.Logger;

import java.io.*;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @author Epikhin Mikhail (epihin-m@yandex.ru)
 * @version 0.2
 * Simple Singleton for store all  apache commons socket-sessions
 */
public class SessionStorage {
    private static final Logger log = LoggingManager.getLoggerForClass();
    private static SessionStorage instance = null;
    private ConcurrentHashMap<String, SocketClient> map = null;
    private ConcurrentHashMap<String, PrintWriter> logMap = null;
    private ConcurrentHashMap<String, proto_type> protoTypeMap = null;

    public enum proto_type {
        PLAIN, SSL, STARTTLS
    }

    public enum PROTOCOL {
        IMAP, POP3, SMTP
    }

    public SessionStorage() {
        map = new ConcurrentHashMap<String, SocketClient>();
        protoTypeMap = new ConcurrentHashMap<String, proto_type>();
    }

    public static SessionStorage getInstance() {
        synchronized (SessionStorage.class) {
            if (instance == null)
                instance = new SessionStorage();
            return instance;
        }
    }

    public SocketClient getClient(String client) {
        log.debug("get client `" + client + "`");
        return map.get(client);
    }

    public proto_type getClientType(String client) {
        log.debug("get client `" + client + "`");
        return protoTypeMap.get(client);
    }

    public void putClient(String client, SocketClient imapClient, proto_type type) {
        map.put(client, imapClient);
        protoTypeMap.put(client, type);
        log.debug("put client `" + client + "` with type " + type);
    }

    public void removeClient(String client) {
        map.remove(client);
        log.debug("remove client `" + client + "`");
    }

    public void installLog(String client, String logPath) {
        if (map.get(client) == null) {
            log.warn("can't install log, because not found client `" + client + "`");
            return;
        }
        try {
            PrintWriter pw = new PrintWriter(
                    new OutputStreamWriter(new BufferedOutputStream(new FileOutputStream(logPath, false)), "UTF-8"),
                    true);

            logMap.put(client, pw);

            map.get(client).addProtocolCommandListener(new PrintCommandListener(pw, true));
        } catch (UnsupportedEncodingException e) {
            log.error("UTF-8 not supported", e);
        } catch (FileNotFoundException e) {
            log.error("Not found `" + logPath + "`", e);
        }
    }

}