Skip to content

Content of file DefaultProcessExecutor.java.html

<?xml version="1.0" encoding="UTF-8"?><!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd"><html xmlns="http://www.w3.org/1999/xhtml" lang="en"><head><meta http-equiv="Content-Type" content="text/html;charset=UTF-8"/><link rel="stylesheet" href="../jacoco-resources/report.css" type="text/css"/><link rel="shortcut icon" href="../jacoco-resources/report.gif" type="image/gif"/><title>DefaultProcessExecutor.java</title><link rel="stylesheet" href="../jacoco-resources/prettify.css" type="text/css"/><script type="text/javascript" src="../jacoco-resources/prettify.js"></script></head><body onload="window['PR_TAB_WIDTH']=4;prettyPrint()"><div class="breadcrumb" id="breadcrumb"><span class="info"><a href="../jacoco-sessions.html" class="el_session">Sessions</a></span><a href="../index.html" class="el_report">TheseFoolishThings :: Utilities</a> &gt; <a href="index.source.html" class="el_package">it.tidalwave.util.spi</a> &gt; <span class="el_source">DefaultProcessExecutor.java</span></div><h1>DefaultProcessExecutor.java</h1><pre class="source lang-java linenums">/*
 * *********************************************************************************************************************
 *
 * TheseFoolishThings: Miscellaneous utilities
 * http://tidalwave.it/projects/thesefoolishthings
 *
 * Copyright (C) 2009 - 2021 by Tidalwave s.a.s. (http://tidalwave.it)
 *
 * *********************************************************************************************************************
 *
 * Licensed under the Apache License, Version 2.0 (the &quot;License&quot;); 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 &quot;AS IS&quot; 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.
 *
 * *********************************************************************************************************************
 *
 * git clone https://bitbucket.org/tidalwave/thesefoolishthings-src
 * git clone https://github.com/tidalwave-it/thesefoolishthings-src
 *
 * *********************************************************************************************************************
 */
package it.tidalwave.util.spi;

import javax.annotation.CheckForNull;
import javax.annotation.Nonnull;
import javax.annotation.concurrent.ThreadSafe;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Map.Entry;
import java.util.Scanner;
import java.util.concurrent.Executors;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import it.tidalwave.util.ProcessExecutor;
import lombok.AccessLevel;
import lombok.Cleanup;
import lombok.Getter;
import lombok.NoArgsConstructor;
import lombok.RequiredArgsConstructor;
import lombok.Setter;
import lombok.extern.slf4j.Slf4j;

/***********************************************************************************************************************
 *
 * @author  Fabrizio Giudici
 * @since   1.39
 *
 **********************************************************************************************************************/
<span class="nc" id="L63">@ThreadSafe @NoArgsConstructor(access=AccessLevel.PRIVATE) @Slf4j</span>
public class DefaultProcessExecutor implements ProcessExecutor
  {
<span class="nc" id="L66">    private final ExecutorService executorService = Executors.newFixedThreadPool(10);</span>

    /*******************************************************************************************************************
     *
     *
     ******************************************************************************************************************/
<span class="nc bnc" id="L72" title="All 4 branches missed.">    @RequiredArgsConstructor(access=AccessLevel.PACKAGE)</span>
    public class DefaultConsoleOutput implements ConsoleOutput
      {
        @Nonnull
        private final String name;

        @Nonnull
        private final InputStream input;

<span class="nc" id="L81">        @Getter</span>
<span class="nc" id="L82">        private final List&lt;String&gt; content = Collections.synchronizedList(new ArrayList&lt;&gt;());</span>

        private volatile String latestLine;

<span class="nc" id="L86">        private volatile int li = 0;</span>

<span class="nc" id="L88">        private final AtomicBoolean started = new AtomicBoolean();</span>
        
<span class="nc" id="L90">        @CheckForNull @Setter @Getter</span>
        private Listener listener;

        /***************************************************************************************************************
         *
         *
         ***************************************************************************************************************/
<span class="nc" id="L97">        private final Runnable reader = new Runnable()</span>
<span class="nc" id="L98">          {</span>
            @Override
            public void run()
              {
                try
                  {
<span class="nc" id="L104">                    read();</span>
                  }
<span class="nc" id="L106">                catch (IOException e)</span>
                  {
<span class="nc" id="L108">                    log.warn(&quot;while reading from &quot; + name, e);</span>
<span class="nc" id="L109">                  }</span>
<span class="nc" id="L110">              }</span>
          };

        /***************************************************************************************************************
         *
         *
         ***************************************************************************************************************/
<span class="nc" id="L117">        private final Runnable logger = new Runnable()</span>
<span class="nc" id="L118">          {</span>
            @Override
            public void run()
              {
<span class="nc" id="L122">                int l = 0;</span>

                for (;;)
                  {
                    try
                      {
<span class="nc bnc" id="L128" title="All 4 branches missed.">                        if ((l != li) &amp;&amp; (latestLine != null))</span>
                          {
<span class="nc" id="L130">                            log.trace(&quot;&gt;&gt;&gt;&gt;&gt;&gt;&gt;&gt; {} {}&quot;, name, latestLine);</span>
                          }

<span class="nc" id="L133">                        l = li;</span>
<span class="nc" id="L134">                        Thread.sleep(500);</span>
                      }
<span class="nc" id="L136">                    catch (Throwable e)</span>
                      {
<span class="nc" id="L138">                        return;</span>
<span class="nc" id="L139">                      }</span>
                  }
              }
          };

        /***************************************************************************************************************
         *
         * Should not be used by the programmer.
         *
         * @return -
         *
         ***************************************************************************************************************/
        @Nonnull
        public ConsoleOutput start()
          {
<span class="nc bnc" id="L154" title="All 2 branches missed.">            if (started.getAndSet(true))</span>
              {
<span class="nc" id="L156">                throw new IllegalStateException(&quot;Already started&quot;);</span>
              }

<span class="nc" id="L159">            log.info(&quot;{} - started&quot;, name);</span>
<span class="nc" id="L160">            executorService.submit(reader);</span>
<span class="nc" id="L161">            executorService.submit(logger);</span>
<span class="nc" id="L162">            return this;</span>
          }

        /***************************************************************************************************************
         *
         * {@inheritDoc}
         *
         ***************************************************************************************************************/
        @Override
        public boolean latestLineMatches (@Nonnull final String regexp)
          {
<span class="nc" id="L173">            String s = null;</span>

<span class="nc bnc" id="L175" title="All 2 branches missed.">            if (latestLine != null)</span>
              {
<span class="nc" id="L177">                s = latestLine;</span>
              }
<span class="nc bnc" id="L179" title="All 2 branches missed.">            else if (!content.isEmpty())</span>
              {
<span class="nc" id="L181">                s = content.get(content.size() - 1);</span>
              }

<span class="nc" id="L184">            log.trace(&quot;&gt;&gt;&gt;&gt; testing '{}' for '{}'&quot;, s, regexp);</span>
<span class="nc bnc" id="L185" title="All 4 branches missed.">            return (s != null) &amp;&amp; Pattern.compile(regexp).matcher(s).matches();</span>
            // FIXME: sync
          }

        /***************************************************************************************************************
         *
         * {@inheritDoc}
         *
         ***************************************************************************************************************/
        @Override @Nonnull
        public Scanner filteredAndSplitBy (@Nonnull final String filterRegexp, @Nonnull final String delimiterRegexp)
          {
<span class="nc" id="L197">            final String string = filteredBy(filterRegexp).get(0);</span>
<span class="nc" id="L198">            return new Scanner(string).useDelimiter(Pattern.compile(delimiterRegexp));</span>
          }

        /***************************************************************************************************************
         *
         * {@inheritDoc}
         *
         ***************************************************************************************************************/
        @Override @Nonnull
        public List&lt;String&gt; filteredBy (@Nonnull final String regexp)
          {
<span class="nc" id="L209">            final Pattern pattern = Pattern.compile(regexp);</span>
<span class="nc" id="L210">            final List&lt;String&gt; result = new ArrayList&lt;&gt;();</span>
<span class="nc" id="L211">            final ArrayList&lt;String&gt; strings = new ArrayList&lt;&gt;(content);</span>

            // TODO: sync
<span class="nc bnc" id="L214" title="All 2 branches missed.">            if (latestLine != null)</span>
              {
<span class="nc" id="L216">                strings.add(latestLine);</span>
              }

<span class="nc bnc" id="L219" title="All 2 branches missed.">            for (final String s : strings)</span>
              {
//                log.trace(&quot;&gt;&gt;&gt;&gt;&gt;&gt;&gt;&gt; matching '{}' with '{}'...&quot;, s, filter);
<span class="nc" id="L222">                final Matcher m = pattern.matcher(s);</span>

<span class="nc bnc" id="L224" title="All 2 branches missed.">                if (m.matches())</span>
                  {
<span class="nc" id="L226">                    result.add(m.group(1));</span>
                  }
<span class="nc" id="L228">              }</span>

<span class="nc" id="L230">            return result;</span>
          }

        /***************************************************************************************************************
         *
         * {@inheritDoc}
         *
         ***************************************************************************************************************/
        @Override @Nonnull
        public ConsoleOutput waitFor (@Nonnull final String regexp)
          throws InterruptedException, IOException
          {
<span class="nc" id="L242">            log.debug(&quot;{} - waitFor({})&quot;, name, regexp);</span>

<span class="nc bnc" id="L244" title="All 2 branches missed.">            while (filteredBy(regexp).isEmpty())</span>
              {
                try
                  {
<span class="nc" id="L248">                    final int exitValue = process.exitValue();</span>
<span class="nc" id="L249">                    throw new IOException(&quot;Process exited with &quot; + exitValue);</span>
                  }
<span class="nc" id="L251">                catch (IllegalThreadStateException e) // ok, process not terminated yet</span>
                  {
<span class="nc" id="L253">                    synchronized (this)</span>
                      {
<span class="nc" id="L255">                        wait(50); // FIXME: polls because it doesn't get notified</span>
polls because it doesn't get notified
<span class="nc" id="L256"> }</span> <span class="nc" id="L257"> }</span> } <span class="nc" id="L260"> return this;</span> } /*************************************************************************************************************** * * {@inheritDoc} * ***************************************************************************************************************/ @Override public void clear() { <span class="nc" id="L271"> content.clear();</span> <span class="nc" id="L272"> latestLine = null;</span> <span class="nc" id="L273"> }</span> /*************************************************************************************************************** * * ***************************************************************************************************************/ private void read() throws IOException { <span class="nc bnc" id="L282" title="All 2 branches missed."> @Cleanup final InputStreamReader is = new InputStreamReader(input);</span> <span class="nc" id="L283"> StringBuilder l = new StringBuilder();</span> for (;;) { <span class="nc" id="L287"> final int c = is.read();</span> <span class="nc bnc" id="L289" title="All 2 branches missed."> if (c &lt; 0)</span> { <span class="nc" id="L291"> break;</span> } // if (c == 10) // { // continue; // } <span class="nc bnc" id="L299" title="All 4 branches missed."> if ((c == 13) || (c == 10))</span> { <span class="nc" id="L301"> latestLine = l.toString();</span> <span class="nc" id="L302"> li++;</span> <span class="nc" id="L303"> content.add(latestLine);</span> <span class="nc" id="L304"> l = new StringBuilder();</span> <span class="nc" id="L305"> log.trace(&quot;&gt;&gt;&gt;&gt;&gt;&gt;&gt;&gt; {} {}&quot;, name, latestLine);</span> <span class="nc bnc" id="L307" title="All 2 branches missed."> if (listener != null)</span> { <span class="nc" id="L309"> listener.onReceived(latestLine);</span> } } else { <span class="nc" id="L314"> l.append((char)c);</span> <span class="nc" id="L315"> latestLine = l.toString();</span> <span class="nc" id="L316"> li++;</span> } <span class="nc" id="L319"> synchronized (this)</span> { <span class="nc" id="L321"> notifyAll();</span> <span class="nc" id="L322"> }</span> <span class="nc" id="L323"> }</span> <span class="nc" id="L325"> log.debug(&quot;&gt;&gt;&gt;&gt;&gt;&gt; {} closed&quot;, name);</span> <span class="nc" id="L326"> is.close();</span> <span class="nc" id="L327"> }</span> } <span class="nc" id="L330"> private final List&lt;String&gt; arguments = new ArrayList&lt;&gt;();</span> private Process process; <span class="nc" id="L334"> @Getter</span> private ConsoleOutput stdout; <span class="nc" id="L337"> @Getter</span> private ConsoleOutput stderr; private PrintWriter stdin; /******************************************************************************************************************* * * Factory method for associating an executable. It returns an intermediate executor that must be configured and * later started. Under Windows, the '.exe' suffix is automatically appended to the name of the executable. * * @see #start() * * @param executable the executable (with the full path) * @return the executor * ******************************************************************************************************************/ @Nonnull public static DefaultProcessExecutor forExecutable (@Nonnull final String executable) { <span class="nc" id="L356"> final DefaultProcessExecutor executor = new DefaultProcessExecutor();</span> <span class="nc bnc" id="L357" title="All 2 branches missed."> executor.arguments.add(new File(executable + (isWindows() ? &quot;.exe&quot; : &quot;&quot;)).getAbsolutePath());</span> <span class="nc" id="L358"> return executor;</span> } // /******************************************************************************************************************* // * // * // ******************************************************************************************************************/ // @Nonnull // private static String findPath (final @Nonnull String executable) // throws NotFoundException // { // for (final String path : System.getenv(&quot;PATH&quot;).split(File.pathSeparator)) // { // final File file = new File(new File(path), executable); // // if (file.canExecute()) // { // return file.getAbsolutePath(); // } // } // // throw new NotFoundException(&quot;Can't find &quot; + executable + &quot; in PATH&quot;); // } /******************************************************************************************************************* * * {@inheritDoc} * ******************************************************************************************************************/ @Override @Nonnull public DefaultProcessExecutor withArgument (@Nonnull final String argument) { <span class="nc" id="L390"> arguments.add(argument);</span> <span class="nc" id="L391"> return this;</span> } /******************************************************************************************************************* * * {@inheritDoc} * ******************************************************************************************************************/ @Override @Nonnull public DefaultProcessExecutor withArguments (@Nonnull final String ... arguments) { <span class="nc" id="L402"> this.arguments.addAll(Arrays.asList(arguments));</span> <span class="nc" id="L403"> return this;</span> } /******************************************************************************************************************* * * {@inheritDoc} * ******************************************************************************************************************/ @Override @Nonnull public DefaultProcessExecutor start() throws IOException { <span class="nc" id="L415"> log.info(&quot;&gt;&gt;&gt;&gt; executing {} ...&quot;, arguments);</span> <span class="nc" id="L417"> final List&lt;String&gt; environment = new ArrayList&lt;&gt;();</span> <span class="nc bnc" id="L419" title="All 2 branches missed."> for (final Entry&lt;String, String&gt; e : System.getenv().entrySet())</span> { <span class="nc" id="L421"> environment.add(String.format(&quot;%s=%s&quot;, e.getKey(), e.getValue()));</span> <span class="nc" id="L422"> }</span> <span class="nc" id="L424"> log.info(&quot;&gt;&gt;&gt;&gt; environment: {}&quot;, environment);</span> <span class="nc" id="L425"> process = Runtime.getRuntime().exec(arguments.toArray(new String[0]),</span> <span class="nc" id="L426"> environment.toArray(new String[0]));</span> <span class="nc" id="L428"> stdout = new DefaultConsoleOutput(&quot;out&quot;, process.getInputStream()).start();</span> <span class="nc" id="L429"> stderr = new DefaultConsoleOutput(&quot;err&quot;, process.getErrorStream()).start();</span> <span class="nc" id="L430"> stdin = new PrintWriter(process.getOutputStream(), true);</span> <span class="nc" id="L432"> return this;</span> } /******************************************************************************************************************* * * {@inheritDoc} * ******************************************************************************************************************/ @Override public void stop() { <span class="nc" id="L443"> log.info(&quot;stop()&quot;);</span> <span class="nc" id="L444"> process.destroy();</span> <span class="nc" id="L445"> executorService.shutdownNow();</span> <span class="nc" id="L446"> }</span> /******************************************************************************************************************* * * {@inheritDoc} * ******************************************************************************************************************/ @Override @Nonnull public DefaultProcessExecutor waitForCompletion() throws IOException, InterruptedException { <span class="nc bnc" id="L457" title="All 2 branches missed."> if (process.waitFor() != 0)</span> { // throw new IOException(&quot;Process exited with &quot; + process.exitValue()); FIXME } <span class="nc" id="L462"> return this;</span> } /******************************************************************************************************************* * * {@inheritDoc} * ******************************************************************************************************************/ @Override @Nonnull public DefaultProcessExecutor send (@Nonnull final String string) { <span class="nc" id="L473"> log.debug(&quot;&gt;&gt;&gt;&gt; sending '{}'...&quot;, string.replaceAll(&quot;\n&quot;, &quot;&lt;CR&gt;&quot;));</span> <span class="nc" id="L474"> stdin.print(string);</span> <span class="nc" id="L475"> stdin.flush();</span> <span class="nc" id="L476"> return this;</span> } /******************************************************************************************************************* * * ******************************************************************************************************************/ private static boolean isWindows() { <span class="nc" id="L485"> return System.getProperty (&quot;os.name&quot;).toLowerCase().startsWith(&quot;windows&quot;);</span> } } </pre><div class="footer"><span class="right">Created with <a href="http://www.jacoco.org/jacoco">JaCoCo</a> 0.8.6.202009150832</span></div></body></html>