对tomcat admin中的datasource管理的扩展（一）

liuming794

　　目的是要实现一个可以通过前台网页，来对tomcat中的jndi数据源行进配置的应用。
　　一下想到了tomcat admin，可惜这东西只在5.5下有支持，6.0后还没支持，说是要下个大版本才会可能有支持。
　　所以自己下载的5.5下的admin，并且build了起来，看看还挺好用，差不多就是我需要的那些。
　　在来看看源码，发现使用的struts1写的mvc，页面上好多的struts1标签，然后对tomcat内部的操作，全部使用了MBean的方式。
　　我只需要对jndi数据源的配置部分，那就着重看这块的代码吧。
　　关键就是org.apache.webapp.admin.resources包下的SaveDataSourceAction类中的：

objectName = (String) mserver.invoke(oname, "addResource", params, signature);
　　这句调用了tomcat内部mbean中实现的addResource方法，我需要使用自己的com.elite.aps.dbservice.http.APSDataSourceFactory类作为数据源工厂类，别且对其中自己扩展的属性进行配置，因此在jsp页面与相关的action中添加了对应的代码，SaveDataSourceAction中也同样，在调用完addResource后，对返回的ObjectName调用其设置属性的方法，类似：

if(StringUtil.isNotNull(dataSourceForm.getDbtype()))
mserver.setAttribute(oname, new Attribute("dbtype",dataSourceForm.getDbtype()));
if(StringUtil.isNotNull(dataSourceForm.getInitialSize()))
mserver.setAttribute(oname, new Attribute("initialSize",dataSourceForm.getInitialSize()));
　　界面上操作完了穿件数据源后，发现并没有保存到对应的server.xml或者context.xml中，发现还有一步commit change的操作，而tomcat6.0中这部操作对应的MBean并没有实现，那就自己实现以下吧。
　　按照MBean的规则，定义一个StoreServerConfigMBean的接口，其中一个方法void storeConfig()，并且提供对应的实现类：

public class StoreServerConfig implements StoreServerConfigMBean {
private static Log log = LogFactory.getLog(StoreServerConfig.class);
private String serverFilename = "conf/server.xml";
private String contextFilename = "conf/context.xml";

public String getServerFilename() {
return serverFilename;
}
public void setServerFilename(String string) {
serverFilename = string;
}
public String getContextFilename() {
return contextFilename;
}
public void setContextFilename(String contextFilename) {
this.contextFilename = contextFilename;
}
public synchronized void storeConfig() {
try{
storeServerFile();//save to server.xml
storeContextFile();//save to context.xml
}catch(Exception e){
log.error("",e);
}
}
private void storeContextFile(){
StoreFileUtil sfContext = new StoreFileUtil(System.getProperty("catalina.base"), getContextFilename(),"utf-8");
try {
PrintWriter writer = sfContext.getWriter();
SAXReader reader = new SAXReader();   
Document doc = reader.read(sfContext.getReader());
Element contextElement=(Element)doc.selectSingleNode("/Context");
List<Element> resourceList=contextElement.selectNodes("ResourceLink[@type='"+ResourceUtils.DATASOURCE_CLASS+"']");
if(resourceList!=null && resourceList.size()>0){
for(Element resource : resourceList){
contextElement.remove(resource);
}
}
MBeanServer mserver = CatalinaUtils.getServer();
DataSourcesForm dsf=ResourceUtils.getDataSourcesForm(mserver, "Global", null, null, "Catalina");
String[] dataSources=dsf.getDataSources();
for(int i=0;i<dataSources.length;i++){
ObjectName oname = new ObjectName(dataSources);
String name=(String)mserver.getAttribute(oname, "name");
Element resourceLinkEle=contextElement.addElement("ResourceLink");
resourceLinkEle.addAttribute("name", name);
resourceLinkEle.addAttribute("type", ResourceUtils.DATASOURCE_CLASS);
resourceLinkEle.addAttribute("global", name);
}
writeOut(writer,doc);
sfContext.move();
} catch (Exception e) {
log.error("",e);
}
}
private void storeServerFile(){
StoreFileUtil sfServer = new StoreFileUtil(System.getProperty("catalina.base"), getServerFilename(),"utf-8");
try {
PrintWriter writer = sfServer.getWriter();
SAXReader reader = new SAXReader();   
Document doc = reader.read(sfServer.getReader());
Element gnrElement=(Element)doc.selectSingleNode("/Server/GlobalNamingResources");
List<Element> resourceList=gnrElement.selectNodes("Resource[@type='"+ResourceUtils.DATASOURCE_CLASS+"']");
if(resourceList!=null && resourceList.size()>0){
for(Element resource : resourceList){
gnrElement.remove(resource);
CatalinaUtils.removeResourceLinkFromNamingContextListener(resource.attributeValue("name"));
}
}
MBeanServer mserver = CatalinaUtils.getServer();
DataSourcesForm dsf=ResourceUtils.getDataSourcesForm(mserver, "Global", null, null, "Catalina");
String[] dataSources=dsf.getDataSources();
for(int i=0;i<dataSources.length;i++){
ObjectName oname = new ObjectName(dataSources);
String name=(String)mserver.getAttribute(oname, "name");
String type=(String)mserver.getAttribute(oname, "type");
String factory=(String)mserver.getAttribute(oname, "factory");
String url=(String)mserver.getAttribute(oname, "url");
String driverClassName=(String)mserver.getAttribute(oname, "driverClassName");
String username=(String)mserver.getAttribute(oname, "username");
String password=(String)mserver.getAttribute(oname, "password");
String maxActive=(String)mserver.getAttribute(oname, "maxActive");
String maxIdle=(String)mserver.getAttribute(oname, "maxIdle");
String maxWait=(String)mserver.getAttribute(oname, "maxWait");
Element resourceEle=gnrElement.addElement("Resource");
resourceEle.addAttribute("name", name);
resourceEle.addAttribute("type", type);
resourceEle.addAttribute("factory", factory);
resourceEle.addAttribute("url", url);
resourceEle.addAttribute("driverClassName", driverClassName);
resourceEle.addAttribute("username", username);
resourceEle.addAttribute("password", password);
resourceEle.addAttribute("maxActive", maxActive);
resourceEle.addAttribute("maxIdle", maxIdle);
resourceEle.addAttribute("maxWait", maxWait);
CatalinaUtils.addResourceLinkFromNamingContextListener(resourceEle.attributeValue("name"));
}
writeOut(writer,doc);
sfServer.move();
} catch (Exception e) {
log.error("",e);
}
}
private void writeOut(PrintWriter writer,Document doc) throws Exception{
try {
OutputFormat format = OutputFormat.createPrettyPrint();
format.setEncoding("utf-8");
format.setLineSeparator(System.getProperty("line.separator"));
XMLWriter xmlWriter = new NotepadXMLWriter(writer, format);
xmlWriter.write(doc);
xmlWriter.flush();
} finally {
try {
writer.flush();
} catch (Exception e) {
log.error(e);
}
try {
writer.close();
} catch (Exception e) {
throw (e);
}
}
}
}

　　这个类中，主要是去做了把当前jmx中的datasource写到响应的配置文件中，server.xml中写入Resource到GlobalNamingResources标签下，context.xml中写入对应的ResourceLink。这样应用就可以直接使用到了，看似不错。
　　这里需要提到的，写出xml到文本时候，默认格式会不好看，使用OutputFormat format = OutputFormat.createPrettyPrint();后，会好看很多。
　　但是，不论怎么设置格式，windows上用notepad打开，总会出现黑色方块这样的难看的符号，
　　原因其实是，虽然标签结束或者有子标签的情况下，使用的format会根据操作系统来替换成对应的换行符号，windows下是\n\r。
　　但是，在注释里面的换行，dom4j里面是被看做一整段text，不受到format的影响的，因此就扩展了一下XMLWriter，重写了writeComment方法，把难看的\n都替换了，这下好看多了。

@Override
protected void writeComment(String text) throws IOException {
text=text.replaceAll("\\n", System.getProperty("line.separator"));
super.writeComment(text);
}
　　至此，到我自己的tomcat admin中，去创建删除修改datasource，然后commit change后，能保存到tomcat的配置文件中，如果重启tomcat，就可以看到效果了。
　　但是，能不能不重启tomcat就有效果呢？