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

目的是要实现一个可以通过前台网页，来对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[i]);
            	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[i]);
            	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就有效果呢？