Spark UDF to filter map's key

Spark UDF filter

// given - column with type Map<String, String>


import static org.apache.spark.sql.functions.callUDF;
import static org.apache.spark.sql.functions.lit;
import static org.apache.spark.sql.functions.col;

// define UDF
private UserDefinedFunction keyContains = functions.udf(
            (scala.collection.Map<String, String> m, String key) -> m.contains(key),
            DataTypes.BooleanType
    );
    
// register UDF
sparkSession.udf().register("keyContains", keyContains);

// use:
Column filter = callUDF("keyContains", col("mapColumnName"), lit("keyToCheck"));