开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 中间件编程 > 查看源码
XMLEnvHandler.java
资源名称：blacktie-2.0.0.M3-src.zip　[点击查看]
上传用户：xfwatch
上传日期：2020-12-14
资源大小：872k
文件大小：17k
源码类别：

中间件编程

开发平台：

Java

XMLEnvHandler.java：源码内容
  1. /*
  2.  * JBoss, Home of Professional Open Source
  3.  * Copyright 2008, Red Hat, Inc., and others contributors as indicated
  4.  * by the @authors tag. All rights reserved.
  5.  * See the copyright.txt in the distribution for a
  6.  * full listing of individual contributors.
  7.  * This copyrighted material is made available to anyone wishing to use,
  8.  * modify, copy, or redistribute it subject to the terms and conditions
  9.  * of the GNU Lesser General Public License, v. 2.1.
  10.  * This program is distributed in the hope that it will be useful, but WITHOUT A
  11.  * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
  12.  * PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more details.
  13.  * You should have received a copy of the GNU Lesser General Public License,
  14.  * v.2.1 along with this distribution; if not, write to the Free Software
  15.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
  16.  * MA  02110-1301, USA.
  17.  */
  18. package org.jboss.blacktie.jatmibroker.core.conf;
  20. import java.util.ArrayList;
  21. import java.util.HashMap;
  22. import java.util.List;
  23. import java.util.Map;
  24. import java.util.Properties;
  26. import org.apache.log4j.LogManager;
  27. import org.apache.log4j.Logger;
  28. import org.xml.sax.Attributes;
  29. import org.xml.sax.SAXException;
  30. import org.xml.sax.helpers.DefaultHandler;
  32. /**
  33.  * XMLEnvHandler extends DefaultHandler to Environment Info
  34.  */
  35. public class XMLEnvHandler extends DefaultHandler {
  36. private static final Logger log = LogManager.getLogger(XMLEnvHandler.class);
  38. private final String DOMAIN = "DOMAIN";
  39. private final String BUFFER = "BUFFER";
  40. private final String ATTRIBUTE = "ATTRIBUTE";
  41. private final String SERVER_NAME = "SERVER";
  42. private final String SERVICE_NAME = "SERVICE";
  43. private final String NAME = "NAME";
  44. private final String VALUE = "VALUE";
  45. private final String ORB = "ORB";
  46. private final String MQ = "MQ";
  47. private final String JMX = "JMX";
  48. private final String MACHINE = "MACHINE";
  49. private final String MACHINE_REF = "MACHINE-REF";
  50. private final String ROLE = "role";
  51. private final String VERSION = "VERSION";
  53. private Properties prop;
  55. private String value;
  56. private String name;
  58. private String serverName;
  59. private String serviceName;
  61. private String jbossasIpAddr = System.getenv("JBOSSAS_IP_ADDR");
  63. private List<String> servers = new ArrayList<String>();
  65. private Map<String, BufferStructure> buffers = new HashMap<String, BufferStructure>();
  67. private Map<String, Machine> machines = new HashMap<String, Machine>();
  69. private List<Server> serverLaunchers = new ArrayList<Server>();
  71. private String currentBufferName;
  73. static int CHAR_SIZE = 1;
  74. static int LONG_SIZE = 8;
  75. static int INT_SIZE = 4;
  76. static int SHORT_SIZE = 2;
  77. static int FLOAT_SIZE = 4;
  78. static int DOUBLE_SIZE = 8;
  80. public XMLEnvHandler(Properties prop) {
  81. this.prop = prop;
  82. prop.put("blacktie.domain.servers", servers);
  83. prop.put("blacktie.domain.buffers", buffers);
  84. prop.put("blacktie.domain.serverLaunchers", serverLaunchers);
  85. }
  87. public void characters(char[] ch, int start, int length)
  88. throws SAXException {
  89. String strValue = new String(ch, start, length);
  90. value += strValue;
  91. }
  93. public void startElement(String namespaceURI, String localName,
  94. String qName, Attributes atts) throws SAXException {
  95. value = "";
  97. if (SERVER_NAME.equals(localName)) {
  98. if (atts != null) {
  99. for (int i = 0; i < atts.getLength(); i++) {
  100. if (atts.getLocalName(i).equals("name")) {
  101. serverName = atts.getValue(i);
  102. }
  103. }
  104. if (serverName == null) {
  105. serverName = "default";
  106. }
  107. if (servers.contains(serverName)) {
  108. throw new SAXException("Duplicate server detected: "
  109. + serverName);
  110. }
  111. servers.add(serverName);
  112. serverLaunchers.add(new Server(serverName));
  113. }
  114. } else if (MACHINE_REF.equals(localName)) {
  115. String value = null;
  116. Machine machine = null;
  117. for (int i = 0; i < atts.getLength(); i++) {
  118. if (atts.getLocalName(i).equals("id")) {
  119. value = atts.getValue(i);
  120. // Get the machine out of the list
  121. machine = machines.get(value);
  122. }
  123. }
  124. if (machine == null) {
  125. throw new SAXException("Machine did not exist: " + value);
  126. } else {
  127. // This will be the last server added
  128. Server server = serverLaunchers.get(serverLaunchers.size() - 1);
  129. server.addMachine(machine);
  130. }
  131. } else if (BUFFER.equals(localName)) {
  132. currentBufferName = atts.getValue(0);
  133. BufferStructure buffer = buffers.get(currentBufferName);
  134. if (buffer == null) {
  135. buffer = new BufferStructure();
  136. buffer.name = currentBufferName;
  137. buffer.wireSize = 0;
  138. buffer.memSize = 0;
  139. buffer.lastPad = 0;
  140. buffers.put(currentBufferName, buffer);
  141. } else {
  142. log.error("Duplicate buffer detected: " + currentBufferName);
  143. currentBufferName = null;
  144. }
  146. } else if (ATTRIBUTE.equals(localName)) {
  147. if (currentBufferName != null) {
  148. BufferStructure buffer = buffers.get(currentBufferName);
  149. AttributeStructure attribute = new AttributeStructure();
  150. attribute.id = null;
  151. attribute.type = null;
  152. attribute.count = 0;
  153. attribute.length = 0;
  154. attribute.wirePosition = 0;
  155. attribute.memPosition = 0;
  156. String type = null;
  157. for (int i = 0; i < atts.getLength(); i++) {
  158. if (atts.getLocalName(i).equals("id")) {
  159. attribute.id = atts.getValue(i);
  160. } else if (atts.getLocalName(i).equals("type")) {
  161. type = atts.getValue(i);
  162. } else if (atts.getLocalName(i).equals("arrayCount")) {
  163. attribute.count = Integer.parseInt(atts.getValue(i));
  164. } else if (atts.getLocalName(i).equals("arrayLength")) {
  165. attribute.length = Integer.parseInt(atts.getValue(i));
  166. }
  167. }
  169. int typeSize = -1;
  170. boolean contains = buffer.attributeNames.contains(attribute.id);
  171. boolean fail = false;
  172. if (!contains) {
  173. // short, int, long, float, double, char
  174. if (type.equals("short")) {
  175. typeSize = SHORT_SIZE;
  176. attribute.instanceSize = SHORT_SIZE;
  177. attribute.type = short.class;
  178. } else if (type.equals("int")) {
  179. typeSize = INT_SIZE;
  180. attribute.instanceSize = INT_SIZE;
  181. attribute.type = int.class;
  182. } else if (type.equals("long")) {
  183. typeSize = LONG_SIZE;
  184. attribute.instanceSize = LONG_SIZE;
  185. attribute.type = long.class;
  186. } else if (type.equals("float")) {
  187. typeSize = FLOAT_SIZE;
  188. attribute.instanceSize = FLOAT_SIZE;
  189. attribute.type = float.class;
  190. } else if (type.equals("double")) {
  191. typeSize = DOUBLE_SIZE;
  192. attribute.instanceSize = DOUBLE_SIZE;
  193. attribute.type = double.class;
  194. } else if (type.equals("char")) {
  195. typeSize = CHAR_SIZE;
  196. attribute.instanceSize = CHAR_SIZE;
  197. attribute.type = byte.class;
  198. } else if (type.equals("char[]")) {
  199. if (attribute.length == 0) {
  200. attribute.length = 1;
  201. }
  202. typeSize = CHAR_SIZE;
  203. attribute.instanceSize = CHAR_SIZE * attribute.length;
  204. attribute.type = byte[].class;
  205. } else if (type.equals("short[]")) {
  206. if (attribute.length == 0) {
  207. attribute.length = 1;
  208. }
  209. typeSize = SHORT_SIZE;
  210. attribute.instanceSize = SHORT_SIZE * attribute.length;
  211. attribute.type = short[].class;
  212. } else if (type.equals("int[]")) {
  213. if (attribute.length == 0) {
  214. attribute.length = 1;
  215. }
  216. typeSize = INT_SIZE;
  217. attribute.instanceSize = INT_SIZE * attribute.length;
  218. attribute.type = int[].class;
  219. } else if (type.equals("long[]")) {
  220. if (attribute.length == 0) {
  221. attribute.length = 1;
  222. }
  223. typeSize = LONG_SIZE;
  224. attribute.instanceSize = LONG_SIZE * attribute.length;
  225. attribute.type = long[].class;
  226. } else if (type.equals("float[]")) {
  227. if (attribute.length == 0) {
  228. attribute.length = 1;
  229. }
  230. typeSize = FLOAT_SIZE;
  231. attribute.instanceSize = FLOAT_SIZE * attribute.length;
  232. attribute.type = float[].class;
  233. } else if (type.equals("double[]")) {
  234. if (attribute.length == 0) {
  235. attribute.length = 1;
  236. }
  237. typeSize = DOUBLE_SIZE;
  238. attribute.instanceSize = DOUBLE_SIZE * attribute.length;
  239. attribute.type = double[].class;
  240. } else if (type.equals("char[][]")) {
  241. if (attribute.length == 0) {
  242. attribute.length = 1;
  243. }
  244. if (attribute.count == 0) {
  245. attribute.count = 1;
  246. }
  247. typeSize = CHAR_SIZE;
  248. attribute.instanceSize = CHAR_SIZE * attribute.length
  249. * attribute.count;
  250. attribute.type = byte[][].class;
  251. } else {
  252. log.error("Unknown attribute type: " + attribute.type);
  253. fail = true;
  254. }
  256. if (!fail) {
  257. buffer.attributes.add(attribute);
  259. // Extend the buffer by the required extra buffer size
  260. if (buffer.lastPad < typeSize) {
  261. buffer.lastPad = typeSize;
  262. }
  264. buffer.memSize = buffer.memSize
  265. + (buffer.memSize % typeSize);
  266. attribute.memPosition = buffer.memSize;
  267. attribute.wirePosition = buffer.wireSize;
  268. buffer.wireSize = buffer.wireSize
  269. + attribute.instanceSize;
  270. buffer.memSize = buffer.memSize
  271. + attribute.instanceSize;
  272. }
  273. } else {
  274. log.error("Duplicate attribute detected: " + attribute.id);
  275. }
  276. } else {
  277. log.error("No buffer is being processed");
  278. }
  279. } else if (ORB.equals(localName)) {
  280. for (int j = 0; j < atts.getLength(); j++) {
  281. if (atts.getLocalName(j).equals("OPT")) {
  282. String[] argv;
  283. argv = atts.getValue(j).split(" ");
  284. int orbargs = argv.length;
  286. for (int i = 1; i <= orbargs; i++) {
  287. String arg = "blacktie.orb.arg." + i;
  288. String toSet = argv[i - 1];
  289. if (jbossasIpAddr != null) {
  290. toSet = toSet.replace("${JBOSSAS_IP_ADDR}",
  291. jbossasIpAddr);
  292. }
  293. prop.setProperty(arg, toSet);
  294. log.debug(arg + " is " + toSet);
  295. }
  297. log.debug("blacktie.orb.args is " + orbargs);
  298. prop.setProperty("blacktie.orb.args", Integer
  299. .toString(orbargs));
  300. } else if (atts.getLocalName(j).equals("TRANS_FACTORY_ID")) {
  301. prop.setProperty("blacktie.trans.factoryid", atts
  302. .getValue(j));
  303. }
  304. }
  305. } else if (MQ.equals(localName)) {
  306. for (int i = 0; i < atts.getLength(); i++) {
  307. if (atts.getLocalName(i).equals("USER")) {
  308. String value = atts.getValue(i);
  309. prop.setProperty("StompConnectUsr", value);
  310. } else if (atts.getLocalName(i).equals("PASSWORD")) {
  311. String value = atts.getValue(i);
  312. prop.setProperty("StompConnectPwd", value);
  313. } else if (atts.getLocalName(i).equals("DESTINATION_TIMEOUT")) {
  314. String value = atts.getValue(i);
  315. prop.setProperty("DestinationTimeout", value);
  316. } else if (atts.getLocalName(i).equals("RECEIVE_TIMEOUT")) {
  317. String value = atts.getValue(i);
  318. prop.setProperty("RequestTimeout", value);
  319. } else if (atts.getLocalName(i).equals("TIME_TO_LIVE")) {
  320. String value = atts.getValue(i);
  321. prop.setProperty("TimeToLive", value);
  322. } else if (atts.getLocalName(i).equals("NAMING_URL")) {
  323. String value = atts.getValue(i);
  324. if (jbossasIpAddr != null) {
  325. value = value.replace("${JBOSSAS_IP_ADDR}",
  326. jbossasIpAddr);
  327. }
  328. prop.setProperty("java.naming.provider.url", value);
  329. }
  330. }
  331. } else if (MACHINE.equals(localName)) {
  332. Machine machine = new Machine();
  333. for (int i = 0; i < atts.getLength(); i++) {
  334. if (atts.getLocalName(i).equals("id")) {
  335. String value = atts.getValue(i);
  336. machine.setId(value);
  337. } else if (atts.getLocalName(i).equals("hostname")) {
  338. String value = atts.getValue(i);
  339. machine.setHostname(value);
  340. } else if (atts.getLocalName(i).equals("pathToExecutable")) {
  341. String value = atts.getValue(i);
  342. machine.setPathToExecutable(value);
  343. } else if (atts.getLocalName(i).equals("workingDirectory")) {
  344. String value = atts.getValue(i);
  345. machine.setWorkingDirectory(value);
  346. } else if (atts.getLocalName(i).equals("serverId")) {
  347. String value = atts.getValue(i);
  348. machine.setServerId(Integer.parseInt(value));
  349. } else if (atts.getLocalName(i).equals("argLine")) {
  350. String value = atts.getValue(i);
  351. machine.setArgLine(value);
  352. }
  353. }
  354. machines.put(machine.getId(), machine);
  355. } else if (JMX.equals(localName)) {
  356. for (int i = 0; i < atts.getLength(); i++) {
  357. if (atts.getLocalName(i).equals("url")) {
  358. String value = atts.getValue(i);
  359. if (jbossasIpAddr != null) {
  360. value = value.replace("${JBOSSAS_IP_ADDR}",
  361. jbossasIpAddr);
  362. }
  363. prop.setProperty("JMXURL", value);
  364. }
  365. }
  366. } else if (SERVICE_NAME.equals(localName)) {
  367. if (atts != null) {
  368. for (int i = 0; i < atts.getLength(); i++) {
  369. String attsLocalName = atts.getLocalName(i);
  370. if (attsLocalName.equals("name")) {
  371. serviceName = atts.getValue(i);
  372. if (serviceName.length() > 15) {
  373. log
  374. .warn("service "
  375. + serviceName
  376. + " is longer than XATMI_SERVICE_NAME_LENGTH and will be ignore");
  377. serviceName = null;
  378. break;
  379. }
  381. String serviceServer = (String) prop.get("blacktie."
  382. + serviceName + ".server");
  383. if (serviceServer != null
  384. && !serviceServer.equals(serverName)) {
  385. log.warn("service " + serviceName
  386. + " has already define in "
  387. + prop.get(serviceServer));
  388. serviceName = null;
  389. throw new SAXException(
  390. "Can not define the same service");
  391. }
  393. if (serviceName.indexOf("_ADMIN") >= 0) {
  394. log.warn("service " + serviceName
  395. + " is admin service");
  396. serviceName = null;
  397. throw new SAXException(
  398. "Can not define ADMIN service");
  399. }
  400. } else if (attsLocalName.equals("function_name")) {
  401. String func_key = "blacktie." + serviceName
  402. + ".function_name";
  403. String function_name = atts.getValue(i);
  404. prop.put(func_key, function_name);
  405. } else if (attsLocalName.equals("java_class_name")) {
  406. String java_key = "blacktie." + serviceName
  407. + ".java_class_name";
  408. String java_class_name = atts.getValue(i);
  409. prop.put(java_key, java_class_name);
  410. } else if (attsLocalName.equals("library_name")) {
  411. String lib_key = "blacktie." + serviceName
  412. + ".library_name";
  413. String library_name = atts.getValue(i);
  414. prop.put(lib_key, library_name);
  415. } else if (attsLocalName.equals("advertised")) {
  416. String advertised = atts.getValue(i);
  417. String ad_key = "blacktie." + serviceName
  418. + ".advertised";
  419. prop.put(ad_key, advertised);
  420. if (advertised.equals("true")) {
  421. String skey = "blacktie." + serverName
  422. + ".services";
  423. String object = (String) prop.get(skey);
  424. if (object == null) {
  425. object = serviceName;
  426. } else {
  427. object = new String(object + "," + serviceName);
  428. }
  429. prop.put(skey, object);
  430. }
  431. } else if (attsLocalName.equals("size")) {
  432. String sizeKey = "blacktie." + serviceName + ".size";
  433. String sizeVal = atts.getValue(i);
  434. prop.setProperty(sizeKey, sizeVal);
  435. }
  436. }
  438. if (serviceName != null) {
  439. // If a function was not defined above
  440. String func_key = "blacktie." + serviceName
  441. + ".function_name";
  442. if (prop.get(func_key) == null) {
  443. prop.put(func_key, serviceName);
  444. }
  446. prop.put("blacktie." + serviceName + ".server", serverName);
  447. prop.put("blacktie." + serviceName + ".transportLib",
  448. "hybrid");
  449. log.trace("Added the service: " + serviceName);
  450. }
  451. }
  452. } else if (ROLE.equals(localName)) {
  453. String key = null;
  454. if (serviceName != null) {
  455. log.debug("Processing role for service" + serviceName);
  456. key = "blacktie." + serviceName + ".security";
  457. } else if (serverName != null) {
  458. log.debug("Processing role for server: " + serverName);
  459. key = "blacktie." + serverName + ".security";
  460. }
  462. if (key != null) {
  463. String roleList = prop.getProperty(key, "");
  464. String name = null;
  465. String read = "false";
  466. String write = "false";
  468. for (int i = 0; i < atts.getLength(); i++) {
  469. String attsLocalName = atts.getLocalName(i);
  470. if (attsLocalName.equals("name")) {
  471. name = atts.getValue(i);
  472. } else if (atts.getLocalName(i).equals("read")) {
  473. read = atts.getValue(i);
  474. } else {
  475. write = atts.getValue(i);
  476. }
  477. }
  478. String role = name + ':' + read + ':' + write;
  479. if (roleList.length() > 0) {
  480. roleList = roleList + ',' + role;
  481. } else {
  482. roleList = role;
  483. }
  484. prop.put(key, roleList);
  485. log.trace("Added the role: " + role);
  486. } else {
  487. log.error("Ignoring role as not processing service or server");
  488. }
  489. }
  490. }
  492. public void endElement(String namespaceURI, String localName, String qName)
  493. throws SAXException {
  494. if (DOMAIN.equals(localName)) {
  495. prop.setProperty("blacktie.domain.name", value);
  496. } else if (NAME.equals(localName)) {
  497. name = value;
  498. } else if (VALUE.equals(localName)) {
  499. if (jbossasIpAddr != null) {
  500. value = value.replace("${JBOSSAS_IP_ADDR}", jbossasIpAddr);
  501. }
  502. prop.setProperty(name, value);
  503. } else if (SERVICE_NAME.equals(localName)) {
  504. if (serviceName != null) {
  505. serviceName = null;
  506. }
  507. } else if (VERSION.equals(localName)) {
  508. prop.setProperty("blacktie.domain.version", value);
  509. }
  510. }
  511. }