GeoIP2雪花Java集成问题

Question

我希望在雪花工作表中创建一个Java，以便查询GeoIp2库并获取给定IP的ISO代码。我有“@AWS_CSV_STAGE/lib/Geoip2-2.8.0.jar”、“@AWS_CSV_STAGE/geodata/GeoLite2-City.mmdb”。如何将函数处理程序定向到创建数据库读取器的方法，正如本文在文档for：https://dev.maxmind.com/geoip/geolocate-an-ip/databases?lang=en#1-install-the-geoip2-client-library中所解释的那样，如何在我的udf中实现以下所有内容？

File database = new File("/path/to/maxmind-database.mmdb")
DatabaseReader reader = new DatabaseReader.Builder(database).build();
InetAddress ipAddress = InetAddress.getByName("128.101.101.101");
CityResponse response = reader.city(ipAddress);
Country country = response.getCountry();

到目前为止，我写了这篇文章，但当然没有用:无论如何，我找不到很多关于如何解决这类问题的材料。

CREATE OR REPLACE FUNCTION GEO()
  returns varchar not null
  language java
  imports = ('@AWS_CSV_STAGE/lib/geoip2-2.8.0.jar','@AWS_CSV_STAGE/geodata/GeoLite2-City.mmdb')
  handler = 'DatabaseReader.Builder';

SELECT GEO();

基本上，我想要实现的是在ip地址表的一列上调用UDF，并在另一列中获取每个ip地址的国家代码。

Answer 1

要在雪花中创建Java用户定义函数(UDF)，您需要在雪花SQL中使用CREATEFunction语句。该语句的语法如下：

CREATE OR REPLACE FUNCTION function_name
RETURNS data_type
LANGUAGE JAVA
IMPORTS = ('file_path_1', 'file_path_2', ...)
HANDLER = 'fully_qualified_class_name.method_name'

在您的示例中，您可以使用以下CREATE语句来创建您的UDF：

CREATE OR REPLACE FUNCTION GEO
RETURNS VARCHAR
LANGUAGE JAVA
IMPORTS = ('@AWS_CSV_STAGE/lib/geoip2-2.8.0.jar','@AWS_CSV_STAGE/geodata/GeoLite2-City.mmdb')
HANDLER = 'com.maxmind.geoip2.DatabaseReader.Builder'

SELECT GEO(ip_address_column) AS country_code
FROM ip_addresses

此查询将使用GEO获取ip_addresses表中每个IP地址的国家代码，并在新的country_code列中返回国家代码。

Answer 2

为了在雪花中创建Java，创建一个定义UDF函数及其行为的Java类。这个类应该包括您为创建DatabaseReader和查询GeoIP2数据库以获得给定IP地址的ISO代码而提供的代码。

定义该类后，使用雪花中的CREATE替换函数语句注册函数并使其可用于查询。CREATE替换函数语句的IMPORTS子句用于指定函数所依赖的外部JAR文件，例如您提到的GeoIP2库JAR文件。

下面是一个示例，说明CREATE替换函数语句的外观：

CREATE OR REPLACE FUNCTION GEO(ipAddress VARCHAR)
RETURNS VARCHAR
LANGUAGE JAVA
IMPORTS ('@AWS_CSV_STAGE/lib/geoip2-2.8.0.jar','@AWS_CSV_STAGE/geodata/GeoLite2-City.mmdb')
HANDLER 'com.example.GeoIpFunction'

在本例中，com.example.GeoIpFunction是您为实现函数而定义的Java类的完全限定名。若要在查询中使用该函数，请像任何其他雪花函数一样调用它，并将IP地址作为参数传递：

SELECT GEO('128.101.101.101')

这将返回指定IP地址的ISO代码。还可以在查询中使用此函数获取表列中每个IP地址的ISO代码：

SELECT GEO(ip_address) AS iso_code FROM my_table