io.datakernel.loader.ResourcesNameLoadingService Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of datakernel-http Show documentation

High-performance asynchronous HTTP clients and servers collection. Package contains a bunch of different built-in servlets for request dispatching, loading of a static content, etc.

There is a newer version: 3.1.0

Show newest version

package io.datakernel.loader;

import io.datakernel.async.Stage;
import io.datakernel.eventloop.Eventloop;
import io.datakernel.eventloop.EventloopService;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.file.Paths;
import java.util.HashSet;
import java.util.Set;
import java.util.concurrent.ExecutorService;

public class ResourcesNameLoadingService implements EventloopService {
	private final Eventloop eventloop;
	private final ClassLoader loader;
	private final ExecutorService executorService;

	private Set names;
	private String resourcePath;

	private ResourcesNameLoadingService(Eventloop eventloop, ExecutorService executorService, ClassLoader loader,
	                                    String resourcePath) {
		this.eventloop = eventloop;
		this.executorService = executorService;
		this.loader = loader;
		this.resourcePath = resourcePath;
	}

	public static ResourcesNameLoadingService create(Eventloop eventloop, ExecutorService executorService,
	                                                 ClassLoader classLoader, String resourcePath) {
		return new ResourcesNameLoadingService(eventloop, executorService, classLoader, resourcePath);
	}

	public static ResourcesNameLoadingService createRoot(Eventloop eventloop, ExecutorService executorService,
	                                                     ClassLoader classLoader) {
		return new ResourcesNameLoadingService(eventloop, executorService, classLoader, "");
	}

	@Override
	public Eventloop getEventloop() {
		return eventloop;
	}

	@Override
	public Stage start() {
		return Stage.ofCallable(executorService,
				() -> {
					Set fileNames = new HashSet<>();
					InputStream in = loader.getResourceAsStream(resourcePath);
					try (BufferedReader reader = new BufferedReader(new InputStreamReader(in))) {
						String line;
						while ((line = reader.readLine()) != null) {
							fileNames.add(Paths.get(resourcePath).resolve(Paths.get(line)).toString());
						}
						return fileNames;
					} catch (IOException e) {
						throw new RuntimeException(e);
					}
				})
				.whenResult(strings -> names = strings)
				.toVoid();
	}

	@Override
	public Stage stop() {
		return Stage.of(null);
	}

	public Set getNames() {
		return names;
	}

	public boolean contains(String value) {
		return names.contains(value);
	}
}