方法my_udf有一个返回语句；需要结果类型

Question

我读到在大多数情况下不需要为udf指定返回类型，但如果您添加返回语句，那么它是必要的。

我有以下udf，它在内部迭代一个列表，我希望在匹配的情况下udf立即返回。基于我在这里找到的其他答案，我尝试了以下两种方法来定义返回类型：

def removeSalutation = udf((name: String) => String {

    val salutationList = List("MRS", "MR", "MSTR", "MISS", "MS", "DR")

    for(salutation <- salutationList){
        if(name.endsWith(salutation)){
            return name.dropRight(salutation.length())
        }
    }
    return name
    
})

上面的脚本只是返回一个错误

def removeSalutation = udf[String, String]((name: String) => {

    val salutationList = List("MRS", "MR", "MSTR", "MISS", "MS", "DR")

    for(salutation <- salutationList){
        if(name.endsWith(salutation)){
            return name.dropRight(salutation.length())
        }
    }
    return name
    
}) 

上面的剧本一直在说:

错误：方法删除别名有返回语句;需要结果类型

指定返回类型的正确方法是什么？

Answer 1

如下分别定义函数:

def _removeSalutation(name: String) : String = {
    val salutationList = List("MRS", "MR", "MSTR", "MISS", "MS", "DR")
    for(salutation <- salutationList){
        if(name.endsWith(salutation)){
            return name.dropRight(salutation.length())
        }
    }
    return name
}

val removeSalutation = udf(_removeSalutation _)