Appeler des fonctions et des procédures stockées dans Snowpark Java¶

Appel de fonctions définies par le système¶

Si vous devez appeler des fonctions SQL définies par le système, utilisez les méthodes statiques équivalentes de la classe Functions.

L’exemple suivant appelle la upper méthode statique de la classe Functions (l’équivalent de la fonction UPPER définie par le système) pour renvoyer les valeurs de la colonne nom avec les lettres en majuscules :

DataFrame df = session.table("sample_product_data");
df.select(Functions.upper(Functions.col("name"))).show();

Si une fonction SQL définie par le système n’est pas disponible dans la classe Functions, vous pouvez utiliser la méthode statique Functions.callUDF pour appeler la fonction définie par le système.

Pour callUDF, transmettez le nom de la fonction définie par le système comme premier argument. Si vous devez transmettre les valeurs des colonnes à la fonction définie par le système, définissez et transmettez les objets colonne comme arguments supplémentaires dans la méthode callUDF.

L’exemple suivant appelle la fonction RADIANS définie par le système, en lui transmettant la valeur de la colonne degrees :

// Call the system-defined function RADIANS() on degrees.
DataFrame dfDegrees = session.range(0, 360, 45).rename("degrees", Functions.col("id"));
dfDegrees.select(Functions.col("degrees"), Functions.callUDF("radians", Functions.col("degrees"))).show();

La méthode callUDF renvoie un Column, que vous pouvez transmettre aux méthodes de transformation DataFrame (par exemple, filtre, sélection, etc.).

Appel de fonctions scalaires définies par l’utilisateur (UDFs)¶

La méthode pour appeler une UDF dépend de la façon dont l’UDF a été créée :

• Pour appeler une UDF anonyme, appelez la méthode apply de l’objet UserDefinedFunction qui a été retourné lorsque vous avez créé l’UDF.

  Les arguments que vous transmettez à une UDF doivent être des objets de colonne. Si vous devez transmettre un littéral, utilisez Functions.lit(), comme expliqué dans Utilisation de littéraux comme objets de colonne.

• Pour appeler des UDFs que vous avez enregistrées par le nom et des UDFs que vous avez créées en exécutant CREATE FUNCTION, utilisez la méthode statique Functions.callUDF.

  Transmettez le nom de l’UDF comme premier argument et tous les paramètres de l’UDF comme arguments supplémentaires.

L’appel d’une UDF renvoie un objet Column contenant la valeur de retour de l’UDF.

L’exemple suivant appelle la fonction UDF doubleUdf, en lui transmettant la valeur des colonnes quantity. L’exemple transmet la valeur de retour de doubleUdf à la méthode select de DataFrame.

import com.snowflake.snowpark_java.types.*;
...
// Create and register a temporary named UDF
// that takes in an integer argument and returns an integer value.
UserDefinedFunction doubleUdf =
  session
    .udf()
    .registerTemporary(
      "doubleUdf",
      (Integer x) -> x + x,
      DataTypes.IntegerType,
      DataTypes.IntegerType);
// Call the named UDF, passing in the "quantity" column.
// The example uses withColumn to return a DataFrame containing
// the UDF result in a new column named "doubleQuantity".
DataFrame df = session.table("sample_product_data");
DataFrame dfWithDoubleQuantity = df.withColumn("doubleQuantity", doubleUdf.apply(Functions.col("quantity")));
dfWithDoubleQuantity.show();

Appel de fonctions de table (fonctions système et UDTFs)¶

Pour appeler une fonction de table ou une fonction de table définie par l’utilisateur (UDTF) :

1. Construisez un objet TableFunction en transmettant le nom de la fonction de table.

2. Appelez la méthode tableFunction de l’objet Session, en lui transmettant l’objet TableFunction et un Map de noms et valeurs d’arguments d’entrée.

table?Function renvoie un DataFrame qui contient la sortie de la fonction de table.

Par exemple, supposons que vous ayez exécuté la commande suivante pour créer une UDTF SQL :

CREATE OR REPLACE FUNCTION product_by_category_id(cat_id INT)
  RETURNS TABLE(id INT, name VARCHAR)
  AS
  $$
    SELECT id, name
      FROM sample_product_data
      WHERE category_id = cat_id
  $$
  ;

Le code suivant appelle cette UDTF et crée un DataFrame pour la sortie de l’UDTF. L’exemple affiche les 10 premières lignes de la sortie dans la console.

import java.util.HashMap;
import java.util.Map;
...

Map<String, Column> arguments = new HashMap<>();
arguments.put("cat_id", Functions.lit(10));
DataFrame dfTableFunctionOutput = session.tableFunction(new TableFunction("product_by_category_id"), arguments);
dfTableFunctionOutput.show();

Si vous devez joindre la sortie d’une fonction de table à un DataFrame, appelez la méthode join qui se transmet dans une TableFunction.

Appel de procédures stockées¶

Pour appeler une procédure stockée, utilisez la méthode sql de la classe Session. Voir Exécution d’instructions SQL.