ひできちの楽天ブログ

< 新しい記事

新着記事一覧(全962件)

過去の記事 >

2025/09/19

snowflakeのユーザー定義関数はちょっと使いにくいとな？

テーマ： snowflake(3)

カテゴリ： snowflake

snowflakeで利用できる言語はSQL以外にも
JAVASCRIPT やPython が使えるということで
ユーザー定義関数でいろいろと試してみておりますよ

JAVASCRIPT ベースの配列を返すUDF

--JAVASCRIPT ベースのUDF
CREATE OR REPLACE FUNCTION make_array_js(val1 STRING,val2 STRING)
RETURNS ARRAY
LANGUAGE JAVASCRIPT
AS
$$
return [VAL1,VAL2];
$$;
select make_array_js('test','test1');
select make_array_js('test','test1')[0];

JAVASCRIPT はつかえるのですが、SQLの発行はできないという制約がありましたよ
なので、活用するには使い分け、そして関数のネストが必要になりそうですな

SQLベースの配列を返すUDF

--配列を返すUDF
CREATE OR REPLACE FUNCTION test_array(id STRING)
RETURNS VARIANT
LANGUAGE SQL
as
declare
val1 integer;
val2 integer;
begin

val1:=1;
val2:=2;

if (id='test') then
return object_construct( val1);
else
return object_construct(val2);
end if;
end;

--実行時エラーとなってしまう
select test_array('test');

単純なスカラーを返却するUDFでは平凡なので
返却値を配列に拡張できるのか？を検証してみたのですが
SQLベースのUDFではどうも無理っぽいですな
JAVASCRIPT ベースのUDFとの併用ができれば回避できそうですが・・

SCALAベースのUDF

--半構造体データを引数に受け取って操作するUDF
CREATE TABLE objectives (o OBJECT);
INSERT INTO objectives SELECT PARSE_JSON('{"outer_key" : {"inner_key" : "inner_value"} }');

CREATE OR REPLACE FUNCTION extract_from_object(x OBJECT, key VARCHAR)
RETURNS VARIANT
LANGUAGE SCALA
RUNTIME_VERSION = 2.12
HANDLER='VariantLibrary.extract'
AS
$$
import scala.collection.immutable.Map

class VariantLibrary {
def extract(m: Map[String, String], key: String): String = {
return m(key)
}
}
$$;

SELECT *,extract_from_object(o, 'outer_key'), extract_from_object(o, 'outer_key')['inner_key'] FROM OBJECTIVES;
SELECT * from OBJECTIVES;

snowflake公式のリソースを見ると
SCALAでのサンプルもあったので試してみましたよ
半構造体データから値を取り出すUDFがありまして
これはかなりありがたいソースですな
SQLベースのUDFで取得した半構造体データから更に値を
取り出して返却するUDFが実装できれば便利ですな

SCALAベースのUDFを参照するUDF

--scalaで書かれたUDFを内包するUDF
CREATE OR REPLACE FUNCTION extract_from_object2(x OBJECT, key VARCHAR)
RETURNS VARIANT
LANGUAGE SQL
as
$$
select extract_from_object(x , key )
$$;
select extract_from_object2(o, 'outer_key')FROM OBJECTIVES;
SELECT *,extract_from_object2(o, 'outer_key'), extract_from_object2(o, 'outer_key')['inner_key'] FROM OBJECTIVES;

scala で書かれたUDFを参照するUDFは作れるのか？を検証してみましたが
問題なく実行できましたよ

SQLベースのUDFで少しだけ複雑な内容のものを検証してましたが
ここでなにやらよくわからない動作にぶつかってしまいましたよ

SQLベースのブロック構造を使ったUDF

create or replace FUNCTION area(x float)
returns float
language SQL
as
$$
declare
radius float;
area_of_circle float;
begin
radius := x;
area_of_circle := pi() * radius * radius;
return area_of_circle;
end;
$$
;

select area(1);

これは問題なく動作するのですが・・・