All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.g2forge.gearbox.functional.proxy.ResultContext Maven / Gradle / Ivy

Go to download

The newest version!
package com.g2forge.gearbox.functional.proxy;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.UncheckedIOException;
import java.lang.reflect.Method;
import java.lang.reflect.Type;
import java.util.Iterator;
import java.util.List;
import java.util.NoSuchElementException;
import java.util.Spliterator;
import java.util.Spliterators;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;

import com.g2forge.alexandria.adt.collection.CircularBuffer;
import com.g2forge.alexandria.java.io.RuntimeIOException;
import com.g2forge.alexandria.java.typed.ATypeRef;
import com.g2forge.alexandria.java.typed.ATypeRefIdentity;
import com.g2forge.alexandria.java.typed.ITypeRef;
import com.g2forge.gearbox.functional.control.IExplicitResultHandler;
import com.g2forge.gearbox.functional.control.IResultContext;
import com.g2forge.gearbox.functional.runner.IProcess;
import com.g2forge.gearbox.functional.runner.redirect.InheritRedirect;
import com.g2forge.gearbox.functional.runner.redirect.Redirects;

import lombok.Data;
import lombok.Getter;

@Data
class ResultContext implements IResultContext {
	protected final Method method;

	@Getter(lazy = true)
	private final ITypeRef type = computeType();

	protected ITypeRef computeType() {
		return new ATypeRefIdentity() {
			@SuppressWarnings("unchecked")
			@Override
			public Class getErasedType() {
				return (Class) method.getReturnType();
			}

			@Override
			public Type getType() {
				return method.getGenericReturnType();
			}
		};
	}

	@Override
	public IExplicitResultHandler getStandard(ITypeRef type) {
		if (type.getErasedType().isAssignableFrom(Boolean.class) || type.getErasedType().isAssignableFrom(Boolean.TYPE)) return (process, context) -> {
			try {
				return process.getExitCode() == 0;
			} finally {
				process.close();
			}
		};
		if (type.getErasedType().isAssignableFrom(Integer.class) || type.getErasedType().isAssignableFrom(Integer.TYPE)) return (process, context) -> {
			try {
				return process.getExitCode();
			} finally {
				process.close();
			}
		};
		if (type.getErasedType().isAssignableFrom(Void.class) || type.getErasedType().isAssignableFrom(Void.TYPE)) return new IExplicitResultHandler() {
			@Override
			public Object apply(IProcess process, IResultContext context) {
				try {
					process.assertSuccess();
					return null;
				} finally {
					process.close();
				}
			}

			@Override
			public Redirects getRedirects() {
				return Redirects.builder().standardInput(InheritRedirect.create()).standardOutput(InheritRedirect.create()).standardError(InheritRedirect.create()).build();
			}
		};
		if (type.getErasedType().isAssignableFrom(String.class)) return (process, context) -> {
			try {
				final StringBuilder retVal = new StringBuilder();
				try (BufferedReader reader = new BufferedReader(new InputStreamReader(process.getStandardOutput()))) {
					String line;
					while ((line = reader.readLine()) != null) {
						retVal.append(line).append("\n");
					}
				} catch (IOException exception) {
					throw new RuntimeIOException(exception);
				}

				process.assertSuccess();
				return retVal.toString();
			} finally {
				process.close();
			}
		};
		if (new ATypeRef>() {}.getType().equals(type.getType())) return (process, context) -> {
			final BufferedReader reader = new BufferedReader(new InputStreamReader(process.getStandardOutput()));
			final Iterator iterator = new Iterator() {
				protected boolean done = false;

				protected String line = null;

				protected final CircularBuffer buffer = new CircularBuffer(10);

				protected void close() {
					done = true;
					if (!process.isSuccess()) {
						final List lines = buffer.getList();
						final StringBuilder builder = new StringBuilder().append("Showing last ").append(lines.size()).append(" lines:");
						for (String line : lines) {
							builder.append('\t').append(line).append('\n');
						}
						throw new RuntimeException(builder.toString());
					}
					try {
						reader.close();
					} catch (IOException e) {
						throw new RuntimeIOException(e);
					} finally {
						process.close();
					}
				}

				@Override
				public boolean hasNext() {
					if (done) return false;
					if (line == null) {
						try {
							line = reader.readLine();
							if (line == null) close();
							else buffer.add(line);
						} catch (IOException e) {
							throw new UncheckedIOException(e);
						}
					}
					return (line != null);
				}

				@Override
				public String next() {
					if (!hasNext()) throw new NoSuchElementException();
					final String retVal = line;
					line = null;
					return retVal;
				}
			};
			return StreamSupport.stream(Spliterators.spliteratorUnknownSize(iterator, Spliterator.ORDERED | Spliterator.NONNULL), false);
		};
		if (type.getErasedType().isAssignableFrom(IProcess.class)) return (process, context) -> process;
		throw new IllegalArgumentException(String.format("Return type \"%1$s\" is not supported!", type.getType()));
	}
}    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy