Class: GeoIP2::Database

Inherits:

Object

• Object
• GeoIP2::Database

Defined in:

lib/geoip2/database.rb,
ext/geoip2/geoip2.c

Instance Method Summary

• #close ⇒ Object
• #initialize(path, symbolize_keys: false) ⇒ Database constructor

  A new instance of Database.

• #lookup(ip) ⇒ Object
• #open_mmdb(path) ⇒ Object

Constructor Details

#initialize(path, symbolize_keys: false) ⇒ Database

Returns a new instance of Database.

# File 'lib/geoip2/database.rb', line 3

def initialize(path, symbolize_keys: false)
  @symbolize_keys = !!symbolize_keys
  open_mmdb(path)
end

Instance Method Details

#close ⇒ Object

# File 'ext/geoip2/geoip2.c', line 274

static VALUE
rb_geoip2_db_close(VALUE self)
{
  MMDB_s *mmdb;

  TypedData_Get_Struct(self, struct MMDB_s, &rb_mmdb_type, mmdb);

  if (!mmdb_is_closed(mmdb)) {
    mmdb_close(mmdb);
  }

  return Qnil;
}

#lookup(ip) ⇒ Object

# File 'ext/geoip2/geoip2.c', line 288

static VALUE
rb_geoip2_db_lookup(VALUE self, VALUE ip)
{
  char *ip_str;
  MMDB_s *mmdb;
  MMDB_lookup_result_s result;
  MMDB_lookup_result_s *result_ptr;
  VALUE obj;

  Check_Type(ip, T_STRING);
  ip_str = StringValueCStr(ip);

  TypedData_Get_Struct(self, struct MMDB_s, &rb_mmdb_type, mmdb);
  result = mmdb_lookup(mmdb, ip_str, false);

  if (!result.found_entry) {
    return Qnil;
  }

  obj = TypedData_Make_Struct(rb_cGeoIP2LookupResult,
                              struct MMDB_lookup_result_s,
                              &rb_lookup_result_type,
                              result_ptr);
  result_ptr->found_entry = result.found_entry;
  result_ptr->entry = result.entry;
  result_ptr->netmask = result.netmask;

  rb_iv_set(obj, "@symbolize_keys", rb_iv_get(self, "@symbolize_keys"));

  return obj;
}

#open_mmdb(path) ⇒ Object

# File 'ext/geoip2/geoip2.c', line 258

static VALUE
rb_geoip2_db_open_mmdb(VALUE self, VALUE path)
{
  char *db_path;
  MMDB_s *mmdb;

  Check_Type(path, T_STRING);

  db_path = StringValueCStr(path);

  TypedData_Get_Struct(self, struct MMDB_s, &rb_mmdb_type, mmdb);
  mmdb_open(db_path, mmdb);

  return Qnil;
}