ruby Celluloid 学习-day11-程序实例

celluloid 官网的文档通读下来，基本有所心得。现结合实际生产，编写一个应用程序，展示如下。
程序目的： 比对新旧两套数据库的同名表记录条数是否一致。预先已将需比对的表名写入 diff_log 表中，添加了两个字段new_count和 old_count用于保存表的记录条数。

1. require 'celluloid'
   
2. require 'dbi'
   
3. 
   
4. class MyWorker
   
5. 
   
6.   # 定义一个 actor
7.   include Celluloid
   
8. 
   
9.   # 析构函数，当主 actor 退出时，所有的子 actor 将调用析构函数并退出
10.   finalizer :my_finalizer
11.   
    
12.   # 构造函数：连接新旧两套数据库
13.   def initialize
    
14.     @newdb = DBI.connect('dbi:OCI8:db_new','new_user','new_pass') or exit
    
15.     @olddb = DBI.connect('dbi:OCI8:db_old','old_user','old_pass') or exit
    
16.   end
    
17.   
    
18.   # 统计记录条数，写入 diff_log 
19.   def calc(table_name)
    
20.     new_count = @newdb.select_one("select count(1) c from #{table_name}").by_field("C");
    
21.     old_count = @olddb.select_one("select count(1) c from #{table_name}").by_field("C");
    
22.     
    
23.     sql = sprintf "update diff_log set new_count = %d,old_count = %d where table_name = '%s'",
    
24.       new_count,old_count,table_name
    
25.     puts sql
    
26.     
    
27.     @newdb.execute(sql)
    
28.     @newdb.commit    
    
29.   end
    
30.   
    
31.   def my_finalizer
    
32.     @newdb.disconnect if @newdb
    
33.     @olddb.disconnect if @olddb
    
34.   end
    
35. end
    
36. # 开启一个进程池，可以同时运行 20 个 actor
37. pool = MyWorker.pool(size: 20)
    
38. 
    
39. dbh = DBI.connect('dbi:OCI8:db_new','new_user','new_pass') or exit
    
40. # 主循环
41. dbh.select_all("select table_name from diff_log where new_count is null") do |row|
    
42.   table_name = row.by_field "TABLE_NAME"
    
43.   # 异步调用
44.   pool.async.calc(table_name)
    
45. end
    
46. 
    
47. dbh.disconnect if dbh
    
48. # 休眠主进程。否则主进程将直接退出，所有子 actor 也将被 kill 掉。
49. # 终止程序只有按 ctrl+c
50. sleep