getModifyMarkCache() {
59 | return this.modifyMarkCache;
60 | }
61 |
62 |
63 | /**
64 | * 获取配置文件路径, 单元测试用
65 | *
66 | * @param dataId
67 | * @param group
68 | * @return
69 | * @throws FileNotFoundException
70 | */
71 | public String getFilePath(String dataId, String group) throws FileNotFoundException {
72 | return getFilePath(Constants.BASE_DIR + "/" + group + "/" + dataId);
73 | }
74 |
75 |
76 | public void saveToDisk(ConfigInfo configInfo) {
77 | String group = configInfo.getGroup();
78 | String dataId = configInfo.getDataId();
79 | String content = configInfo.getContent();
80 | String cacheKey = generateCacheKey(group, dataId);
81 | // 标记正在写磁盘
82 | if (this.modifyMarkCache.putIfAbsent(cacheKey, true) == null) {
83 | File tempFile = null;
84 | try {
85 | // 目标目录
86 | String groupPath = getFilePath(Constants.BASE_DIR + "/" + group);
87 | createDirIfNessary(groupPath);
88 | // 目标文件
89 | File targetFile = createFileIfNessary(groupPath, dataId);
90 | // 创建临时文件
91 | tempFile = createTempFile(dataId, group);
92 | // 写数据至临时文件
93 | FileUtils.writeStringToFile(tempFile, content, Constants.ENCODE);
94 | // 用临时文件覆盖目标文件, 完成本次磁盘操作
95 | FileUtils.copyFile(tempFile, targetFile);
96 | }
97 | catch (Exception e) {
98 | String errorMsg = "save disk error, dataId=" + dataId + ",group=" + group;
99 | log.error(errorMsg, e);
100 | throw new ConfigServiceException(errorMsg, e);
101 | }
102 | finally {
103 | // 删除临时文件
104 | if (tempFile != null && tempFile.exists()) {
105 | FileUtils.deleteQuietly(tempFile);
106 | }
107 | // 清除标记
108 | this.modifyMarkCache.remove(cacheKey);
109 | }
110 | }
111 | else {
112 | throw new ConfigServiceException("config info is being motified, dataId=" + dataId + ",group=" + group);
113 | }
114 |
115 | }
116 |
117 |
118 | public boolean isModified(String dataId, String group) {
119 | return this.modifyMarkCache.get(generateCacheKey(group, dataId)) != null;
120 | }
121 |
122 |
123 | /**
124 | * 生成缓存key，用于标记文件是否正在被修改
125 | *
126 | * @param group
127 | * @param dataId
128 | *
129 | * @return
130 | */
131 | public final String generateCacheKey(String group, String dataId) {
132 | return group + "/" + dataId;
133 | }
134 |
135 |
136 | public void removeConfigInfo(String dataId, String group) {
137 | String cacheKey = generateCacheKey(group, dataId);
138 | // 标记正在写磁盘
139 | if (this.modifyMarkCache.putIfAbsent(cacheKey, true) == null) {
140 | try {
141 | String basePath = getFilePath(Constants.BASE_DIR);
142 | createDirIfNessary(basePath);
143 |
144 | String groupPath = getFilePath(Constants.BASE_DIR + "/" + group);
145 | File groupDir = new File(groupPath);
146 | if (!groupDir.exists()) {
147 | return;
148 | }
149 |
150 | String dataPath = getFilePath(Constants.BASE_DIR + "/" + group + "/" + dataId);
151 | File dataFile = new File(dataPath);
152 | if (!dataFile.exists()) {
153 | return;
154 | }
155 |
156 | FileUtils.deleteQuietly(dataFile);
157 | }
158 | catch (Exception e) {
159 | String errorMsg = "delete config info error, dataId=" + dataId + ",group=" + group;
160 | log.error(errorMsg, e);
161 | throw new ConfigServiceException(errorMsg, e);
162 | }
163 | finally {
164 | // 清除标记
165 | this.modifyMarkCache.remove(cacheKey);
166 | }
167 | }
168 | else {
169 | throw new ConfigServiceException("config info is being motified, dataId=" + dataId + ",group=" + group);
170 | }
171 | }
172 |
173 |
174 | private String getFilePath(String dir) throws FileNotFoundException {
175 | return WebUtils.getRealPath(servletContext, dir);
176 | }
177 |
178 |
179 | private void createDirIfNessary(String path) {
180 | final File dir = new File(path);
181 | if (!dir.exists()) {
182 | dir.mkdirs();
183 | }
184 | }
185 |
186 |
187 | private File createFileIfNessary(String parent, String child) throws IOException {
188 | final File file = new File(parent, child);
189 | if (!file.exists()) {
190 | file.createNewFile();
191 | // 设置文件权限
192 | changeFilePermission(file);
193 | }
194 | return file;
195 | }
196 |
197 |
198 | private void changeFilePermission(File file) {
199 | // 文件权限设置为600
200 | file.setExecutable(false, false);
201 | file.setWritable(false, false);
202 | file.setReadable(false, false);
203 | file.setExecutable(false, true);
204 | file.setWritable(true, true);
205 | file.setReadable(true, true);
206 | }
207 |
208 |
209 | private File createTempFile(String dataId, String group) throws IOException {
210 | return File.createTempFile(group + "-" + dataId, ".tmp");
211 | }
212 |
213 | }
214 |
--------------------------------------------------------------------------------
/diamond-server/src/main/webapp/js/tooltips.js:
--------------------------------------------------------------------------------
1 | //compatible with prototype
2 | if(typeof Prototype != 'undefined' && (typeof $ != 'undefined')) {
3 | $prototype = $;
4 | }
5 |
6 |
7 | // Tooltip Object
8 | var Tooltip = Class.create();
9 | Tooltip.prototype = {
10 | initialize: function(el, options) {
11 | this.el = $prototype(el);
12 | this.initialized = false;
13 | this.setOptions(options);
14 |
15 | // Event handlers
16 | this.showEvent = this.show.bindAsEventListener(this);
17 | this.hideEvent = this.hide.bindAsEventListener(this);
18 | this.updateEvent = this.update.bindAsEventListener(this);
19 | Event.observe(this.el, "mouseover", this.showEvent );
20 | Event.observe(this.el, "mouseout", this.hideEvent );
21 |
22 | // Removing title from DOM element to avoid showing it
23 | this.content = this.el.title;
24 | this.el.title = "";
25 |
26 | // If descendant elements has 'alt' attribute defined, clear it
27 | this.el.descendants().each(function(el){
28 | if(Element.readAttribute(el, 'alt'))
29 | el.alt = "";
30 | });
31 | },
32 | setOptions: function(options) {
33 | this.options = {
34 | backgroundColor: '#999', // Default background color
35 | borderColor: '#666', // Default border color
36 | textColor: '', // Default text color (use CSS value)
37 | textShadowColor: '', // Default text shadow color (use CSS value)
38 | maxWidth: 250, // Default tooltip width
39 | align: "left", // Default align
40 | delay: 250, // Default delay before tooltip appears in ms
41 | mouseFollow: true, // Tooltips follows the mouse moving
42 | opacity: .75, // Default tooltips opacity
43 | appearDuration: .25, // Default appear duration in sec
44 | hideDuration: .25 // Default disappear duration in sec
45 | };
46 | Object.extend(this.options, options || {});
47 | },
48 | show: function(e) {
49 | this.xCord = Event.pointerX(e);
50 | this.yCord = Event.pointerY(e);
51 | if(!this.initialized)
52 | this.timeout = window.setTimeout(this.appear.bind(this), this.options.delay);
53 | },
54 | hide: function(e) {
55 | if(this.initialized) {
56 | //this.appearingFX.cancel();
57 | if(this.options.mouseFollow)
58 | Event.stopObserving(this.el, "mousemove", this.updateEvent);
59 | this.tooltip.hide();
60 | //new Effect.Fade(this.tooltip, {duration: this.options.hideDuration, afterFinish: function() { Element.remove(this.tooltip) }.bind(this) });
61 | }
62 | this._clearTimeout(this.timeout);
63 |
64 | this.initialized = false;
65 | },
66 | update: function(e){
67 | this.xCord = Event.pointerX(e);
68 | this.yCord = Event.pointerY(e);
69 | this.setup();
70 | },
71 | appear: function() {
72 | // Building tooltip container
73 | /*
74 | this.tooltip = Builder.node("div", {className: "tooltip", style: "display: none;" }, [
75 | Builder.node("div", {className:"xtop"}, [
76 | Builder.node("div", {className:"xb1", style:"background-color:" + this.options.borderColor + ";"}),
77 | Builder.node("div", {className:"xb2", style: "background-color:" + this.options.backgroundColor + "; border-color:" + this.options.borderColor + ";"}),
78 | Builder.node("div", {className:"xb3", style: "background-color:" + this.options.backgroundColor + "; border-color:" + this.options.borderColor + ";"}),
79 | Builder.node("div", {className:"xb4", style: "background-color:" + this.options.backgroundColor + "; border-color:" + this.options.borderColor + ";"})
80 | ]),
81 | Builder.node("div", {className: "xboxcontent", style: "background-color:" + this.options.backgroundColor +
82 | "; border-color:" + this.options.borderColor +
83 | ((this.options.textColor != '') ? "; color:" + this.options.textColor : "") +
84 | ((this.options.textShadowColor != '') ? "; text-shadow:2px 2px 0" + this.options.textShadowColor + ";" : "")}, this.content),
85 | Builder.node("div", {className:"xbottom"}, [
86 | Builder.node("div", {className:"xb4", style: "background-color:" + this.options.backgroundColor + "; border-color:" + this.options.borderColor + ";"}),
87 | Builder.node("div", {className:"xb3", style: "background-color:" + this.options.backgroundColor + "; border-color:" + this.options.borderColor + ";"}),
88 | Builder.node("div", {className:"xb2", style: "background-color:" + this.options.backgroundColor + "; border-color:" + this.options.borderColor + ";"}),
89 | Builder.node("div", {className:"xb1", style:"background-color:" + this.options.borderColor + ";"})
90 | ]),
91 | ]);
92 | */
93 |
94 | var tooltipString = ''+
95 | ''
112 | new Insertion.Before(document.body.childNodes[0], tooltipString)
113 | this.tooltip = document.body.childNodes[0];
114 | //document.body.insertBefore(this.tooltip, document.body.childNodes[0]);
115 |
116 | Element.extend(this.tooltip); // IE needs element to be manually extended
117 | this.options.width = this.tooltip.getWidth();
118 | this.tooltip.style.width = this.options.width + 'px'; // IE7 needs width to be defined
119 |
120 | this.setup();
121 |
122 | if(this.options.mouseFollow)
123 | Event.observe(this.el, "mousemove", this.updateEvent);
124 |
125 | this.initialized = true;
126 | this.tooltip.show();
127 | //this.appearingFX = new Effect.Appear(this.tooltip, {duration: this.options.appearDuration, to: this.options.opacity });
128 | },
129 | setup: function(){
130 | // If content width is more then allowed max width, set width to max
131 | if(this.options.width > this.options.maxWidth) {
132 | this.options.width = this.options.maxWidth;
133 | this.tooltip.style.width = this.options.width + 'px';
134 | }
135 |
136 | // Tooltip doesn't fit the current document dimensions
137 | if(this.xCord + this.options.width >= Element.getWidth(document.body)) {
138 | this.options.align = "right";
139 | this.xCord = this.xCord - this.options.width + 20;
140 | }
141 |
142 | this.tooltip.style.left = this.xCord - 7 + "px";
143 | this.tooltip.style.top = this.yCord + 12 + "px";
144 | },
145 | stop: function () {
146 | this.hide();
147 | Event.stopObserving(this.el, "mouseover", this.showEvent);
148 | Event.stopObserving(this.el, "mouseout", this.hideEvent);
149 | Event.stopObserving(this.el, "mousemove", this.updateEvent);
150 | },
151 | _clearTimeout: function(timer) {
152 | clearTimeout(timer);
153 | clearInterval(timer);
154 | return null;
155 | }
156 | };
--------------------------------------------------------------------------------
/diamond-client/src/main/java/com/taobao/diamond/manager/impl/DefaultDiamondManager.java:
--------------------------------------------------------------------------------
1 | /*
2 | * (C) 2007-2012 Alibaba Group Holding Limited.
3 | *
4 | * This program is free software; you can redistribute it and/or modify
5 | * it under the terms of the GNU General Public License version 2 as
6 | * published by the Free Software Foundation.
7 | * Authors:
8 | * leiwen , boyan
9 | */
10 | package com.taobao.diamond.manager.impl;
11 |
12 | import java.io.IOException;
13 | import java.io.StringReader;
14 | import java.util.LinkedList;
15 | import java.util.List;
16 | import java.util.Properties;
17 |
18 | import org.apache.commons.logging.Log;
19 | import org.apache.commons.logging.LogFactory;
20 |
21 | import com.taobao.diamond.client.DiamondConfigure;
22 | import com.taobao.diamond.client.DiamondSubscriber;
23 | import com.taobao.diamond.client.impl.DefaultSubscriberListener;
24 | import com.taobao.diamond.client.impl.DiamondClientFactory;
25 | import com.taobao.diamond.manager.DiamondManager;
26 | import com.taobao.diamond.manager.ManagerListener;
27 |
28 |
29 | /**
30 | * 需要注意的是：一个JVM中一个DataID只能对应一个DiamondManager
31 | *
32 | * @author aoqiong
33 | *
34 | */
35 | public class DefaultDiamondManager implements DiamondManager {
36 |
37 | private static final Log log = LogFactory.getLog(DefaultDiamondManager.class);
38 |
39 | private DiamondSubscriber diamondSubscriber = null;
40 | private final List managerListeners = new LinkedList();
41 | private final String dataId;
42 | private final String group;
43 |
44 |
45 | public DefaultDiamondManager(String dataId, ManagerListener managerListener) {
46 | this(null, dataId, managerListener);
47 | }
48 |
49 |
50 | public DefaultDiamondManager(String group, String dataId, ManagerListener managerListener) {
51 | this.dataId = dataId;
52 | this.group = group;
53 |
54 | diamondSubscriber = DiamondClientFactory.getSingletonDiamondSubscriber();
55 |
56 | this.managerListeners.add(managerListener);
57 | ((DefaultSubscriberListener) diamondSubscriber.getSubscriberListener()).addManagerListeners(this.dataId,
58 | this.group, this.managerListeners);
59 | diamondSubscriber.addDataId(this.dataId, this.group);
60 | diamondSubscriber.start();
61 |
62 | }
63 |
64 |
65 | public DefaultDiamondManager(String dataId, List managerListenerList) {
66 | this(null, dataId, managerListenerList);
67 | }
68 |
69 |
70 | /**
71 | * 使用指定的集群类型clusterType
72 | *
73 | * @param group
74 | * @param dataId
75 | * @param managerListenerList
76 | * @param clusterType
77 | */
78 | public DefaultDiamondManager(String group, String dataId, List managerListenerList) {
79 | this.dataId = dataId;
80 | this.group = group;
81 |
82 | diamondSubscriber = DiamondClientFactory.getSingletonDiamondSubscriber();
83 |
84 | this.managerListeners.addAll(managerListenerList);
85 | ((DefaultSubscriberListener) diamondSubscriber.getSubscriberListener()).addManagerListeners(this.dataId,
86 | this.group, this.managerListeners);
87 | diamondSubscriber.addDataId(this.dataId, this.group);
88 | diamondSubscriber.start();
89 | }
90 |
91 |
92 | public void setManagerListener(ManagerListener managerListener) {
93 | this.managerListeners.clear();
94 | this.managerListeners.add(managerListener);
95 |
96 | ((DefaultSubscriberListener) diamondSubscriber.getSubscriberListener()).removeManagerListeners(this.dataId,
97 | this.group);
98 | ((DefaultSubscriberListener) diamondSubscriber.getSubscriberListener()).addManagerListeners(this.dataId,
99 | this.group, this.managerListeners);
100 | }
101 |
102 |
103 | public void close() {
104 | /**
105 | * 因为同一个DataID只能对应一个MnanagerListener，所以，关闭时一次性关闭所有ManagerListener即可
106 | */
107 | ((DefaultSubscriberListener) diamondSubscriber.getSubscriberListener()).removeManagerListeners(this.dataId,
108 | this.group);
109 |
110 | diamondSubscriber.removeDataId(dataId, group);
111 | if (diamondSubscriber.getDataIds().size() == 0) {
112 | diamondSubscriber.close();
113 | }
114 |
115 | }
116 |
117 |
118 | public String getConfigureInfomation(long timeout) {
119 | return diamondSubscriber.getConfigureInfomation(this.dataId, this.group, timeout);
120 | }
121 |
122 |
123 | public String getAvailableConfigureInfomation(long timeout) {
124 | return diamondSubscriber.getAvailableConfigureInfomation(dataId, group, timeout);
125 | }
126 |
127 |
128 | public String getAvailableConfigureInfomationFromSnapshot(long timeout) {
129 | return diamondSubscriber.getAvailableConfigureInfomationFromSnapshot(dataId, group, timeout);
130 | }
131 |
132 |
133 | public Properties getAvailablePropertiesConfigureInfomation(long timeout) {
134 | String configInfo = this.getAvailableConfigureInfomation(timeout);
135 | Properties properties = new Properties();
136 | try {
137 | properties.load(new StringReader(configInfo));
138 | return properties;
139 | }
140 | catch (IOException e) {
141 | log.warn("装载properties失败：" + configInfo, e);
142 | throw new RuntimeException("装载properties失败：" + configInfo, e);
143 | }
144 | }
145 |
146 |
147 | public Properties getAvailablePropertiesConfigureInfomationFromSnapshot(long timeout) {
148 | String configInfo = this.getAvailableConfigureInfomationFromSnapshot(timeout);
149 | Properties properties = new Properties();
150 | try {
151 | properties.load(new StringReader(configInfo));
152 | return properties;
153 | }
154 | catch (IOException e) {
155 | log.warn("装载properties失败：" + configInfo, e);
156 | throw new RuntimeException("装载properties失败：" + configInfo, e);
157 | }
158 | }
159 |
160 |
161 | public void setManagerListeners(List managerListenerList) {
162 | this.managerListeners.clear();
163 | this.managerListeners.addAll(managerListenerList);
164 |
165 | ((DefaultSubscriberListener) diamondSubscriber.getSubscriberListener()).removeManagerListeners(this.dataId,
166 | this.group);
167 | ((DefaultSubscriberListener) diamondSubscriber.getSubscriberListener()).addManagerListeners(this.dataId,
168 | this.group, this.managerListeners);
169 | }
170 |
171 |
172 | public DiamondConfigure getDiamondConfigure() {
173 | return diamondSubscriber.getDiamondConfigure();
174 | }
175 |
176 |
177 | public void setDiamondConfigure(DiamondConfigure diamondConfigure) {
178 | diamondSubscriber.setDiamondConfigure(diamondConfigure);
179 | }
180 |
181 |
182 | public Properties getPropertiesConfigureInfomation(long timeout) {
183 | String configInfo = this.getConfigureInfomation(timeout);
184 | Properties properties = new Properties();
185 | try {
186 | properties.load(new StringReader(configInfo));
187 | return properties;
188 | }
189 | catch (IOException e) {
190 | log.warn("装载properties失败：" + configInfo, e);
191 | throw new RuntimeException("装载properties失败：" + configInfo, e);
192 | }
193 | }
194 |
195 |
196 | public List getManagerListeners() {
197 | return this.managerListeners;
198 | }
199 |
200 | }
201 |
--------------------------------------------------------------------------------
/diamond-sdk/src/main/java/com/taobao/diamond/util/ResourceUtils.java:
--------------------------------------------------------------------------------
1 | /*
2 | * (C) 2007-2012 Alibaba Group Holding Limited.
3 | *
4 | * This program is free software; you can redistribute it and/or modify
5 | * it under the terms of the GNU General Public License version 2 as
6 | * published by the Free Software Foundation.
7 | * Authors:
8 | * leiwen , boyan
9 | */
10 | package com.taobao.diamond.util;
11 |
12 | import java.io.File;
13 | import java.io.IOException;
14 | import java.io.InputStream;
15 | import java.io.InputStreamReader;
16 | import java.io.Reader;
17 | import java.net.URL;
18 | import java.util.Properties;
19 |
20 |
21 | /**
22 | *
23 | * @author boyan
24 | * @date 2010-5-4
25 | */
26 | public class ResourceUtils extends Object {
27 |
28 | /** */
29 | /**
30 | * Returns the URL of the resource on the classpath
31 | *
32 | * @param resource
33 | * The resource to find
34 | * @throws IOException
35 | * If the resource cannot be found or read
36 | * @return The resource
37 | */
38 | public static URL getResourceURL(String resource) throws IOException {
39 | URL url = null;
40 | ClassLoader loader = ResourceUtils.class.getClassLoader();
41 | if (loader != null)
42 | url = loader.getResource(resource);
43 | if (url == null)
44 | url = ClassLoader.getSystemResource(resource);
45 | if (url == null)
46 | throw new IOException("Could not find resource " + resource);
47 | return url;
48 | }
49 |
50 |
51 | /** */
52 | /**
53 | * Returns the URL of the resource on the classpath
54 | *
55 | * @param loader
56 | * The classloader used to load the resource
57 | * @param resource
58 | * The resource to find
59 | * @throws IOException
60 | * If the resource cannot be found or read
61 | * @return The resource
62 | */
63 | public static URL getResourceURL(ClassLoader loader, String resource) throws IOException {
64 | URL url = null;
65 | if (loader != null)
66 | url = loader.getResource(resource);
67 | if (url == null)
68 | url = ClassLoader.getSystemResource(resource);
69 | if (url == null)
70 | throw new IOException("Could not find resource " + resource);
71 | return url;
72 | }
73 |
74 |
75 | /** */
76 | /**
77 | * Returns a resource on the classpath as a Stream object
78 | *
79 | * @param resource
80 | * The resource to find
81 | * @throws IOException
82 | * If the resource cannot be found or read
83 | * @return The resource
84 | */
85 | public static InputStream getResourceAsStream(String resource) throws IOException {
86 | InputStream in = null;
87 | ClassLoader loader = ResourceUtils.class.getClassLoader();
88 | if (loader != null)
89 | in = loader.getResourceAsStream(resource);
90 | if (in == null)
91 | in = ClassLoader.getSystemResourceAsStream(resource);
92 | if (in == null)
93 | throw new IOException("Could not find resource " + resource);
94 | return in;
95 | }
96 |
97 |
98 | /** */
99 | /**
100 | * Returns a resource on the classpath as a Stream object
101 | *
102 | * @param loader
103 | * The classloader used to load the resource
104 | * @param resource
105 | * The resource to find
106 | * @throws IOException
107 | * If the resource cannot be found or read
108 | * @return The resource
109 | */
110 | public static InputStream getResourceAsStream(ClassLoader loader, String resource) throws IOException {
111 | InputStream in = null;
112 | if (loader != null)
113 | in = loader.getResourceAsStream(resource);
114 | if (in == null)
115 | in = ClassLoader.getSystemResourceAsStream(resource);
116 | if (in == null)
117 | throw new IOException("Could not find resource " + resource);
118 | return in;
119 | }
120 |
121 |
122 | /** */
123 | /**
124 | * Returns a resource on the classpath as a Properties object
125 | *
126 | * @param resource
127 | * The resource to find
128 | * @throws IOException
129 | * If the resource cannot be found or read
130 | * @return The resource
131 | */
132 | public static Properties getResourceAsProperties(String resource) throws IOException {
133 | Properties props = new Properties();
134 | InputStream in = null;
135 | String propfile = resource;
136 | in = getResourceAsStream(propfile);
137 | props.load(in);
138 | in.close();
139 | return props;
140 | }
141 |
142 |
143 | /** */
144 | /**
145 | * Returns a resource on the classpath as a Properties object
146 | *
147 | * @param loader
148 | * The classloader used to load the resource
149 | * @param resource
150 | * The resource to find
151 | * @throws IOException
152 | * If the resource cannot be found or read
153 | * @return The resource
154 | */
155 | public static Properties getResourceAsProperties(ClassLoader loader, String resource) throws IOException {
156 | Properties props = new Properties();
157 | InputStream in = null;
158 | String propfile = resource;
159 | in = getResourceAsStream(loader, propfile);
160 | props.load(in);
161 | in.close();
162 | return props;
163 | }
164 |
165 |
166 | /** */
167 | /**
168 | * Returns a resource on the classpath as a Reader object
169 | *
170 | * @param resource
171 | * The resource to find
172 | * @throws IOException
173 | * If the resource cannot be found or read
174 | * @return The resource
175 | */
176 | public static InputStreamReader getResourceAsReader(String resource) throws IOException {
177 | return new InputStreamReader(getResourceAsStream(resource));
178 | }
179 |
180 |
181 | /** */
182 | /**
183 | * Returns a resource on the classpath as a Reader object
184 | *
185 | * @param loader
186 | * The classloader used to load the resource
187 | * @param resource
188 | * The resource to find
189 | * @throws IOException
190 | * If the resource cannot be found or read
191 | * @return The resource
192 | */
193 | public static Reader getResourceAsReader(ClassLoader loader, String resource) throws IOException {
194 | return new InputStreamReader(getResourceAsStream(loader, resource));
195 | }
196 |
197 |
198 | /** */
199 | /**
200 | * Returns a resource on the classpath as a File object
201 | *
202 | * @param resource
203 | * The resource to find
204 | * @throws IOException
205 | * If the resource cannot be found or read
206 | * @return The resource
207 | */
208 | public static File getResourceAsFile(String resource) throws IOException {
209 | return new File(getResourceURL(resource).getFile());
210 | }
211 |
212 |
213 | /** */
214 | /**
215 | * Returns a resource on the classpath as a File object
216 | *
217 | * @param loader
218 | * The classloader used to load the resource
219 | * @param resource
220 | * The resource to find
221 | * @throws IOException
222 | * If the resource cannot be found or read
223 | * @return The resource
224 | */
225 | public static File getResourceAsFile(ClassLoader loader, String resource) throws IOException {
226 | return new File(getResourceURL(loader, resource).getFile());
227 | }
228 |
229 | }
--------------------------------------------------------------------------------
/diamond-utils/src/main/java/com/taobao/diamond/utils/ResourceUtils.java:
--------------------------------------------------------------------------------
1 | /*
2 | * (C) 2007-2012 Alibaba Group Holding Limited.
3 | *
4 | * This program is free software; you can redistribute it and/or modify
5 | * it under the terms of the GNU General Public License version 2 as
6 | * published by the Free Software Foundation.
7 | * Authors:
8 | * leiwen , boyan
9 | */
10 | package com.taobao.diamond.utils;
11 |
12 | import java.io.File;
13 | import java.io.IOException;
14 | import java.io.InputStream;
15 | import java.io.InputStreamReader;
16 | import java.io.Reader;
17 | import java.net.URL;
18 | import java.util.Properties;
19 |
20 |
21 | /**
22 | *
23 | * @author boyan
24 | * @date 2010-5-4
25 | */
26 | public class ResourceUtils extends Object {
27 |
28 | /** */
29 | /**
30 | * Returns the URL of the resource on the classpath
31 | *
32 | * @param resource
33 | * The resource to find
34 | * @throws IOException
35 | * If the resource cannot be found or read
36 | * @return The resource
37 | */
38 | public static URL getResourceURL(String resource) throws IOException {
39 | URL url = null;
40 | ClassLoader loader = ResourceUtils.class.getClassLoader();
41 | if (loader != null)
42 | url = loader.getResource(resource);
43 | if (url == null)
44 | url = ClassLoader.getSystemResource(resource);
45 | if (url == null)
46 | throw new IOException("Could not find resource " + resource);
47 | return url;
48 | }
49 |
50 |
51 | /** */
52 | /**
53 | * Returns the URL of the resource on the classpath
54 | *
55 | * @param loader
56 | * The classloader used to load the resource
57 | * @param resource
58 | * The resource to find
59 | * @throws IOException
60 | * If the resource cannot be found or read
61 | * @return The resource
62 | */
63 | public static URL getResourceURL(ClassLoader loader, String resource) throws IOException {
64 | URL url = null;
65 | if (loader != null)
66 | url = loader.getResource(resource);
67 | if (url == null)
68 | url = ClassLoader.getSystemResource(resource);
69 | if (url == null)
70 | throw new IOException("Could not find resource " + resource);
71 | return url;
72 | }
73 |
74 |
75 | /** */
76 | /**
77 | * Returns a resource on the classpath as a Stream object
78 | *
79 | * @param resource
80 | * The resource to find
81 | * @throws IOException
82 | * If the resource cannot be found or read
83 | * @return The resource
84 | */
85 | public static InputStream getResourceAsStream(String resource) throws IOException {
86 | InputStream in = null;
87 | ClassLoader loader = ResourceUtils.class.getClassLoader();
88 | if (loader != null)
89 | in = loader.getResourceAsStream(resource);
90 | if (in == null)
91 | in = ClassLoader.getSystemResourceAsStream(resource);
92 | if (in == null)
93 | throw new IOException("Could not find resource " + resource);
94 | return in;
95 | }
96 |
97 |
98 | /** */
99 | /**
100 | * Returns a resource on the classpath as a Stream object
101 | *
102 | * @param loader
103 | * The classloader used to load the resource
104 | * @param resource
105 | * The resource to find
106 | * @throws IOException
107 | * If the resource cannot be found or read
108 | * @return The resource
109 | */
110 | public static InputStream getResourceAsStream(ClassLoader loader, String resource) throws IOException {
111 | InputStream in = null;
112 | if (loader != null)
113 | in = loader.getResourceAsStream(resource);
114 | if (in == null)
115 | in = ClassLoader.getSystemResourceAsStream(resource);
116 | if (in == null)
117 | throw new IOException("Could not find resource " + resource);
118 | return in;
119 | }
120 |
121 |
122 | /** */
123 | /**
124 | * Returns a resource on the classpath as a Properties object
125 | *
126 | * @param resource
127 | * The resource to find
128 | * @throws IOException
129 | * If the resource cannot be found or read
130 | * @return The resource
131 | */
132 | public static Properties getResourceAsProperties(String resource) throws IOException {
133 | Properties props = new Properties();
134 | InputStream in = null;
135 | String propfile = resource;
136 | in = getResourceAsStream(propfile);
137 | props.load(in);
138 | in.close();
139 | return props;
140 | }
141 |
142 |
143 | /** */
144 | /**
145 | * Returns a resource on the classpath as a Properties object
146 | *
147 | * @param loader
148 | * The classloader used to load the resource
149 | * @param resource
150 | * The resource to find
151 | * @throws IOException
152 | * If the resource cannot be found or read
153 | * @return The resource
154 | */
155 | public static Properties getResourceAsProperties(ClassLoader loader, String resource) throws IOException {
156 | Properties props = new Properties();
157 | InputStream in = null;
158 | String propfile = resource;
159 | in = getResourceAsStream(loader, propfile);
160 | props.load(in);
161 | in.close();
162 | return props;
163 | }
164 |
165 |
166 | /** */
167 | /**
168 | * Returns a resource on the classpath as a Reader object
169 | *
170 | * @param resource
171 | * The resource to find
172 | * @throws IOException
173 | * If the resource cannot be found or read
174 | * @return The resource
175 | */
176 | public static InputStreamReader getResourceAsReader(String resource) throws IOException {
177 | return new InputStreamReader(getResourceAsStream(resource));
178 | }
179 |
180 |
181 | /** */
182 | /**
183 | * Returns a resource on the classpath as a Reader object
184 | *
185 | * @param loader
186 | * The classloader used to load the resource
187 | * @param resource
188 | * The resource to find
189 | * @throws IOException
190 | * If the resource cannot be found or read
191 | * @return The resource
192 | */
193 | public static Reader getResourceAsReader(ClassLoader loader, String resource) throws IOException {
194 | return new InputStreamReader(getResourceAsStream(loader, resource));
195 | }
196 |
197 |
198 | /** */
199 | /**
200 | * Returns a resource on the classpath as a File object
201 | *
202 | * @param resource
203 | * The resource to find
204 | * @throws IOException
205 | * If the resource cannot be found or read
206 | * @return The resource
207 | */
208 | public static File getResourceAsFile(String resource) throws IOException {
209 | return new File(getResourceURL(resource).getFile());
210 | }
211 |
212 |
213 | /** */
214 | /**
215 | * Returns a resource on the classpath as a File object
216 | *
217 | * @param loader
218 | * The classloader used to load the resource
219 | * @param resource
220 | * The resource to find
221 | * @throws IOException
222 | * If the resource cannot be found or read
223 | * @return The resource
224 | */
225 | public static File getResourceAsFile(ClassLoader loader, String resource) throws IOException {
226 | return new File(getResourceURL(loader, resource).getFile());
227 | }
228 |
229 | }
--------------------------------------------------------------------------------
/diamond-server/src/main/java/com/taobao/diamond/server/controller/ConfigController.java:
--------------------------------------------------------------------------------
1 | /*
2 | * (C) 2007-2012 Alibaba Group Holding Limited.
3 | *
4 | * This program is free software; you can redistribute it and/or modify
5 | * it under the terms of the GNU General Public License version 2 as
6 | * published by the Free Software Foundation.
7 | * Authors:
8 | * leiwen , boyan
9 | */
10 | package com.taobao.diamond.server.controller;
11 |
12 | import static com.taobao.diamond.common.Constants.LINE_SEPARATOR;
13 | import static com.taobao.diamond.common.Constants.WORD_SEPARATOR;
14 |
15 | import java.net.URLEncoder;
16 | import java.util.LinkedList;
17 | import java.util.List;
18 |
19 | import javax.servlet.http.HttpServletRequest;
20 | import javax.servlet.http.HttpServletResponse;
21 |
22 | import org.apache.commons.lang.StringUtils;
23 | import org.springframework.beans.factory.annotation.Autowired;
24 | import org.springframework.stereotype.Controller;
25 |
26 | import com.taobao.diamond.common.Constants;
27 | import com.taobao.diamond.server.service.ConfigService;
28 | import com.taobao.diamond.server.service.DiskService;
29 | import com.taobao.diamond.server.utils.GlobalCounter;
30 |
31 |
32 | /**
33 | * 处理配置信息获取和提交的controller
34 | *
35 | * @author boyan
36 | * @date 2010-5-4
37 | */
38 | @Controller
39 | public class ConfigController {
40 |
41 | @Autowired
42 | private ConfigService configService;
43 |
44 | @Autowired
45 | private DiskService diskService;
46 |
47 |
48 | public String getConfig(HttpServletRequest request, HttpServletResponse response, String dataId, String group) {
49 | response.setHeader("Content-Type", "text/html;charset=GBK");
50 | final String address = getRemortIP(request);
51 | if (address == null) {
52 | // 未找到远端地址，返回400错误
53 | response.setStatus(HttpServletResponse.SC_BAD_REQUEST);
54 | return "400";
55 | }
56 |
57 | if (GlobalCounter.getCounter().decrementAndGet() >= 0) {
58 | response.setStatus(HttpServletResponse.SC_SERVICE_UNAVAILABLE);
59 | return "503";
60 | }
61 |
62 | String md5 = this.configService.getContentMD5(dataId, group);
63 | if (md5 == null) {
64 | return "404";
65 | }
66 |
67 | response.setHeader(Constants.CONTENT_MD5, md5);
68 |
69 | // 正在被修改，返回304，这里的检查并没有办法保证一致性，因此做double-check尽力保证
70 | if (diskService.isModified(dataId, group)) {
71 | response.setStatus(HttpServletResponse.SC_NOT_MODIFIED);
72 | return "304";
73 | }
74 | String path = configService.getConfigInfoPath(dataId, group);
75 | // 再次检查
76 | if (diskService.isModified(dataId, group)) {
77 | response.setStatus(HttpServletResponse.SC_NOT_MODIFIED);
78 | return "304";
79 | }
80 | // 禁用缓存
81 | response.setHeader("Pragma", "no-cache");
82 | response.setDateHeader("Expires", 0);
83 | response.setHeader("Cache-Control", "no-cache,no-store");
84 | return "forward:" + path;
85 | }
86 |
87 |
88 | public String getProbeModifyResult(HttpServletRequest request, HttpServletResponse response, String probeModify) {
89 | response.setHeader("Content-Type", "text/html;charset=GBK");
90 | final String address = getRemortIP(request);
91 | if (address == null) {
92 | // 未找到远端地址，返回400错误
93 | response.setStatus(HttpServletResponse.SC_BAD_REQUEST);
94 | return "400";
95 | }
96 |
97 | if (GlobalCounter.getCounter().decrementAndGet() >= 0) {
98 | response.setStatus(HttpServletResponse.SC_SERVICE_UNAVAILABLE);
99 | return "503";
100 | }
101 |
102 | final List configKeyList = getConfigKeyList(probeModify);
103 |
104 | StringBuilder resultBuilder = new StringBuilder();
105 | for (ConfigKey key : configKeyList) {
106 | String md5 = this.configService.getContentMD5(key.getDataId(), key.getGroup());
107 | if (!StringUtils.equals(md5, key.getMd5())) {
108 | resultBuilder.append(key.getDataId()).append(WORD_SEPARATOR).append(key.getGroup())
109 | .append(LINE_SEPARATOR);
110 | }
111 | }
112 |
113 | String returnHeader = resultBuilder.toString();
114 | try {
115 | returnHeader = URLEncoder.encode(resultBuilder.toString(), "UTF-8");
116 | }
117 | catch (Exception e) {
118 | // ignore
119 | }
120 |
121 | request.setAttribute("content", returnHeader);
122 | // 禁用缓存
123 | response.setHeader("Pragma", "no-cache");
124 | response.setDateHeader("Expires", 0);
125 | response.setHeader("Cache-Control", "no-cache,no-store");
126 | return "200";
127 | }
128 |
129 |
130 | public ConfigService getConfigService() {
131 | return configService;
132 | }
133 |
134 |
135 | public void setConfigService(ConfigService configService) {
136 | this.configService = configService;
137 | }
138 |
139 |
140 | public DiskService getDiskService() {
141 | return diskService;
142 | }
143 |
144 |
145 | public void setDiskService(DiskService diskService) {
146 | this.diskService = diskService;
147 | }
148 |
149 |
150 | /**
151 | * 查找真实的IP地址
152 | *
153 | * @param request
154 | * @return
155 | */
156 | public String getRemortIP(HttpServletRequest request) {
157 | if (request.getHeader("x-forwarded-for") == null) {
158 | return request.getRemoteAddr();
159 | }
160 | return request.getHeader("x-forwarded-for");
161 | }
162 |
163 |
164 | public static List getConfigKeyList(String configKeysString) {
165 | List configKeyList = new LinkedList();
166 | if (null == configKeysString || "".equals(configKeysString)) {
167 | return configKeyList;
168 | }
169 | String[] configKeyStrings = configKeysString.split(LINE_SEPARATOR);
170 | for (String configKeyString : configKeyStrings) {
171 | String[] configKey = configKeyString.split(WORD_SEPARATOR);
172 | if (configKey.length > 3) {
173 | continue;
174 | }
175 | ConfigKey key = new ConfigKey();
176 | if ("".equals(configKey[0])) {
177 | continue;
178 | }
179 | key.setDataId(configKey[0]);
180 | if (configKey.length >= 2 && !"".equals(configKey[1])) {
181 | key.setGroup(configKey[1]);
182 | }
183 | if (configKey.length == 3 && !"".equals(configKey[2])) {
184 | key.setMd5(configKey[2]);
185 | }
186 | configKeyList.add(key);
187 | }
188 |
189 | return configKeyList;
190 | }
191 |
192 | public static class ConfigKey {
193 | private String dataId;
194 | private String group;
195 | private String md5;
196 |
197 |
198 | public String getDataId() {
199 | return dataId;
200 | }
201 |
202 |
203 | public void setDataId(String dataId) {
204 | this.dataId = dataId;
205 | }
206 |
207 |
208 | public String getGroup() {
209 | return group;
210 | }
211 |
212 |
213 | public void setGroup(String group) {
214 | this.group = group;
215 | }
216 |
217 |
218 | public String getMd5() {
219 | return md5;
220 | }
221 |
222 |
223 | public void setMd5(String md5) {
224 | this.md5 = md5;
225 | }
226 |
227 |
228 | @Override
229 | public String toString() {
230 | StringBuilder sb = new StringBuilder();
231 | sb.append("DataID: ").append(dataId).append("\r\n");
232 | sb.append("Group: ").append((null == group ? "" : group)).append("\r\n");
233 | sb.append("MD5: ").append((null == md5 ? "" : md5)).append("\r\n");
234 | return sb.toString();
235 | }
236 | }
237 | }
238 |
--------------------------------------------------------------------------------
/diamond-utils/src/main/java/com/taobao/diamond/io/watch/WatchKey.java:
--------------------------------------------------------------------------------
1 | /*
2 | * (C) 2007-2012 Alibaba Group Holding Limited.
3 | *
4 | * This program is free software; you can redistribute it and/or modify
5 | * it under the terms of the GNU General Public License version 2 as
6 | * published by the Free Software Foundation.
7 | * Authors:
8 | * leiwen , boyan
9 | */
10 | package com.taobao.diamond.io.watch;
11 |
12 | import java.io.File;
13 | import java.util.HashSet;
14 | import java.util.Iterator;
15 | import java.util.LinkedList;
16 | import java.util.List;
17 | import java.util.Set;
18 |
19 | import com.taobao.diamond.io.Path;
20 | import com.taobao.diamond.io.watch.util.PathNode;
21 |
22 |
23 | /**
24 | * WatchKey，表示一个注册的的凭证
25 | *
26 | * @author boyan
27 | * @date 2010-5-4
28 | */
29 | public class WatchKey {
30 |
31 | private volatile boolean valid;
32 |
33 | private final PathNode root;
34 |
35 | private List> changedEvents;
36 |
37 | private final Set> filterSet = new HashSet>();
38 |
39 | private final WatchService watcher;
40 |
41 |
42 | public WatchKey(final Path path, final WatchService watcher, boolean fireCreatedEventOnIndex,
43 | WatchEvent.Kind... events) {
44 | valid = true;
45 | this.watcher = watcher;
46 | // 建立内存索引
47 | this.root = new PathNode(path, true);
48 | if (events != null) {
49 | for (WatchEvent.Kind event : events) {
50 | filterSet.add(event);
51 | }
52 | }
53 | LinkedList> changedEvents = new LinkedList>();
54 | index(this.root, fireCreatedEventOnIndex, changedEvents);
55 | this.changedEvents = changedEvents;
56 | }
57 |
58 |
59 | /**
60 | * 索引目录
61 | *
62 | * @param node
63 | */
64 | private void index(PathNode node, boolean fireCreatedEventOnIndex, LinkedList> changedEvents) {
65 | File file = node.getPath().getFile();
66 | if (!file.isDirectory()) {
67 | return;
68 | }
69 | File[] subFiles = file.listFiles();
70 | if (subFiles != null) {
71 | for (File subFile : subFiles) {
72 | PathNode subNode = new PathNode(new Path(subFile), false);
73 | if (fireCreatedEventOnIndex) {
74 | changedEvents.add(new WatchEvent(StandardWatchEventKind.ENTRY_CREATE, 1, subNode.getPath()));
75 | }
76 | node.addChild(subNode);
77 | if (subNode.getPath().isDirectory()) {
78 | index(subNode, fireCreatedEventOnIndex, changedEvents);
79 | }
80 | }
81 | }
82 | }
83 |
84 |
85 | public void cancel() {
86 | this.valid = false;
87 | }
88 |
89 |
90 | @Override
91 | public String toString() {
92 | return "WatchKey [root=" + root + ", valid=" + valid + "]";
93 | }
94 |
95 |
96 | public boolean isValid() {
97 | return valid && root != null;
98 | }
99 |
100 |
101 | public List> pollEvents() {
102 | if (changedEvents != null) {
103 | List> result = changedEvents;
104 | changedEvents = null;
105 | return result;
106 | }
107 | return null;
108 | }
109 |
110 |
111 | /**
112 | * 检测是否有变化
113 | *
114 | * @return
115 | */
116 | boolean check() {
117 | if (this.changedEvents != null && this.changedEvents.size() > 0)
118 | return true;
119 | if (!this.valid)
120 | return false;
121 | List> list = new LinkedList>();
122 | if (check(root, list)) {
123 | this.changedEvents = list;
124 | return true;
125 | }
126 | else {
127 | return false;
128 | }
129 | }
130 |
131 |
132 | private boolean check(PathNode node, List> changedEvents) {
133 | Path nodePath = node.getPath();
134 | File nodeNewFile = new File(nodePath.getAbsolutePath());
135 | if (nodePath != null) {
136 | if (node.isRoot()) {
137 | if (!nodeNewFile.exists())
138 | return fireOnRootDeleted(changedEvents, nodeNewFile);
139 | else {
140 | return checkNodeChildren(node, changedEvents, nodeNewFile);
141 | }
142 | }
143 | else {
144 | return checkNodeChildren(node, changedEvents, nodeNewFile);
145 | }
146 | }
147 | else
148 | throw new IllegalStateException("PathNode没有path");
149 | }
150 |
151 |
152 | private boolean checkNodeChildren(PathNode node, List> changedEvents, File nodeNewFile) {
153 | boolean changed = false;
154 | Iterator it = node.getChildren().iterator();
155 | // 用于判断是否有新增文件或者目录的现有名称集合
156 | Set childNameSet = new HashSet();
157 | while (it.hasNext()) {
158 | PathNode child = it.next();
159 | Path childPath = child.getPath();
160 | childNameSet.add(childPath.getName());
161 | File childNewFile = new File(childPath.getAbsolutePath());
162 | // 1、判断文件是否还存在
163 | if (!childNewFile.exists() && filterSet.contains(StandardWatchEventKind.ENTRY_DELETE)) {
164 | changed = true;
165 | changedEvents.add(new WatchEvent(StandardWatchEventKind.ENTRY_DELETE, 1, childPath));
166 | it.remove();// 移除节点
167 | }
168 | // 2、如果是文件，判断是否被修改
169 | if (childPath.isFile()) {
170 | if (checkFile(changedEvents, child, childNewFile) && !changed) {
171 | changed = true;
172 | }
173 |
174 | }
175 | // 3、递归检测目录
176 | if (childPath.isDirectory()) {
177 | if (check(child, changedEvents) && !changed) {
178 | changed = true;
179 | }
180 | }
181 | }
182 |
183 | // 查看是否有新增文件
184 | File[] newChildFiles = nodeNewFile.listFiles();
185 | if(newChildFiles!=null)
186 | for (File newChildFile : newChildFiles) {
187 | if (!childNameSet.contains(newChildFile.getName())
188 | && filterSet.contains(StandardWatchEventKind.ENTRY_CREATE)) {
189 | changed = true;
190 | Path newChildPath = new Path(newChildFile);
191 | changedEvents.add(new WatchEvent(StandardWatchEventKind.ENTRY_CREATE, 1, newChildPath));
192 | PathNode newSubNode = new PathNode(newChildPath, false);
193 | node.addChild(newSubNode);// 新增子节点
194 | // 如果是目录，递归调用
195 | if (newChildFile.isDirectory()) {
196 | checkNodeChildren(newSubNode, changedEvents, newChildFile);
197 | }
198 | }
199 | }
200 | return changed;
201 | }
202 |
203 |
204 | private boolean checkFile(List> changedEvents, PathNode child, File childNewFile) {
205 | boolean changed = false;
206 | // 查看文件是否被修改
207 | if (childNewFile.lastModified() != child.lastModified()
208 | && filterSet.contains(StandardWatchEventKind.ENTRY_MODIFY)) {
209 | changed = true;
210 | Path newChildPath = new Path(childNewFile);
211 | changedEvents.add(new WatchEvent(StandardWatchEventKind.ENTRY_MODIFY, 1, newChildPath));
212 | child.setPath(newChildPath);// 更新path
213 | }
214 | return changed;
215 | }
216 |
217 |
218 | private boolean fireOnRootDeleted(List> changedEvents, File nodeNewFile) {
219 | this.valid = false;
220 | if (filterSet.contains(StandardWatchEventKind.ENTRY_DELETE)) {
221 | changedEvents.add(new WatchEvent(StandardWatchEventKind.ENTRY_DELETE, 1, new Path(nodeNewFile)));
222 | return true;
223 | }
224 | return false;
225 | }
226 |
227 |
228 | public boolean reset() {
229 | if (!valid)
230 | return false;
231 | if (root == null)
232 | return false;
233 | return this.watcher.resetKey(this);
234 | }
235 | }
236 |
--------------------------------------------------------------------------------
/pom.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 | 4.0.0
5 | 2010
6 | com.taobao.diamond
7 | diamond-all
8 | pom
9 | 2.0.5.4.taocode-SNAPSHOT
10 | taobao diamond v${project.version}
11 |
12 |
13 | UTF-8
14 |
15 |
16 |
17 | diamond-client
18 | diamond-sdk
19 | diamond-server
20 | diamond-utils
21 |
22 |
23 |
24 |
25 |
26 | com.atlassian.maven.plugins
27 | maven-clover2-plugin
28 |
29 | clover.license
30 | true
31 |
32 | **/notjunit/*.java
33 |
34 | true
35 | true
36 |
37 |
38 |
39 | org.apache.maven.plugins
40 | maven-eclipse-plugin
41 | 2.5.1
42 |
43 | true
44 | false
45 |
46 |
47 |
48 | org.apache.maven.plugins
49 | maven-surefire-plugin
50 | 2.3
51 |
52 | -Xms512m -Xmx1024m
53 | always
54 |
55 | **/*Test.java
56 |
57 |
58 |
59 |
60 | org.apache.maven.plugins
61 | maven-site-plugin
62 |
63 | zh_CN
64 | GBK
65 | GBK
66 |
67 |
68 |
69 |
70 | org.apache.maven.plugins
71 | maven-release-plugin
72 | 2.0-beta-9
73 |
74 | http://www.xx.com/repos/diamond/tags
75 |
76 |
77 |
78 |
79 |
80 |
81 |
82 |
83 |
84 | ${pom.groupId}
85 | diamond-client
86 | ${project.version}
87 |
88 |
89 | ${pom.groupId}
90 | diamond-utils
91 | ${project.version}
92 |
93 |
94 | javax.servlet
95 | servlet-api
96 | 2.5
97 | provided
98 |
99 |
100 | jstl
101 | jstl
102 | 1.1.2
103 |
104 |
105 | taglibs
106 | standard
107 | 1.1.2
108 |
109 |
110 | org.springframework
111 | spring-jdbc
112 | 3.1.1.RELEASE
113 |
114 |
115 | org.springframework
116 | spring-webmvc
117 | 3.1.1.RELEASE
118 |
119 |
120 |
121 | mysql
122 | mysql-connector-java
123 | 5.1.5
124 |
125 |
126 |
127 | log4j
128 | log4j
129 | 1.2.14
130 |
131 |
132 | commons-logging
133 | commons-logging
134 | 1.1
135 |
136 |
137 | commons-httpclient
138 | commons-httpclient
139 | 3.1
140 |
141 |
142 | commons-dbcp
143 | commons-dbcp
144 | 1.2.2
145 |
146 |
147 | commons-io
148 | commons-io
149 | 1.4
150 |
151 |
152 | commons-lang
153 | commons-lang
154 | 2.6
155 |
156 |
157 |
158 | org.codehaus.jackson
159 | jackson-core-lgpl
160 | 1.4.0
161 |
162 |
163 | org.codehaus.jackson
164 | jackson-mapper-lgpl
165 | 1.4.0
166 |
167 |
168 |
169 | junit
170 | junit
171 | 4.4
172 | test
173 |
174 |
175 |
176 |
177 |
178 |
179 |
180 | org.apache.maven.plugins
181 | maven-javadoc-plugin
182 | 2.5
183 |
184 | GBK
185 | GBK
186 |
187 |
188 |
189 |
190 | javadoc
191 |
192 |
193 |
194 |
195 |
196 |
197 | org.apache.maven.plugins
198 | maven-surefire-report-plugin
199 | 2.4.2
200 |
201 |
202 | org.apache.maven.plugins
203 | maven-checkstyle-plugin
204 | 2.3
205 |
206 | checkstyle.xml
207 |
208 |
209 |
210 | org.apache.maven.plugins
211 | maven-clover-plugin
212 |
213 | clover.license
214 | true
215 | true
216 | true
217 |
218 |
219 |
220 | org.apache.maven.plugins
221 | maven-pmd-plugin
222 |
223 | true
224 | 1.6
225 |
226 |
227 |
228 | org.codehaus.mojo
229 | findbugs-maven-plugin
230 | 2.0.1
231 |
232 | true
233 | true
234 | true
235 |
236 |
237 |
238 | maven-compiler-plugin
239 |
240 | 1.7
241 | 1.7
242 | UTF-8
243 |
244 |
245 |
246 |
247 |
248 |
249 |
250 |
251 |
252 |
253 |
254 |
255 |
256 |
257 |
258 |
259 |
260 |
261 |
262 |
--------------------------------------------------------------------------------
/diamond-client/src/main/java/com/taobao/diamond/client/processor/LocalConfigInfoProcessor.java:
--------------------------------------------------------------------------------
1 | /*
2 | * (C) 2007-2012 Alibaba Group Holding Limited.
3 | *
4 | * This program is free software; you can redistribute it and/or modify
5 | * it under the terms of the GNU General Public License version 2 as
6 | * published by the Free Software Foundation.
7 | * Authors:
8 | * leiwen , boyan
9 | */
10 | package com.taobao.diamond.client.processor;
11 |
12 | import java.io.File;
13 | import java.io.IOException;
14 | import java.util.HashMap;
15 | import java.util.HashSet;
16 | import java.util.Map;
17 | import java.util.Set;
18 | import java.util.concurrent.Executors;
19 | import java.util.concurrent.ScheduledExecutorService;
20 |
21 | import org.apache.commons.logging.Log;
22 | import org.apache.commons.logging.LogFactory;
23 |
24 | import com.taobao.diamond.common.Constants;
25 | import com.taobao.diamond.configinfo.CacheData;
26 | import com.taobao.diamond.io.FileSystem;
27 | import com.taobao.diamond.io.Path;
28 | import com.taobao.diamond.io.watch.StandardWatchEventKind;
29 | import com.taobao.diamond.io.watch.WatchEvent;
30 | import com.taobao.diamond.io.watch.WatchKey;
31 | import com.taobao.diamond.io.watch.WatchService;
32 | import com.taobao.diamond.utils.FileUtils;
33 |
34 |
35 | public class LocalConfigInfoProcessor {
36 | private static final Log log = LogFactory.getLog(LocalConfigInfoProcessor.class);
37 | private ScheduledExecutorService singleExecutor = Executors.newSingleThreadScheduledExecutor();;
38 |
39 | private final Map existFiles = new HashMap();
40 |
41 | private volatile boolean isRun;
42 | private String rootPath = null;
43 |
44 |
45 | /**
46 | * 获取本地配置
47 | *
48 | * @param cacheData
49 | * @param force
50 | * 强制获取，在没有变更的时候不返回null
51 | * @return
52 | * @throws IOException
53 | */
54 | public String getLocalConfigureInfomation(CacheData cacheData, boolean force) throws IOException {
55 | String filePath = getFilePath(cacheData.getDataId(), cacheData.getGroup());
56 | if (!existFiles.containsKey(filePath)) {
57 | if (cacheData.isUseLocalConfigInfo()) {
58 | cacheData.setLastModifiedHeader(Constants.NULL);
59 | cacheData.setMd5(Constants.NULL);
60 | cacheData.setLocalConfigInfoFile(null);
61 | cacheData.setLocalConfigInfoVersion(0L);
62 | cacheData.setUseLocalConfigInfo(false);
63 | }
64 | return null;
65 | }
66 | if (force) {
67 | log.info("主动从本地获取配置数据, dataId:" + cacheData.getDataId() + ", group:" + cacheData.getGroup());
68 |
69 | String content = FileUtils.getFileContent(filePath);
70 | return content;
71 | }
72 | // 判断是否变更，没有变更，返回null
73 | if (!filePath.equals(cacheData.getLocalConfigInfoFile())
74 | || existFiles.get(filePath) != cacheData.getLocalConfigInfoVersion()) {
75 | String content = FileUtils.getFileContent(filePath);
76 | cacheData.setLocalConfigInfoFile(filePath);
77 | cacheData.setLocalConfigInfoVersion(existFiles.get(filePath));
78 | cacheData.setUseLocalConfigInfo(true);
79 |
80 | if (log.isInfoEnabled()) {
81 | log.info("本地配置数据发生变化, dataId:" + cacheData.getDataId() + ", group:" + cacheData.getGroup());
82 | }
83 |
84 | return content;
85 | }
86 | else {
87 | cacheData.setUseLocalConfigInfo(true);
88 |
89 | if (log.isInfoEnabled()) {
90 | log.debug("本地配置数据没有发生变化, dataId:" + cacheData.getDataId() + ", group:" + cacheData.getGroup());
91 | }
92 |
93 | return null;
94 | }
95 | }
96 |
97 |
98 | String getFilePath(String dataId, String group) {
99 | StringBuilder filePathBuilder = new StringBuilder();
100 | filePathBuilder.append(rootPath).append("/").append(Constants.BASE_DIR).append("/").append(group).append("/")
101 | .append(dataId);
102 | File file = new File(filePathBuilder.toString());
103 | return file.getAbsolutePath();
104 | }
105 |
106 |
107 | public synchronized void start(String rootPath) {
108 | if (this.isRun) {
109 | return;
110 | }
111 | this.rootPath = rootPath;
112 | this.isRun = true;
113 | if (this.singleExecutor == null || singleExecutor.isTerminated()) {
114 | singleExecutor = Executors.newSingleThreadScheduledExecutor();
115 | }
116 | initDataDir(rootPath);
117 | startCheckLocalDir(rootPath);
118 | }
119 |
120 |
121 | private void initDataDir(String rootPath) {
122 | try {
123 | File flie = new File(rootPath);
124 | flie.mkdir();
125 | }
126 | catch (Exception e) {
127 | }
128 | }
129 |
130 |
131 | public synchronized void stop() {
132 | if (!this.isRun) {
133 | return;
134 | }
135 | this.isRun = false;
136 | this.singleExecutor.shutdownNow();
137 | this.singleExecutor = null;
138 | }
139 |
140 |
141 | private void startCheckLocalDir(final String filePath) {
142 | final WatchService watcher = FileSystem.getDefault().newWatchService();
143 |
144 | Path path = new Path(new File(filePath));
145 | // 注册事件
146 | watcher.register(path, true, StandardWatchEventKind.ENTRY_CREATE, StandardWatchEventKind.ENTRY_DELETE,
147 | StandardWatchEventKind.ENTRY_MODIFY);
148 | // 第一次运行，主动check
149 | checkAtFirst(watcher);
150 | singleExecutor.execute(new Runnable() {
151 | public void run() {
152 | log.debug(">>>>>>已经开始监控目录<<<<<<");
153 | // 无限循环等待事件
154 | while (isRun) {
155 | // 凭证
156 | WatchKey key;
157 | try {
158 | key = watcher.take();
159 | }
160 | catch (InterruptedException x) {
161 | continue;
162 | }
163 | // reset，如果无效，跳出循环,无效可能是监听的目录被删除
164 | if (!processEvents(key)) {
165 | log.error("reset unvalid,监控服务失效");
166 | break;
167 | }
168 | }
169 | log.debug(">>>>>>退出监控目录<<<<<<");
170 | watcher.close();
171 |
172 | }
173 |
174 | });
175 | }
176 |
177 |
178 | private void checkAtFirst(final WatchService watcher) {
179 | watcher.check();
180 | WatchKey key = null;
181 | while ((key = watcher.poll()) != null) {
182 | processEvents(key);
183 | }
184 | }
185 |
186 |
187 | /**
188 | * 处理触发的事件
189 | *
190 | * @param key
191 | * @return
192 | */
193 | @SuppressWarnings({ "unchecked" })
194 | private boolean processEvents(WatchKey key) {
195 | /**
196 | * 获取事件集合
197 | */
198 | for (WatchEvent event : key.pollEvents()) {
199 | // 事件的类型
200 | // WatchEvent.Kind kind = event.kind();
201 |
202 | // 通过context方法得到发生事件的path
203 | WatchEvent ev = (WatchEvent) event;
204 | Path eventPath = ev.context();
205 |
206 | String realPath = eventPath.getAbsolutePath();
207 | if (ev.kind() == StandardWatchEventKind.ENTRY_CREATE || ev.kind() == StandardWatchEventKind.ENTRY_MODIFY) {
208 |
209 | String grandpaDir = null;
210 | try {
211 | grandpaDir = FileUtils.getGrandpaDir(realPath);
212 | }
213 | catch (Exception e1) {
214 |
215 | }
216 | if (!Constants.BASE_DIR.equals(grandpaDir)) {
217 | log.error("无效的文件进入监控目录: " + realPath);
218 | continue;
219 | }
220 | existFiles.put(realPath, System.currentTimeMillis());
221 | if (log.isInfoEnabled()) {
222 | log.info(realPath + "文件被添加或更新");
223 | }
224 | }
225 | else if (ev.kind() == StandardWatchEventKind.ENTRY_DELETE) {
226 | String grandpaDir = null;
227 | try {
228 | grandpaDir = FileUtils.getGrandpaDir(realPath);
229 | }
230 | catch (Exception e1) {
231 |
232 | }
233 | if (Constants.BASE_DIR.equals(grandpaDir)) {
234 | // 删除的是文件
235 | existFiles.remove(realPath);
236 | if (log.isInfoEnabled()) {
237 | log.info(realPath + "文件被被删除");
238 | }
239 | }
240 | else {
241 | // 删除的是目录
242 | Set keySet = new HashSet(existFiles.keySet());
243 | for (String filePath : keySet) {
244 | if (filePath.startsWith(realPath)) {
245 | existFiles.remove(filePath);
246 | if (log.isInfoEnabled()) {
247 | log.info(filePath + "文件被删除");
248 | }
249 | }
250 | }
251 |
252 | }
253 | }
254 | }
255 | return key.reset();
256 | }
257 | }
258 |
--------------------------------------------------------------------------------