JRuby 调用 maxmind-java 测试

#!/usr/bin/env jruby
require "geoip"
require "maxminddb"
require "thread"
require "java"

# 测试数据
ip = '202.106.0.20'

# 加载 maxmind-java 的所有 jar 包
Dir["/Users/raochenlin/geoip2-2.1.0/lib/*.jar"].each { |jar| require jar }

# 导入关键性的 java 类
import com.maxmind.geoip2.DatabaseReader
import java.net.InetAddress
# 这个原生的 java 写法是：
#   File database = new File("/Users/raochenlin/GeoLite2-City.mmdb")
#   DatabaseReader reader = new DatabaseReader.Builder(database).build()
# 之前对 java 不太懂，想直接 import Builder 进来
# 其实 Builder 是DatabaseReader 类里的静态类(public final static class)，不能直接 import
database = java.io.File.new("/Users/raochenlin/GeoLite2-City.mmdb")
@reader = DatabaseReader::Builder.new(database).build()

# 纯 Ruby 实现的库
@db = MaxMindDB.new('/Users/raochenlin/GeoLite2-City.mmdb')

# 老的 GeoIP 库，需要制定不同的数据文件类型，这部分直接抄自 logstash 源码
@geo = GeoIP.new('/Users/raochenlin/Downloads/logstash-1.4.2/vendor/geoip/GeoLiteCity.dat')
@geoip_type = case @geo.database_type
when GeoIP::GEOIP_CITY_EDITION_REV0, GeoIP::GEOIP_CITY_EDITION_REV1
  :city
when GeoIP::GEOIP_COUNTRY_EDITION
  :country
when GeoIP::GEOIP_ASNUM_EDITION
  :asn
when GeoIP::GEOIP_ISP_EDITION, GeoIP::GEOIP_ORG_EDITION
  :isp
else
  raise RuntimeException.new "This GeoIP database is not currently supported"
end

# 开始 logstash 流程
# 创建从 input 到 filter 的缓冲队列，固定大小 20
# SizedQueue 是 thread 库导入的
@input_to_filter = SizedQueue.new(20)

# 具体的 geoip 过滤器线程
def geoworker
    begin
        while true
            ip = @input_to_filter.pop

# GeoIP 查询方法
#            data = @geo.send(@geoip_type, ip)
#            puts data.to_hash[:city_name]

# MaxMind-java 查询方法，注意传入的是 InetAddress 对象
            data = @reader.city(InetAddress.getByName(ip))
#            puts data.getCountry().getName()

# maxminddb 查询方法
#            data = @db.lookup(ip)
#            puts data.country.name
        end
    end
end

# 定义 input 线程，传入一百万次 IP 到缓冲队列
lines_num = 1000000
input = Thread.new do
    lines_num.times.each do |i|
        @input_to_filter.push(ip)
    end
# IP 发送完毕，计算每秒处理的速率
    end_time = Time.now.to_f * 1000
    puts lines_num * 1000 / (end_time - @start_time)
end

# 定义 filter 线程，启动 20 个
arr = 20.times.collect do
    Thread.new do
        geoworker
    end
end

# 记录开始时间，运行定义好的各线程
@start_time = Time.now.to_f * 1000
input.join
arr.each{|t| t.join}