HashMap緩衝測試

    昨天做了個HashMap緩衝的測試，但這個測試代碼應該還有點問題，我不知道應該怎麼改，所以發上來給大家看看，希望有點協助！

1. package motor.sql;
2. import java.sql.Connection;
3. import java.sql.PreparedStatement;
4. import java.sql.DriverManager;
5. import java.sql.ResultSet;
6. import java.sql.Statement;
7. import java.util.*;
8. import java.util.concurrent.ConcurrentHashMap;
10. /**
11.  * @author lzk
12.  * ConrurrentHashMap緩衝測試程式
13.  *
14.  */
15. public class BufferHashMapTest {
16.     private Statement stmt = null;
17.     private Connection con = null;
18.     private ResultSet rs = null;
19.     private String drivce = "com.microsoft.sqlserver.jdbc.SQLServerDriver";
20.     private String URL = "jdbc:sqlserver://127.0.0.1:1433;DatabaseName = motortestv1";
21.     public static ConcurrentHashMap updatehashmap = new ConcurrentHashMap();
22.     
23.     public Connection getcon(){             //得到資料庫連接
24.         try{
25.             Class.forName(drivce);
26.             con = DriverManager.getConnection(URL,"sa","ba123");
27.         }catch(Exception e){
28.             System.out.println(e.getMessage());
29.         }
30.         return con;
31.     }
32.     
33.     public ConcurrentHashMap queryHashMap(String sql) throws Exception {      
34.         ConcurrentHashMap conhashmap = new ConcurrentHashMap();
35.         Connection con = new BufferHashMapTest().getcon();
36.         stmt = con.createStatement();
37.         rs = stmt.executeQuery(sql);
38.         
39.         while (rs.next()) {
40.             Integer array[] = new Integer[2];
41.             array[0] = rs.getInt(1);
42.             array[1] = rs.getInt(2);
43.             conhashmap.put(new Integer(array[0]), array[1]);
44.         }
45.         rs.close();
46.         return conhashmap;
47.     }
48.     
49.     public static void main(String []args){
50.         String selectsql = "select id,status from users"; 
51.         BufferHashMapTest bhmt = new BufferHashMapTest();
52.         
53.         try{
54.             ConcurrentHashMap conhashmap = bhmt.queryHashMap(selectsql);
55.         
56.             new Login(conhashmap).start();    //啟動登陸線程
57.             new Logout(conhashmap).start();   //啟動登出線程
58.             new updateDB(bhmt).start();       //啟動更新資料庫線程
59.         }catch(Exception e){
60.             System.out.println("Error2 :"+e.getMessage());
61.         }
62.     }
63. }
65. class Login extends Thread{              //登陸線程
66.     private ConcurrentHashMap conhashmap = null;
67.     
68.     public Login(ConcurrentHashMap conhashmap){
69.         this.conhashmap = conhashmap;
70.     }
71.     
72.     public void run(){
73.         try{
74.             while(true){
75.                 int id = (int)(Math.random()*10000+1);
76.                 int status = 1;             //status=1表示已經登陸
77.                 if(conhashmap.containsKey(id)){
78.                     int values = Integer.parseInt(conhashmap.get(id).toString());
79.                     if( values == status){
80.                         System.out.println("使用者已經登陸！");
81.                     }else{
82.                         conhashmap.put(id, status);                 //更新記憶體中conhashmap的當前資訊
83.                         BufferHashMapTest.updatehashmap.put(id, status);  //臨時儲存更新資訊
84.                     }
85.                 }else{
86.                     //System.out.println(id);
87.                     System.out.println("id不存在！");
88.                 }
89.                 Thread.sleep(4);
90.             }
91.         }catch(Exception e){
92.             System.out.println("Error Login.run():"+e.getMessage());
93.         }
94.     }
95. }
97. class Logout extends Thread{                 //登出線程
98.     private ConcurrentHashMap conhashmap = null;
99.     
100.     public Logout(ConcurrentHashMap conhashmap){
101.         this.conhashmap = conhashmap;
102.     }
103.     
104.     public void run(){
105.         try{
106.             while(true){
107.                 int id = (int)(Math.random()*10000+1);
108.                 int status = 0;             //status=0表示已經登出
109.                 if(conhashmap.containsKey(id)){
110.                     int values = Integer.parseInt(conhashmap.get(id).toString()); 
111.                     if(values == status){
112.                         System.out.println("使用者已經登出!");
113.                     }else{
114.                         conhashmap.put(id, status);         //更新記憶體中conhashmap的當前資訊
115.                         BufferHashMapTest.updatehashmap.put(id, status);  //臨時儲存更新資訊
116.                     }
117.                 }else{
118.                     //System.out.println(id);
119.                     System.out.println("id不存在！");
120.                 }
121.                 Thread.sleep(5);
122.             }
123.         }catch(Exception e){
124.             System.out.println("Error Logout.run()"+e.getMessage());
125.         }
126.     }
127. }
129. class updateDB extends Thread{               //更新資料庫線程 
130.     private BufferHashMapTest bhmt = null;
131.     private Connection conn= null;    
132.     private PreparedStatement pstmt = null;  
133.     private int length = 0; 
134.     
135.     public updateDB(BufferHashMapTest bhmt){
136.         this.bhmt = bhmt;
137.     }
138.     
139.     public void run(){
140.         try{
141.             while(true){
142.                 conn = bhmt.getcon();
143.                 length = bhmt.updatehashmap.size();
144.                 pstmt =conn.prepareStatement("UPDATE users SET status = ? WHERE id = ?");  
145.                 Set entryset = bhmt.updatehashmap.entrySet(); 
146.                 Iterator iter = entryset.iterator(); 
147.                 
148.                 while(iter.hasNext()) { 
149.                     Map.Entry entry = (Map.Entry)iter.next(); 
150.                     int key = Integer.parseInt(entry.getKey().toString());
151.                     int value = Integer.parseInt(entry.getValue().toString());
152.                     System.out.print(key+" ");
153.                     System.out.println(value);
154.                     pstmt.setInt(1,value);
155.                     pstmt.setInt(2, key);
156.                     int tmp = pstmt.executeUpdate();
157.                     //System.out.println(tmp);
158.                     bhmt.updatehashmap.remove(key);         //更新資料庫後移除臨時hashmap中的元素
159.                 }
160.                 Thread.sleep(2000);
161.             }
162.         }catch(Exception e){
163.             System.out.println("Error updateDB.run()"+e.getMessage());
164.         }
165.     }
166. }