Browse Source
Rename ApplicationPidListener to ApplicationPidFileWriter (keeping the old class in a deprecated form) Fixes gh-1673pull/1840/head
4 changed files with 142 additions and 109 deletions
@ -0,0 +1,131 @@ |
|||||||
|
/* |
||||||
|
* Copyright 2010-2014 the original author or authors. |
||||||
|
* |
||||||
|
* 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 org.springframework.boot.actuate.system; |
||||||
|
|
||||||
|
import java.io.File; |
||||||
|
import java.util.concurrent.atomic.AtomicBoolean; |
||||||
|
|
||||||
|
import org.apache.commons.logging.Log; |
||||||
|
import org.apache.commons.logging.LogFactory; |
||||||
|
import org.springframework.boot.ApplicationPid; |
||||||
|
import org.springframework.boot.context.event.ApplicationStartedEvent; |
||||||
|
import org.springframework.context.ApplicationListener; |
||||||
|
import org.springframework.core.Ordered; |
||||||
|
import org.springframework.util.Assert; |
||||||
|
|
||||||
|
/** |
||||||
|
* An {@link ApplicationListener} that saves application PID into file. This application |
||||||
|
* listener will be triggered exactly once per JVM, and the file name can be overridden at |
||||||
|
* runtime with a System property or environment variable named "PIDFILE" (or "pidfile"). |
||||||
|
* |
||||||
|
* @author Jakub Kubrynski |
||||||
|
* @author Dave Syer |
||||||
|
* @author Phillip Webb |
||||||
|
* @since 1.2.0 |
||||||
|
*/ |
||||||
|
public class ApplicationPidFileWriter implements |
||||||
|
ApplicationListener<ApplicationStartedEvent>, Ordered { |
||||||
|
|
||||||
|
private static final Log logger = LogFactory.getLog(ApplicationPidFileWriter.class); |
||||||
|
|
||||||
|
private static final String DEFAULT_FILE_NAME = "application.pid"; |
||||||
|
|
||||||
|
private static final String[] PROPERTY_VARIABLES = { "PIDFILE", "pidfile" }; |
||||||
|
|
||||||
|
private static final AtomicBoolean created = new AtomicBoolean(false); |
||||||
|
|
||||||
|
private int order = Ordered.HIGHEST_PRECEDENCE + 13; |
||||||
|
|
||||||
|
private final File file; |
||||||
|
|
||||||
|
/** |
||||||
|
* Create a new {@link ApplicationPidFileWriter} instance using the filename |
||||||
|
* 'application.pid'. |
||||||
|
*/ |
||||||
|
public ApplicationPidFileWriter() { |
||||||
|
this.file = new File(DEFAULT_FILE_NAME); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* Create a new {@link ApplicationPidFileWriter} instance with a specified filename. |
||||||
|
* @param filename the name of file containing pid |
||||||
|
*/ |
||||||
|
public ApplicationPidFileWriter(String filename) { |
||||||
|
this(new File(filename)); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* Create a new {@link ApplicationPidFileWriter} instance with a specified file. |
||||||
|
* @param file the file containing pid |
||||||
|
*/ |
||||||
|
public ApplicationPidFileWriter(File file) { |
||||||
|
Assert.notNull(file, "File must not be null"); |
||||||
|
String override = getOverride(); |
||||||
|
if (override != null) { |
||||||
|
this.file = new File(override); |
||||||
|
} |
||||||
|
else { |
||||||
|
this.file = file; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private String getOverride() { |
||||||
|
for (String property : PROPERTY_VARIABLES) { |
||||||
|
try { |
||||||
|
String override = System.getProperty(property); |
||||||
|
override = (override != null ? override : System.getenv(property)); |
||||||
|
if (override != null) { |
||||||
|
return override; |
||||||
|
} |
||||||
|
} |
||||||
|
catch (Throwable ex) { |
||||||
|
System.err.println("Could not resolve '" + property |
||||||
|
+ "' as system property: " + ex); |
||||||
|
} |
||||||
|
} |
||||||
|
return null; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void onApplicationEvent(ApplicationStartedEvent event) { |
||||||
|
if (created.compareAndSet(false, true)) { |
||||||
|
try { |
||||||
|
new ApplicationPid().write(this.file); |
||||||
|
this.file.deleteOnExit(); |
||||||
|
} |
||||||
|
catch (Exception ex) { |
||||||
|
logger.warn(String.format("Cannot create pid file %s", this.file)); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
public void setOrder(int order) { |
||||||
|
this.order = order; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public int getOrder() { |
||||||
|
return this.order; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* Reset the created flag for testing purposes. |
||||||
|
*/ |
||||||
|
static void reset() { |
||||||
|
created.set(false); |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue