ID生成服务

ID生成服务本身应该串行化,避免锁竞争.

1. import java.sql.Connection;
   
2. import java.sql.DriverManager;
   
3. import java.sql.ResultSet;
   
4. import java.sql.Statement;
   
5. import java.util.concurrent.Callable;
   
6. import java.util.concurrent.ExecutionException;
   
7. import java.util.concurrent.ExecutorService;
   
8. import java.util.concurrent.Executors;
   
9. import java.util.concurrent.Future;
   
10. 
    
11. public class SeqGenerator {
    
12.     private static ExecutorService singleThreadExecutor = Executors.newSingleThreadExecutor();
    
13. 
    
14.     private static int currentVal = -1;
    
15.     private static int maxVal = -1;
    
16.     private static int fetchSize = 10000;
    
17. 
    
18.     static{
    
19.         try {
    
20.             fetchFromDB();
    
21.         } catch (Exception e) {
    
22.             // TODO Auto-generated catch block
    
23.             e.printStackTrace();
    
24.         }
    
25.     }
    
26.     
    
27.     public static int getSeq() throws InterruptedException, ExecutionException {
    
28.         Callable<Integer> c = new Callable<Integer>() {
    
29.             @Override
    
30.             public Integer call() throws Exception {
    
31.                 int result = currentVal;
    
32.                 if (currentVal > maxVal) {
    
33.                     fetchFromDB();
    
34.                     result = currentVal;
    
35.                 }
    
36.                 currentVal++;
    
37.                 return result;
    
38. 
    
39.             }
    
40.         };
    
41.         Future<Integer> task = singleThreadExecutor.submit(c);
    
42.         return task.get().intValue();
    
43.     }
    
44. 
    
45.     private static void fetchFromDB() throws Exception {
    
46.         Class.forName("com.mysql.jdbc.Driver");
    
47.         Connection connection = DriverManager.getConnection("jdbc:mysql://127.0.0.1:3306/test", "xx", "xx");
    
48.         connection.setAutoCommit(false);
    
49.         Statement st = connection.createStatement();
    
50.         ResultSet rs = st.executeQuery("select * from test for update");
    
51.         rs.next();
    
52.         currentVal = rs.getInt(1) + 1;
    
53.         rs.close();
    
54.         st.executeUpdate("update test set id=id+" + fetchSize);
    
55.         rs = st.executeQuery("select * from test for update");
    
56.         rs.next();
    
57.         maxVal = rs.getInt(1);
    
58.         connection.commit();
    
59.         rs.close();
    
60.         st.close();
    
61.         connection.close();
    
62.     }
    
63. 
    
64.     public static void main(String[] args) throws Exception {
    
65.         int i = 1000000;
    
66.         long start = System.currentTimeMillis();
    
67. 
    
68.         while (i > 0) {
    
69.             System.out.println(SeqGenerator.getSeq());
    
70.             i--;
    
71.         }
    
72.         long end = System.currentTimeMillis();
    
73.         System.out.println(end - start);
    
74.     }
    
75. }

    使用这种自定义序列,1百万ID的生成时间是14s.效果非常明显.