如何使这段代码更具功能性

时间:2013-01-09 09:52:10

标签: scala functional-programming refactoring

我是scala和函数式编程的新手。我正在尝试通常的初学者应用程序和脚本(显然使用了一些过度技术)

无论如何,我有一个计算器的代码,它带有参数和一个开关来指示要在参数上使用的操作。

object Main {
    def main(args: Array[String]): Unit = {

        var calc = calculate( "" , _:List[Int])
        var values:List[Int] = List()

        if(args.size < 1) println("No arguments supplied") else{
            args collect {_ match{
                    case arg if arg.contains("-") => {
                            if(values.size>0){
                                calc(values)
                                values = List()}
                            calc = calculate( arg , _:List[Int])
                        }
                    case arg => {
                            try{
                                val value=arg.toInt
                                values = values ::: List(value)
                            }catch{case e:NumberFormatException=>println("\nError:Invalid argument:\""+arg+"\"\nCannot convert to Integer.\n")}
                    }
                }
            }
            calc(values)
        }
    }

    def sum(values:List[Int]) { println("The sum is:"+(values.foldLeft(0)((sum,value) => sum + value))) }

    def subtract(values:List[Int]) {
        val initial:Int = values(0)

        var total:Int = 0
        for(i <- 1 until values.size){
            total = total + values(i)
        }
        val diff:Int = initial - total
        println("The difference is:"+diff)
    }

    def calculate(operation:String,values:List[Int]) = operation match {
        case "-sum" => sum(values)
        case "-diff" => subtract(values)
        case _ => println("Default operation \"Sum\" will be applied");sum(values)
    }
}

有些人想知道是否有更好的方法,就像删除try catch语句一样。

非常欢迎更好的撰写此应用程序的方法。

1 个答案:

答案 0 :(得分:4)

这个怎么样?

object Main extends App {
    require(args.size > 0, "Please, supply more arguments")

    @annotation.tailrec
    def parseArguments(arguments: List[String], operation: String, values: List[Int]() = Nil) {
        if(values.nonEmpty) calculate(operation, values)
        arguments match {
            case op::unprocessed if op.startsWith("-") => parseArguments(unprocessed, op)
            case maybeNumber::unprocessed => {
                val newValues = try {
                    maybeNumber.toInt::values
                } catch {
                    case _: NumberFormatException => 
                        println("\nError:Invalid argument:\""+maybeNumber+"\"\nCannot convert to Integer.\n")
                        values
                }             
                parseArguments(unprocessed, operation, newValues)   
            }
            case Nil => //done processing, exiting
        }
    }

    parseArguments(args.toList, "")

    def diff(values:List[Int]) = {
        val initial::tail = values
        val total = tail.sum
        initial - total
    }

    def calculate(operation:String, values:List[Int]) = operation match {
        case "-sum"  => println("The sum is " + values.sum)
        case "-diff" => println("The difference is: " + diff(values))
        case _       => 
                        println("""Default operation "Sum" will be applied""")
                        sum(values)
    }
}