be.bagofwords.application.status.ListUrlsController Maven / Gradle / Ivy

Go to download
package be.bagofwords.application.status;

import be.bagofwords.application.CloseableComponent;
import be.bagofwords.application.annotations.BowController;
import be.bagofwords.util.Pair;
import be.bagofwords.web.BaseController;
import org.springframework.beans.factory.annotation.Autowired;
import spark.Request;
import spark.Response;

import java.util.ArrayList;
import java.util.List;

/**
 * Created by Koen Deschacht ([email protected]) on 07/10/14.
 */

@BowController
public class ListUrlsController extends BaseController implements CloseableComponent {

    private final List> urls;
    private final RegisterUrlsServer registerUrlsServer;

    @Autowired
    public ListUrlsController(RemoteRegisterUrlsServerProperties properties) {
        super("/paths");
        this.urls = new ArrayList<>();
        this.registerUrlsServer = new RegisterUrlsServer(properties.getRegisterUrlServerPort(), this);
        this.registerUrlsServer.start();
    }

    @Override
    protected String handleRequest(Request request, Response response) throws Exception {
        StringBuilder result = new StringBuilder();
        synchronized (urls) {
            for (Pair url : urls) {
                result.append("" + url.getFirst() + " " + url.getSecond() + "
");
            }
        }
        return result.toString();
    }

    public void registerUrl(String name, String url) {
        synchronized (urls) {
            Pair toRegister = new Pair<>(name, url);
            if (!urls.contains(toRegister)) {
                urls.add(0, toRegister);
            }
            while (urls.size() > 20) {
                urls.remove(urls.get(urls.size() - 1));
            }
        }
    }

    @Override
    public void terminate() {
        registerUrlsServer.terminateAndWaitForFinish();
    }
}