liuming794 发表于 2017-2-4 12:47:09

对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);
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就有效果呢?
页: [1]
查看完整版本: 对tomcat admin中的datasource管理的扩展(一)