Zookeeper权限控制

 

1. package org.I0Itec.zkclient; 
2.  
3. import java.util.ArrayList; 
4. import java.util.List; 
5.  
6. import org.apache.zookeeper.WatchedEvent; 
7. import org.apache.zookeeper.Watcher; 
8. import org.apache.zookeeper.ZooDefs.Ids; 
9. import org.apache.zookeeper.data.ACL; 
10.  
11. /** 
12. * Description: ZooKeepre ACL权限控制 测试 
13. * @author nileader / nileader@gmail.com 
14. * @Date Feb 2, 2012 
15. */ 
16. public class DemoAuth implements Watcher { 
17.  
18. final static String SERVER_LIST = “127.0.0.1:4711″; 
19.  
20. final static String PATH = “/yinshi_auth_test”; 
21. final static String PATH_DEL = “/yinshi_auth_test/will_be_del”; 
22.  
23. final static String authentication_type = “digest”; 
24.  
25. final static String correctAuthentication = “taokeeper:true”; 
26. final static String badAuthentication = “taokeeper:errorCode”; 
27.  
28. static ZkClient zkClient = null; 
29.  
30. public static void main( String[] args ) throws Exception { 
31.  
32. List< ACL > acls = new ArrayList< ACL >( 1 ); 
33. for ( ACL ids_acl : Ids.CREATOR_ALL_ACL ) { 
34. acls.add( ids_acl ); 
35. } 
36.  
37. try { 
38. zkClient = new ZkClient( SERVER_LIST, 50000); 
39. zkClient.addAuthInfo( authentication_type, correctAuthentication.getBytes() ); 
40. } catch ( Exception e ) { 
41. // TODO Auto-generated catch block 
42. e.printStackTrace(); 
43. } 
44.  
45. try { 
46. zkClient.createPersistent( PATH, acls, “init content” ); 
47. System.out.println( “使用授权key：” + correctAuthentication + “创建节点：” + PATH + “, 初始内容是: init content” ); 
48. } catch ( Exception e ) { 
49. e.printStackTrace(); 
50. } 
51. try { 
52. zkClient.createPersistent( PATH_DEL, acls, “待删节点” ); 
53. System.out.println( “使用授权key：” + correctAuthentication + “创建节点：” + PATH_DEL + “, 初始内容是: init content” ); 
54. } catch ( Exception e ) { 
55. // TODO Auto-generated catch block 
56. e.printStackTrace(); 
57. } 
58.  
59. // 获取数据 
60. getDataByNoAuthentication(); 
61. getDataByBadAuthentication(); 
62. getDataByCorrectAuthentication(); 
63.  
64. // 更新数据 
65. updateDataByNoAuthentication(); 
66. updateDataByBadAuthentication(); 
67. updateDataByCorrectAuthentication(); 
68.  
69. // 获取数据 
70. getDataByNoAuthentication(); 
71. getDataByBadAuthentication(); 
72. getDataByCorrectAuthentication(); 
73.  
74. //删除数据 
75. deleteNodeByBadAuthentication(); 
76. deleteNodeByNoAuthentication(); 
77. deleteNodeByCorrectAuthentication(); 
78.  
79. deleteParent(); 
80.  
81. zkClient.close(); 
82. } 
83.  
84. /** 获取数据：采用错误的密码 */ 
85. static void getDataByBadAuthentication() { 
86. String prefix = “[使用错误的授权信息]“; 
87. try { 
88. System.out.println( prefix + “获取数据：” + PATH ); 
89. zkClient = new ZkClient( SERVER_LIST, 50000); 
90. zkClient.addAuthInfo( authentication_type, badAuthentication.getBytes() ); 
91. System.out.println( prefix + “成功获取数据：” + zkClient.readData( PATH ) ); 
92. } catch ( Exception e ) { 
93. System.err.println( prefix + “获取数据失败，原因：” + e.getMessage() ); 
94. } 
95. } 
96.  
97. /** 获取数据：不采用密码 */ 
98. static void getDataByNoAuthentication() { 
99. String prefix = “[不使用任何授权信息]“; 
100. try { 
101. System.out.println( prefix + “获取数据：” + PATH ); 
102. zkClient = new ZkClient( SERVER_LIST, 50000); 
103. /*
104. zkClient.addAuthInfo( authentication_type, correctAuthentication.getBytes() ); 
105. */
106. System.out.println( prefix + “成功获取数据：” + zkClient.readData( PATH ) ); 
107. } catch ( Exception e ) { 
108. System.err.println( prefix + “获取数据失败，原因：” + e.getMessage() ); 
109. } 
110. } 
111.  
112. /** 采用正确的密码 */ 
113. static void getDataByCorrectAuthentication() { 
114. String prefix = “[使用正确的授权信息]“; 
115. try { 
116. System.out.println( prefix + “获取数据：” + PATH ); 
117. zkClient = new ZkClient( SERVER_LIST, 50000); 
118. zkClient.addAuthInfo( authentication_type, correctAuthentication.getBytes() ); 
119. System.out.println( prefix + “成功获取数据：” + zkClient.readData( PATH ) ); 
120. } catch ( Exception e ) { 
121. System.out.println( prefix + “获取数据失败，原因：” + e.getMessage() ); 
122. } 
123. } 
124.  
125. /** 
126. * 更新数据：不采用密码 
127. */ 
128. static void updateDataByNoAuthentication() { 
129.  
130. String prefix = “[不使用任何授权信息]“; 
131.  
132. System.out.println( prefix + “更新数据： ” + PATH ); 
133. try { 
134. zkClient = new ZkClient( SERVER_LIST, 50000); 
135. if( zkClient.exists( PATH ) ){ 
136. zkClient.writeData( PATH, prefix ); 
137. System.out.println( prefix + “更新成功” ); 
138. } 
139. } catch ( Exception e ) { 
140. System.err.println( prefix + “更新失败，原因是：” + e.getMessage() ); 
141. } 
142. } 
143.  
144. /** 
145. * 更新数据：采用错误的密码 
146. */ 
147. static void updateDataByBadAuthentication() { 
148.  
149. String prefix = “[使用错误的授权信息]“; 
150.  
151. System.out.println( prefix + “更新数据：” + PATH ); 
152. try { 
153. zkClient = new ZkClient( SERVER_LIST, 50000); 
154. zkClient.addAuthInfo( authentication_type, badAuthentication.getBytes() ); 
155. if( zkClient.exists( PATH ) ){ 
156. zkClient.writeData( PATH, prefix ); 
157. System.out.println( prefix + “更新成功” ); 
158. } 
159. } catch ( Exception e ) { 
160. System.err.println( prefix + “更新失败，原因是：” + e.getMessage() ); 
161. } 
162. } 
163.  
164. /** 
165. * 更新数据：采用正确的密码 
166. */ 
167. static void updateDataByCorrectAuthentication() { 
168.  
169. String prefix = “[使用正确的授权信息]“; 
170.  
171. System.out.println( prefix + “更新数据：” + PATH ); 
172. try { 
173. zkClient = new ZkClient( SERVER_LIST, 50000); 
174. zkClient.addAuthInfo( authentication_type, correctAuthentication.getBytes() ); 
175. if( zkClient.exists( PATH ) ){ 
176. zkClient.writeData( PATH, prefix ); 
177. System.out.println( prefix + “更新成功” ); 
178. } 
179. } catch ( Exception e ) { 
180. System.err.println( prefix + “更新失败，原因是：” + e.getMessage() ); 
181. } 
182. } 
183.  
184. /** 
185. * 不使用密码 删除节点 
186. */ 
187. static void deleteNodeByNoAuthentication() throws Exception { 
188.  
189. String prefix = “[不使用任何授权信息]“; 
190.  
191. try { 
192. System.out.println( prefix + “删除节点：” + PATH_DEL ); 
193. zkClient = new ZkClient( SERVER_LIST, 50000); 
194. if( zkClient.exists( PATH_DEL ) ){ 
195. zkClient.delete( PATH_DEL ); 
196. System.out.println( prefix + “删除成功” ); 
197. } 
198. } catch ( Exception e ) { 
199. System.err.println( prefix + “删除失败，原因是：” + e.getMessage() ); 
200. } 
201. } 
202.  
203. /** 
204. * 采用错误的密码删除节点 
205. */ 
206. static void deleteNodeByBadAuthentication() throws Exception { 
207.  
208. String prefix = “[使用错误的授权信息]“; 
209.  
210. try { 
211. System.out.println( prefix + “删除节点：” + PATH_DEL ); 
212. zkClient = new ZkClient( SERVER_LIST, 50000); 
213. zkClient.addAuthInfo( authentication_type, badAuthentication.getBytes() ); 
214. if( zkClient.exists( PATH_DEL ) ){ 
215. zkClient.delete( PATH_DEL ); 
216. System.out.println( prefix + “删除成功” ); 
217. } 
218. } catch ( Exception e ) { 
219. System.err.println( prefix + “删除失败，原因是：” + e.getMessage() ); 
220. } 
221. } 
222.  
223. /** 
224. * 使用正确的密码删除节点 
225. */ 
226. static void deleteNodeByCorrectAuthentication() throws Exception { 
227.  
228. String prefix = “[使用正确的授权信息]“; 
229.  
230. try { 
231. System.out.println( prefix + “删除节点：” + PATH_DEL ); 
232. zkClient = new ZkClient( SERVER_LIST, 50000); 
233. zkClient.addAuthInfo( authentication_type, correctAuthentication.getBytes() ); 
234. if( zkClient.exists( PATH_DEL ) ){ 
235. zkClient.delete( PATH_DEL ); 
236. System.out.println( prefix + “删除成功” ); 
237. } 
238. } catch ( Exception e ) { 
239. System.out.println( prefix + “删除失败，原因是：” + e.getMessage() ); 
240. } 
241. } 
242.  
243. /** 
244. * 使用正确的密码删除节点 
245. */ 
246. static void deleteParent() throws Exception { 
247. try { 
248. zkClient = new ZkClient( SERVER_LIST, 50000); 
249. zkClient.addAuthInfo( authentication_type, correctAuthentication.getBytes() ); 
250. if( zkClient.exists( PATH ) ){ 
251. zkClient.delete( PATH ); 
252. } 
253. } catch ( Exception e ) { 
254. e.printStackTrace(); 
255. } 
256. } 
257.  
258. @Override 
259. public void process( WatchedEvent event ) { 
260. // TODO Auto-generated method stub 
261.  
262. } 
263.  
264. }