读取tomcat jndi配置

具体看代码的resolveJNDIProps方法。首先在tomcatd的config/context.xml里添加代码

<?xml version='1.0' encoding='utf-8'?>
<!--
Licensed to the Apache Software Foundation (ASF) under one or more
contributor license agreements. See the NOTICE file distributed with
this work for additional information regarding copyright ownership.
The ASF licenses this file to You under the Apache License, Version 2.0
(the "License"); 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 "AS IS" 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.
-->
<!-- The contents of this file will be loaded for each web application -->
<Context>

<!-- Default set of monitored resources -->
<WatchedResource>WEB-INF/web.xml</WatchedResource>
<Environment name="redis-0-port" override="false" type="java.lang.Integer" value="6379"/>
<Environment name="redis-0-user" override="false" type="java.lang.String" value="root"/>
<!-- Uncomment this to disable session persistence across Tomcat restarts -->
<!--
<Manager pathname="" />
-->

<!-- Uncomment this to enable Comet connection tacking (provides events
on session expiration as well as webapp lifecycle) -->
<!--
<Valve className="org.apache.catalina.valves.CometConnectionManagerValve" />
-->

</Context>

 具体分析resolveJNDIProps方法,代码如下:

/*
 * Copyright 2010 LinkedIn, Inc
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); 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 "AS IS" 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.
 */

package azkaban.jobs.builtin;

import azkaban.app.JobDescriptor;
import azkaban.common.jobs.Job;
import azkaban.common.utils.Props;
import azkaban.jobs.AbstractProcessJob;

import org.apache.log4j.Level;
import org.apache.log4j.Logger;

import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.InputStreamReader;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;

import javax.naming.InitialContext;
import javax.naming.NameClassPair;
import javax.naming.NamingEnumeration;
import javax.naming.NamingException;

/**
 * A job that runs a simple unix command
 * 
 * @author jkreps
 * 
 */
public class ProcessJob extends AbstractProcessJob implements Job {
	private static final Logger log = Logger.getLogger(ProcessJob.class);
    public static final String COMMAND = "command";
    public static final int CLEAN_UP_TIME_MS = 1000;
    
    private volatile Process _process;
    private volatile boolean _isComplete;
    
    public ProcessJob(JobDescriptor descriptor) {
        super(descriptor);
    }
    
    public void resolveJNDIProps(){
    	log.info("*****************************************");
    	log.info("resolve JNDI props from InitialContext");
    	log.info("*****************************************");
    	//read all settings from jndi context 
		InitialContext ic;
		try {
			ic = new InitialContext();
	        NamingEnumeration<NameClassPair> em = ic.list("java:/comp/env");
			while(em.hasMoreElements()){
				NameClassPair nPair = em.nextElement();
				String sClass = nPair.getClassName();
				String sName = nPair.getName();
				Object value = ic.lookup("java:/comp/env/"+sName);
				if(sClass.contains("String")||sClass.contains("Integer")||sClass.contains("Long")||sClass.contains("Double")){
					this._env.put("jndi."+sName, value.toString());
					
					log.info("resolve jndi prop. name=" + sName + ", value="
							+ value);
				}else{
					log.info("skip jndi prop. name=" + sName + ", value="
							+ value);
				}
				
			}
			this._env.put("jndi.resolved", Boolean.TRUE.toString());
		} catch (NamingException e) {
			// ignored 
			log.error("exception at resolve all jndi props.", e);
			this._env.put("jndi.resolved", e.getClass().getCanonicalName());
		}
    }

    public void run() {
        
        resolveProps();
        resolveJNDIProps();
        // Sets a list of all the commands that need to be run.
        List<String> commands = getCommandList();
        info(commands.size() + " commands to execute.");

        File[] propFiles =initPropsFiles();
        
        //System.err.println("in process job outputFile=" +propFiles[1]);
 
        // For each of the jobs, set up a process and run them.
        for(String command: commands) {
            info("Executing command: " + command);
            String[] cmdPieces = partitionCommandLine(command);

            ProcessBuilder builder = new ProcessBuilder(cmdPieces);

            builder.directory(new File(getCwd()));
            builder.environment().putAll(getEnv());

            try {
                _process = builder.start();
            } catch(IOException e) {
                for (File file: propFiles)   if (file != null && file.exists()) file.delete();
                throw new RuntimeException(e);
            }
            Thread outputGobbler = new LoggingGobbler(new InputStreamReader(_process.getInputStream()),
                                                      Level.INFO);
            Thread errorGobbler = new LoggingGobbler(new InputStreamReader(_process.getErrorStream()),
                                                     Level.ERROR);

            int processId = getProcessId();
            if(processId == 0) {
                info("Spawned thread. Unknowned processId");
            } else {
                info("Spawned thread with processId " + processId);
            }
            outputGobbler.start();
            errorGobbler.start();
            int exitCode = -999;
            try {
                exitCode = _process.waitFor();
            } catch(InterruptedException e) {
            }

            _isComplete = true;
            if(exitCode != 0) {
                for (File file: propFiles)   if (file != null && file.exists()) file.delete();
                throw new RuntimeException("Processes ended with exit code " + exitCode + ".");
            }

            // try to wait for everything to get logged out before exiting
            try {
                outputGobbler.join(1000);
                errorGobbler.join(1000);
            } catch(InterruptedException e) {
            }
        }
        
        // Get the output properties from this job.
       generateProperties(propFiles[1]);
               
       for (File file: propFiles)
           if (file != null && file.exists()) file.delete();
    
    }
       

    protected List<String> getCommandList() {
        List<String> commands = new ArrayList<String>();
        commands.add(_props.getString(COMMAND));
        for(int i = 1; _props.containsKey(COMMAND + "." + i); i++)
            commands.add(_props.getString(COMMAND + "." + i));

        return commands;
    }


   

    @Override
    public void cancel() throws Exception {
        if(_process != null) {
            int processId = getProcessId();
            if(processId != 0) {
                warn("Attempting to kill the process " + processId);
                try {
                    Runtime.getRuntime().exec("kill " + processId);
                    synchronized(this) {
                        wait(CLEAN_UP_TIME_MS);
                    }
                } catch(InterruptedException e) {
                    // Do nothing. We don't really care.
                }
                if(!_isComplete) {
                    error("After " + CLEAN_UP_TIME_MS
                          + " ms, the job hasn't terminated. Will force terminate the job.");
                }
            } else {
                info("Cound not get process id");
            }

            if(!_isComplete) {
                warn("Force kill the process");
                _process.destroy();
            }
        }
    }

    public int getProcessId() {
        int processId = 0;

        try {
            Field f = _process.getClass().getDeclaredField("pid");
            f.setAccessible(true);

            processId = f.getInt(_process);
        } catch(Throwable e) {}

        return processId;
    }

    @Override
    public double getProgress() {
        return _isComplete ? 1.0 : 0.0;
    }

    private class LoggingGobbler extends Thread {

        private final BufferedReader _inputReader;
        private final Level _loggingLevel;

        public LoggingGobbler(InputStreamReader inputReader, Level level) {
            _inputReader = new BufferedReader(inputReader);
            _loggingLevel = level;
        }

        @Override
        public void run() {
            try {
                while(!Thread.currentThread().isInterrupted()) {
                    String line = _inputReader.readLine();
                    if(line == null)
                        return;

                    logMessage(line);
                }
            } catch(IOException e) {
                error("Error reading from logging stream:", e);
            }
        }

        private void logMessage(String message) {
            if(message.startsWith(Level.DEBUG.toString())) {
                String newMsg = message.substring(Level.DEBUG.toString().length());
                getLog().debug(newMsg);
            } else if(message.startsWith(Level.ERROR.toString())) {
                String newMsg = message.substring(Level.ERROR.toString().length());
                getLog().error(newMsg);
            } else if(message.startsWith(Level.INFO.toString())) {
                String newMsg = message.substring(Level.INFO.toString().length());
                getLog().info(newMsg);
            } else if(message.startsWith(Level.WARN.toString())) {
                String newMsg = message.substring(Level.WARN.toString().length());
                getLog().warn(newMsg);
            } else if(message.startsWith(Level.FATAL.toString())) {
                String newMsg = message.substring(Level.FATAL.toString().length());
                getLog().fatal(newMsg);
            } else if(message.startsWith(Level.TRACE.toString())) {
                String newMsg = message.substring(Level.TRACE.toString().length());
                getLog().trace(newMsg);
            } else {
                getLog().log(_loggingLevel, message);
            }

        }
    }

    public Props getProps() {
        return _props;
    }

    public String getPath() {
        return _jobPath;
    }

    public String getJobName() {
        return getId();
    }

    public JobDescriptor getJobDescriptor() {
        return _descriptor;
    }

    /**
     * Splits the command into a unix like command line structure. Quotes and
     * single quotes are treated as nested strings.
     * 
     * @param command
     * @return
     */
    public static String[] partitionCommandLine(String command) {
        ArrayList<String> commands = new ArrayList<String>();

        int index = 0;

        StringBuffer buffer = new StringBuffer(command.length());

        boolean isApos = false;
        boolean isQuote = false;
        while(index < command.length()) {
            char c = command.charAt(index);

            switch(c) {
                case ' ':
                    if(!isQuote && !isApos) {
                        String arg = buffer.toString();
                        buffer = new StringBuffer(command.length() - index);
                        if(arg.length() > 0) {
                            commands.add(arg);
                        }
                    } else {
                        buffer.append(c);
                    }
                    break;
                case '\'':
                    if(!isQuote) {
                        isApos = !isApos;
                    } else {
                        buffer.append(c);
                    }
                    break;
                case '"':
                    if(!isApos) {
                        isQuote = !isQuote;
                    } else {
                        buffer.append(c);
                    }
                    break;
                default:
                    buffer.append(c);
            }

            index++;
        }

        if(buffer.length() > 0) {
            String arg = buffer.toString();
            commands.add(arg);
        }

        return commands.toArray(new String[commands.size()]);
    }

}
 

猜你喜欢

转载自01jiangwei01.iteye.com/blog/1835391