package io.hawt.web;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URISyntaxException;
import java.net.URL;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Properties;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.json.simple.JSONArray;
import org.json.simple.JSONObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import static io.hawt.web.ServletHelpers.writeEmpty;
public class SpringBatchConfigServlet extends HttpServlet {
private static final transient Logger LOG = LoggerFactory.getLogger(SpringBatchConfigServlet.class);
@Override
public void doGet(HttpServletRequest httpServletRequest,
HttpServletResponse httpServletResponse) throws IOException, ServletException {
InputStream propsIn = SpringBatchConfigServlet.class.getClassLoader().getResourceAsStream("springbatch.properties");
httpServletResponse.setHeader("Content-type", "application/json");
if (propsIn == null) {
writeEmpty(httpServletResponse.getWriter());
return;
}
Properties properties = new Properties();
properties.load(propsIn);
JSONObject responseJson = new JSONObject();
JSONArray springBatchServersJson = new JSONArray();
List<? extends String> springBatchServers = Arrays.asList(properties.getProperty("springBatchServerList").split(","));
springBatchServersJson.addAll(springBatchServers);
responseJson.put("springBatchServerList", springBatchServersJson);
String res = "success";
httpServletResponse.getWriter().println(responseJson.toJSONString());
}
@Override
protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
File file = getPropertiesFile("springbatch.properties");
Properties properties = getProperties(file);
String server = req.getParameter("server");
String replaceServer = req.getParameter("replaceServer");
if ((replaceServer != null && !replaceServer.isEmpty()) && (server != null && !server.isEmpty())) {
String[] servers = properties.getProperty("springBatchServerList").split(",");
List<String> serverList = new ArrayList<String>(Arrays.asList(servers));
if (serverList.contains(replaceServer)) {
serverList.remove(replaceServer);
serverList.add(server);
}
properties.setProperty("springBatchServerList", join(serverList, ","));
properties.store(new FileOutputStream(file), null);
resp.getWriter().print("updated");
} else if (server != null && !server.isEmpty()) {
properties.setProperty("springBatchServerList", properties.getProperty("springBatchServerList") + "," + server);
properties.store(new FileOutputStream(file), null);
resp.getWriter().print("added");
} else {
resp.getWriter().print("failed");
}
}
@Override
protected void doDelete(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
File file = getPropertiesFile("springbatch.properties");
Properties properties = getProperties(file);
String server = req.getParameter("server");
if (server != null && !server.isEmpty()) {
String[] servers = properties.getProperty("springBatchServerList").split(",");
List<String> serverList = new ArrayList<String>(Arrays.asList(servers));
serverList.remove(server);
properties.setProperty("springBatchServerList", join(serverList, ","));
properties.store(new FileOutputStream(file), null);
resp.getWriter().print("deleted");
} else {
resp.getWriter().print("failed");
}
}
private File getPropertiesFile(String name) {
URL propsUrl = SpringBatchConfigServlet.class.getClassLoader().getResource(name);
File file = null;
try {
file = new File(propsUrl.toURI());
} catch (URISyntaxException s) {
LOG.error(s.getMessage());
}
return file;
}
private Properties getProperties(File file) throws IOException {
FileInputStream propsIn = new FileInputStream(file);
Properties properties = new Properties();
properties.load(propsIn);
return properties;
}
private String join(List<String> list, String div) {
StringBuffer buffer = new StringBuffer();
for (String e : list) {
buffer.append((list.size() == (list.indexOf(e) + 1)) ? e : e + div);
}
return buffer.toString();
}
}