com.alibaba.antx.config.resource.ResourceManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of antx-autoconfig Show documentation
There is a newer version: 1.2
/*
 * Copyright (c) 2002-2012 Alibaba Group Holding Limited.
 * All rights reserved.
 *
 * Licensed 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 com.alibaba.antx.config.resource;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.URI;
import java.util.Collections;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

import com.alibaba.antx.config.ConfigException;
import com.alibaba.antx.config.resource.file.FileResourceDriver;
import com.alibaba.antx.config.resource.http.HttpResourceDriver;
import com.alibaba.antx.config.resource.ssh.SshResourceDriver;

public class ResourceManager {
    private final Map                   drivers            = new HashMap();
    private final Map                   sessions           = Collections.synchronizedMap(new HashMap()); // driver -> session
    private       AuthenticationHandler authHandlerDefault = new DefaultAuthenticationHandler(this);
    private AuthenticationHandler authHandler;
    private PrintWriter           out;
    private BufferedReader        in;

    public ResourceManager() {
        registerDefaultDrivers();
    }

    private void registerDefaultDrivers() {
        drivers.put("file", new FileResourceDriver(this));
        drivers.put("http", new HttpResourceDriver(this));
        drivers.put("https", drivers.get("http"));
        drivers.put("ssh", new SshResourceDriver(this));
    }

    public AuthenticationHandler getAuthenticationHandler() {
        return authHandler == null ? authHandlerDefault : authHandler;
    }

    public void setAuthenticationHandler(AuthenticationHandler authHandler) {
        this.authHandler = authHandler;
    }

    public Resource getResource(URI uri) {
        Session session = getSession(uri.getScheme());

        return session.getResource(new ResourceURI(uri, session));
    }

    private Session getSession(String type) {
        ResourceDriver driver = (ResourceDriver) drivers.get(type);

        if (driver == null) {
            throw new ConfigException("No drivers for resource type: " + type);
        }

        Session session;

        synchronized (sessions) {
            session = (Session) sessions.get(driver);

            if (session == null) {
                session = driver.open();
                sessions.put(driver, session);
            }
        }

        return session;
    }

    public void close() {
        synchronized (sessions) {
            for (Iterator i = sessions.values().iterator(); i.hasNext(); ) {
                Session session = (Session) i.next();

                i.remove();
                session.close();
            }
        }
    }

    public void log(String message) {
        getOut().println(message);
        getOut().flush();
    }

    public PrintWriter getOut() {
        if (out == null) {
            out = new PrintWriter(System.out, true);
        }

        return out;
    }

    public void setOut(PrintWriter out) {
        this.out = out;
    }

    public BufferedReader getIn() {
        if (in == null) {
            in = new BufferedReader(new InputStreamReader(System.in));
        }

        return in;
    }

    public void setIn(BufferedReader in) {
        this.in = in;
    }
}