org.rzo.yajsw.os.posix.PosixService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of wrapper Show documentation
YAJSW is a java centric implementation of the java service wrapper by tanuki (JSW). It aims at being mostly configuration compliant with the original. It should therefore be easy to switch from JSW to YAJSW.
The newest version!
package org.rzo.yajsw.os.posix;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Map.Entry;
import java.util.logging.Logger;

import org.apache.velocity.Template;
import org.apache.velocity.VelocityContext;
import org.apache.velocity.app.VelocityEngine;
import org.rzo.yajsw.Constants;
import org.rzo.yajsw.boot.WrapperLoader;
import org.rzo.yajsw.config.YajswConfigurationImpl;
import org.rzo.yajsw.os.AbstractService;
import org.rzo.yajsw.os.JavaHome;
import org.rzo.yajsw.os.OperatingSystem;
import org.rzo.yajsw.os.posix.PosixProcess.CLibrary;
import org.rzo.yajsw.util.Utils;

public class PosixService extends AbstractService implements Constants
{
	protected String	_daemonDir;
	String				_daemonTemplate;
	String				_daemonScript;
	String				_wrapperPidFile;
	String				_appPidFile;
	String				_pidDir;
	String				_confFile;
	int					_stopTimeout;

	String[]			_startCmd;
	String[]			_execCmd;
	String[]			_stopCmd;
	String[]			_statusCmd;

	List		_ksLinks	= new ArrayList();
	UpdateRcParser		_updateRcParser;
	PosixUtils				_utils		= new PosixUtils();

	@Override
	public void setLogger(Logger logger)
	{
		super.setLogger(logger);
		_utils.setLog(logger);
	}
	

	public void init()
	{
		if (_name == null)
		{
			_logger.warning("no name for daemon -> abort");
			return;
		}
		_daemonDir = _config.getString("wrapper.daemon.dir", getDefaultDaemonDir());
		File daemonDir = new File(_daemonDir);
		if (!daemonDir.exists() || !daemonDir.isDirectory())
		{
			_logger.warning("Error " + _daemonDir + " : is not a directory");
			return;
		}
		_pidDir = _config.getString("wrapper.daemon.pid.dir", Constants.DEFAULT_DAEMON_PID_DIR);
		File pidDir = new File(_pidDir);
		if (!pidDir.exists() || !pidDir.isDirectory())
		{
			_logger.warning("Error " + _pidDir + " : is not a directory");
			return;
		}
		String wrapperJar = WrapperLoader.getWrapperJar().trim();
		String wrapperHome = ".";
		try
		{
			wrapperHome = new File(wrapperJar).getParentFile().getCanonicalPath();
		}
		catch (IOException e1)
		{
			e1.printStackTrace();
		}

		String confFile = _config.getString("wrapper.config");
		String confDir = null;
		if (confFile != null)
		{
			File f = new File(confFile);
			if (f.exists())
				try
				{
					confDir = f.getParentFile().getCanonicalPath();
				}
				catch (IOException e)
				{
				}
		}
		if (confDir == null)
			confDir = wrapperHome + "/conf";
		if (confFile == null)
		{
			_logger.warning("no conf file found -> abort");
			return;
		}
		try
		{
			_confFile = new File(confFile).getCanonicalPath();
		}
		catch (IOException e)
		{
			_logger.throwing(this.getClass().getName(), "init", e);
		}

    // Find working java
    JavaHome javaHome = OperatingSystem.instance().getJavaHome( _config );
    String java = javaHome.findJava( null, null );

    boolean bDone = false;
    File fJava = ((java == null) ? null : new File( java ));

    // Create symlink to Java Home for CustomProcName if necessary
    String custProcName = _config.getString( "wrapper.java.customProcName" );
    if ( custProcName != null && custProcName.length() > 0 )
    {
      File fcustom = new File( custProcName );
      bDone = false;

      while ( !bDone )
      {
        // If customProcName File/Symlink already exists - continue, do nothing...
        if ( fcustom.exists() && fcustom.canExecute() )
        {
          _config.setProperty( "wrapper.java.custProcName", fcustom.getAbsolutePath() );
          bDone = true;
          break;
        }
        else
        {
          // Find Path to Regular Java
          if ( fJava == null )
          {
            // not found - exit with tears
            bDone = true;
            break;
          }

          // is CustProcName Complete Path
          if ( fcustom.isAbsolute() && fcustom.getParentFile().exists() )
          {
            // Create Symlink for java at the specified location.
            if ( CLibrary.INSTANCE.symlink( fJava.getAbsolutePath(), fcustom.getAbsolutePath() ) != 0 )
            {
              if ( _logger != null )
                _logger.info( "error on creating script link " + fcustom );
            }
          }
          else
          {
            // create Symlink in tmp Directory
            File tmpPath = new File( _config.getString( "wrapper.tmp.path", "/tmp" ) );
            if ( tmpPath.exists() )
            {
              File link = new File( tmpPath, fcustom.getName() );
              if ( CLibrary.INSTANCE.symlink( fJava.getAbsolutePath(), link.getAbsolutePath() ) != 0 )
              {
                if ( _logger != null )
                  _logger.info( "error on creating script link " + link );
              }
              else
              {
                _config.setProperty( "wrapper.java.custProcName", link.getAbsolutePath() );
              }

            }
            else
            {
              
              if ( _logger != null )
                _logger.info( "error on creating customProcName link - temp directory [" + tmpPath.getAbsolutePath()
                    + "] does not exist" );
            }

          }
        }

        bDone = true; // do not want to loop
      } // while( !bDone )
      
		}
    
    
		_daemonTemplate = _config.getString("wrapper.daemon.template", wrapperHome + "/templates/daemon.vm");
		File daemonTemplate = new File(_daemonTemplate);
		if (!daemonTemplate.exists() || !daemonTemplate.isFile())
		{
			if (_logger != null)
				_logger.warning("Error " + _daemonTemplate + " : template file not found");
			return;
		}
		File daemonScript = getDaemonScript();
		if (daemonScript.exists())
			if (_logger != null)
				_logger.info(daemonScript.getAbsolutePath() + " already exists");
		String pidName = null;
		try
		{
			pidName = _config.getString("wrapper.pidfile", new File(pidDir, "wrapper." + getName() + ".pid").getCanonicalPath());
		}
		catch (IOException e)
		{
			if (_logger != null)
				_logger.throwing(this.getClass().getName(), "init", e);
		}
		File pidFile = new File(pidName);
		String apidName = null;
		try
		{
			apidName = _config.getString("wrapper.java.pidfile", new File(pidDir, "wrapper.java." + getName() + ".pid").getCanonicalPath());
		}
		catch (IOException e)
		{
			if (_logger != null)
				_logger.throwing(this.getClass().getName(), "init", e);
		}
		File apidFile = new File(apidName);
		try
		{
			_daemonTemplate = daemonTemplate.getCanonicalPath();
			_wrapperPidFile = pidFile.getCanonicalPath();
			_appPidFile = apidFile.getCanonicalPath();
			_daemonScript = daemonScript.getCanonicalPath();
		}
		catch (Exception ex)
		{
			if (_logger != null)
				_logger.throwing(this.getClass().getName(), "init", ex);
		}
		//JavaHome javaHome = OperatingSystem.instance().getJavaHome(_config);
    //String java = javaHome.findJava(_config.getString("wrapper.ntservice.java.command"), _config.getString("wrapper.java.customProcName"));
    java = javaHome.findJava(_config.getString("wrapper.ntservice.java.command"), _config.getString("wrapper.java.customProcName"));
		String tmpDir = _config.getString("wrapper.tmp.path", System.getProperty("jna_tmpdir", null));
		ArrayList result = new ArrayList();
		String opt = null;
		if (tmpDir != null)
		{
			opt = Utils.getDOption("jna_tmpdir", tmpDir);
			result.add(opt);
		}
		YajswConfigurationImpl config = (YajswConfigurationImpl) _config;
		for (Iterator it=config.subset("wrapper").getKeys(); it.hasNext(); )
		try
		{
			config.getProperty((String) it.next());
		}
		catch (Exception ex)
		{
			
		}
		// first add lookup vars eg ${lookup}
		if (_config.getBoolean("wrapper.save_interpolated", true))
		{
		
		for (Entry e : config.getEnvLookupSet().entrySet())
		{
			opt = Utils.getDOption(e.getKey(),e.getValue());
			if (!result.contains(opt))
			   result.add(opt);
		}
		}
		
		for (Iterator it = _config.getKeys("wrapper.ntservice.additional"); it.hasNext();)
		{
			String key = (String) it.next();
			String value = _config.getString(key);
			result.add(value);
		}
		List configs = _config.getList("wrapperx.config");
		
		_startCmd = new String[10+result.size()+configs.size()];
		_startCmd[0] = java;
		_startCmd[1] = Utils.getDOption("wrapper.pidfile", _wrapperPidFile);
		_startCmd[2] = "-Dwrapper.service=true";
		_startCmd[3] = "-Dwrapper.visible=false";
		for (int i=0; i abort");
			return false;
		}
		try
		{
			File daemonTemplate = new File(_daemonTemplate);
			VelocityEngine ve = new VelocityEngine();
			ve.setProperty(VelocityEngine.RESOURCE_LOADER, "file");
			ve.setProperty("file.resource.loader.path", daemonTemplate.getParent());
			ve.setProperty("runtime.log.logsystem.class", VelocityLog.class.getCanonicalName());
			// TODO find a way to set a non static logger
			VelocityLog.setLogger(_logger);
			ve.init();
			Template t = ve.getTemplate(daemonTemplate.getName());
			VelocityContext context = new VelocityContext();
			context.put("w_name", _name);
			context.put("w_long_name", _displayName);
			context.put("w_start_cmd", toStrCommand(_startCmd));
			context.put("w_stop_cmd", toStrCommand(_stopCmd));
			context.put("w_status_cmd", toStrCommand(_statusCmd));
			context.put("w_description", _description);
			context.put("w_conf_file", _confFile);
			context.put("w_app_pid_file", _appPidFile);
			context.put("w_wrapper_pid_file", _wrapperPidFile);
			context.put("w_start_levels", _updateRcParser.getStartLevels());
			context.put("w_stop_levels", _updateRcParser.getStopLevels());
			context.put("start_dependencies", getStartDependencies());
			context.put("stop_dependencies", getInternalStopDependencies());
			FileWriter writer = new FileWriter(_daemonScript);

			t.merge(context, writer);
			writer.flush();
			writer.close();
			File daemonScript = new File(_daemonScript);
			if (daemonScript.exists())
			{
				if (_logger != null)
					_logger.warning("created daemon script: " + _daemonScript);
			}
			else
			{
				if (_logger != null)
					_logger.warning("error creating daemon script: " + _daemonScript);
			}
			// only jdk 1.6 daemonScript.setExecutable(true);
			Runtime.getRuntime().exec("chmod 755 " + _daemonScript);
			if ("AUTO_START".equals(_config.getString("wrapper.ntservice.starttype", DEFAULT_SERVICE_START_TYPE))
					|| "DELAYED_AUTO_START".equals(_config.getString("wrapper.ntservice.starttype", DEFAULT_SERVICE_START_TYPE)))
			{

				for (String link : _ksLinks)
					if (CLibrary.INSTANCE.symlink(_daemonScript, link) != 0)
					{
						if (_logger != null)
							_logger.info("error on creating script link " + link);
					}
					else
					{
						if (new File(link).exists())
						{
							if (_logger != null)
								_logger.info("created link : " + link);
						}
						else
						{
							if (_logger != null)
								_logger.info("error on creating script link " + link);
						}
					}

				// Runtime.getRuntime().exec(String.format("ln -s %1$s %2$s",
				// _klink, _daemonScript));
				// Runtime.getRuntime().exec(String.format("ln -s %1$s %2$s",
				// _slink, _daemonScript));
			}

		}
		catch (Exception ex)
		{
			if (_logger != null)
				_logger.throwing(this.getClass().getName(), "install", ex);
			return false;
		}
		return true;
	}

	private String getInternalStopDependencies()
	{
		String result = "";
		String[] dependencies = super.getStopDependencies();
		if (dependencies != null)
		for (String dep : dependencies)
			result = result + dep + " ";
		return result;
	}


	private String getStartDependencies()
	{
		String result = "";
		String[] dependencies = super.getDependencies();
		if (dependencies != null)
		for (String dep : dependencies)
			result = result + dep + " ";
		return result;
	}


	private String toStrCommand(String[] cmd)
	{
		StringBuffer  tmp = new StringBuffer();
		for (String s : cmd)
		{
			if (s != null && s.contains(" ") && !s.contains("\""))
			{
				tmp.append("\"");
				tmp.append(s);
				tmp.append("\"");
			}
			else if (s != null)
			{
				tmp.append(s);
			}
			if (s != null && s != cmd[cmd.length-1])
				tmp.append(" ");

		}
		return tmp.toString();
	}

	public boolean isInstalled()
	{
		return _daemonScript != null && new File(_daemonScript).exists();
	}

	public boolean isRunning()
	{
		int pid = getPid();
		if (pid < 0)
			return false;
		org.rzo.yajsw.os.Process p = OperatingSystem.instance().processManagerInstance().getProcess(pid);
		return p != null && p.isRunning();
	}

	public boolean start()
	{
		if (isRunning())
		{
			if (_logger != null)
				_logger.info("already running");
			return true;
		}
		File f = new File(_daemonScript);
		if (!OperatingSystem.instance().setWorkingDir(f.getParent()))
		{
			if (_logger != null)
				_logger.warning("could not set working dir. pls check configuration or user rights :"+f.getParent());
			else
				System.out.println("could not set working dir. pls check configuration or user rights :"+f.getParent());
		}
		String txt = _utils.osCommand(_daemonScript + " start", 45000);
		if (_logger != null)
			_logger.info(txt);
		return isRunning();
	}

	public boolean stop()
	{
		if (_logger != null)
			_logger.info(_utils.osCommand(_daemonScript + " stop", _stopTimeout));
		return !isRunning();
	}

	public boolean startProcess()
	{
		if (isRunning())
		{
			_logger.info("already running");
			return true;
		}
		/*
		 * org.rzo.yajsw.os.Process p =
		 * OperatingSystem.instance().processManagerInstance().createProcess();
		 * p.setCommand(cleanCmd(_execCmd)); p.setVisible(false);
		 * p.setPipeStreams(false, false);
		 * p.setDebug(_config.getBoolean("wrapper.debug", false)); p.start();
		 */
		try
		{
			if (_logger != null)
				_logger.info("calling " + toStrCommand(_execCmd));
			Runtime.getRuntime().exec(cleanCmd(_execCmd));
		}
		catch (Exception e)
		{
			if (_logger != null)
				_logger.throwing(this.getClass().getName(), "startProcess", e);

		}
		return true; // p.isRunning();
	}

	public boolean stopProcess()
	{
		int pid = getPid();
		if (_logger != null)
			_logger.info("stop daemon with pid " + pid);
		if (pid <= 0)
			return false;
		org.rzo.yajsw.os.Process p = OperatingSystem.instance().processManagerInstance().getProcess(pid);
		if (p == null)
		{
			if (_logger != null)
				_logger.info("process not running");
			return true;
		}

		p.stop(_stopTimeout, 0);
		int apid = getAppPid();
		if (_logger != null)
			_logger.info("stop daemon app with pid " + apid);
		if (apid <= 0)
			return false;
		org.rzo.yajsw.os.Process ap = OperatingSystem.instance().processManagerInstance().getProcess(apid);

		if (ap != null)
			ap.kill(999);
		return true;
	}

	public boolean uninstall()
	{
		if (isRunning())
			stop();
		new File(_daemonScript).delete();
		for (String link : _ksLinks)
			new File(link).delete();
		return true;
	}

	public int state()
	{
		int result = 0;
		if (new File(_daemonScript).exists())
			result |= STATE_INSTALLED;
		if (isRunning())
			result |= STATE_RUNNING;
		return result;
	}

	public int getPid()
	{
if ( _logger != null )
		_logger.info("wrapper pid file: " + _wrapperPidFile);
		if (_wrapperPidFile != null && new File(_wrapperPidFile).exists())
			try
			{
				BufferedReader reader = new BufferedReader(new FileReader(_wrapperPidFile));
				String pid = reader.readLine();
				reader.close();
				return Integer.parseInt(pid);
			}
			catch (Exception e)
			{
				if (_logger != null)
					_logger.throwing(this.getClass().getName(), "getPid", e);
			}
		return -1;

	}

	public int getAppPid()
	{
		_logger.info("app pid file: " + _appPidFile);
		if (_appPidFile != null && new File(_appPidFile).exists())
			try
			{
				BufferedReader reader = new BufferedReader(new FileReader(_appPidFile));
				String pid = reader.readLine();
				reader.close();
				return Integer.parseInt(pid);
			}
			catch (Exception e)
			{
				if (_logger != null)
					_logger.throwing(this.getClass().getName(), "getAppPid", e);
			}
		return -1;

	}

	public String[] cleanCmd(String[] cmd)
	{
		List cmdl = new ArrayList();
		for (String s : cmd)
		{
			if (s != null && !"".equals(s))
				cmdl.add(s);
		}
		String[] result = new String[cmdl.size()];
		int i = 0;
		for (String s : cmdl)
			result[i++] = s;
		return result;

	}
	

}