org.jutils.jprocesses.info.UnixProcessesService Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of jProcesses Show documentation
Show all versions of jProcesses Show documentation
Library to manage system processes using Java
/*
* Copyright 2016 Javier Garcia Alonso.
*
* 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.jutils.jprocesses.info;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import org.jutils.jprocesses.model.JProcessesResponse;
import org.jutils.jprocesses.model.ProcessInfo;
import org.jutils.jprocesses.util.OSDetector;
import org.jutils.jprocesses.util.ProcessesUtils;
/**
* Service implementation for Unix/BSD systems
*
* @author Javier Garcia Alonso
*/
class UnixProcessesService extends AbstractProcessesService {
//Use BSD sytle to get data in order to be compatible with Mac Systems(thanks to jkuharev for this tip)
private static final String PS_COLUMNS = "pid,ruser,vsize,rss,%cpu,lstart,cputime,nice,ucomm";
private static final String PS_FULL_COMMAND = "pid,command";
private static final int PS_COLUMNS_SIZE = PS_COLUMNS.split(",").length;
private static final int PS_FULL_COMMAND_SIZE = PS_FULL_COMMAND.split(",").length;
private String nameFilter = null;
@Override
protected List> parseList(String rawData) {
List> processesDataList = new ArrayList>();
String[] dataStringLines = rawData.split("\\r?\\n");
int index;
for (final String dataLine : dataStringLines) {
String line = dataLine.trim();
if (line.startsWith("PID")) {
// skip header
} else {
// LinkedHashMap keeps the insertion order, thus easier to debug
Map element = new LinkedHashMap();
String[] elements = line.split("\\s+", PS_COLUMNS_SIZE + 5);
index = 0;
element.put("pid", elements[index++]);
element.put("user", elements[index++]);
element.put("virtual_memory", elements[index++]);
element.put("physical_memory", elements[index++]);
element.put("cpu_usage", elements[index++]);
index++; // Skip weekday
String longDate = elements[index++] + " "
+ elements[index++] + " "
+ elements[index++]+ " "
+ elements[index++];
element.put("start_time", elements[index - 2]);
try {
element.put("start_datetime",
ProcessesUtils.parseUnixLongTimeToFullDate(longDate));
} catch (ParseException e) {
element.put("start_datetime", "01/01/2000 00:00:00");
System.err.println("Failed formatting date from ps: " + longDate + ", using \"01/01/2000 00:00:00\"");
}
element.put("proc_time", elements[index++]);
element.put("priority", elements[index++]);
element.put("proc_name", elements[index++]);
// first init full command by content of proc_name
element.put("command", elements[index - 1]);
processesDataList.add(element);
}
}
loadFullCommandData(processesDataList);
if (nameFilter != null) {
filterByName(processesDataList);
}
return processesDataList;
}
@Override
protected String getProcessesData(String name) {
if (name != null) {
if (OSDetector.isLinux()) {
return ProcessesUtils.executeCommand("ps",
"-o", PS_COLUMNS, "-C", name);
} else {
this.nameFilter = name;
}
}
return ProcessesUtils.executeCommand("ps",
"-e", "-o", PS_COLUMNS);
}
@Override
protected JProcessesResponse kill(int pid) {
JProcessesResponse response = new JProcessesResponse();
if (ProcessesUtils.executeCommandAndGetCode("kill", "-9", String.valueOf(pid)) == 0) {
response.setSuccess(true);
}
return response;
}
@Override
protected JProcessesResponse killGracefully(int pid) {
JProcessesResponse response = new JProcessesResponse();
if (ProcessesUtils.executeCommandAndGetCode("kill", "-15", String.valueOf(pid)) == 0) {
response.setSuccess(true);
}
return response;
}
public JProcessesResponse changePriority(int pid, int priority) {
JProcessesResponse response = new JProcessesResponse();
if (ProcessesUtils.executeCommandAndGetCode("renice", String.valueOf(priority),
"-p", String.valueOf(pid)) == 0) {
response.setSuccess(true);
}
return response;
}
public ProcessInfo getProcess(int pid) {
return getProcess (pid, false);
}
public ProcessInfo getProcess(int pid, boolean fastMode) {
this.fastMode = fastMode;
List> processList
= parseList(ProcessesUtils.executeCommand("ps",
"-o", PS_COLUMNS, "-p", String.valueOf(pid)));
if (processList != null && !processList.isEmpty()) {
Map processData = processList.get(0);
ProcessInfo info = new ProcessInfo();
info.setPid(processData.get("pid"));
info.setName(processData.get("proc_name"));
info.setTime(processData.get("proc_time"));
info.setCommand(processData.get("command"));
info.setCpuUsage(processData.get("cpu_usage"));
info.setPhysicalMemory(processData.get("physical_memory"));
info.setStartTime(processData.get("start_time"));
info.setUser(processData.get("user"));
info.setVirtualMemory(processData.get("virtual_memory"));
info.setPriority(processData.get("priority"));
return info;
}
return null;
}
private static void loadFullCommandData(List> processesDataList) {
Map commandsMap = new HashMap();
String data = ProcessesUtils.executeCommand("ps",
"-e", "-o", PS_FULL_COMMAND);
String[] dataStringLines = data.split("\\r?\\n");
for (final String dataLine : dataStringLines) {
if (!(dataLine.trim().startsWith("PID"))) {
String[] elements = dataLine.trim().split("\\s+", PS_FULL_COMMAND_SIZE);
if (elements.length == PS_FULL_COMMAND_SIZE) {
commandsMap.put(elements[0], elements[1]);
}
}
}
for (final Map process : processesDataList) {
if (commandsMap.containsKey(process.get("pid"))) {
process.put("command", commandsMap.get(process.get("pid")));
}
}
}
private void filterByName(List> processesDataList) {
List> processesToRemove = new ArrayList>();
for (final Map process : processesDataList) {
if (!nameFilter.equals(process.get("proc_name"))) {
processesToRemove.add(process);
}
}
processesDataList.removeAll(processesToRemove);
}
}