登录
首页 >  Golang >  Go问答

如何实现多态变参数的求和?

来源:SegmentFault

时间:2023-01-08 16:05:23 344浏览 收藏

怎么入门Golang编程?需要学习哪些知识点?这是新手们刚接触编程时常见的问题;下面golang学习网就来给大家整理分享一些知识点,希望能够给初学者一些帮助。本篇文章就来介绍《如何实现多态变参数的求和?》,涉及到go、多态,有需要的可以收藏一下

问题内容

求问下面的Add2这里怎么实现?sum为interface好像没法任意求和?

func Add(vars ... int)(ret int){
    var sum int
    for _,v := range vars {
        //fmt.Println("v",v)
        sum += v
    }

    return sum
}


//这里怎么实现?sum为interface好像没法任意求和?
func Add2(vars ... interface{})(ret interface{}){
    /* var sum interface{}
    for _,v := range vars {
        switch v.(type){
            case int:
            sum += v
            case float64:
            sum += v
            case string:
            sum += v
        }
    } */

    return sum
}



func main() {
    fmt.Println(Add2(1,2))
    fmt.Println(Add2(1,2,3))
    fmt.Println(Add2(2.4, 3.6, 1.0))
    fmt.Println(Add2("Hello", "world", "zhong"))
}

正确答案

你的应该存在一个潜在的前提,就是传入的参数是一致的类型。

下面的例子,取第一个值的类型,作为整个相加运算的类型,如果有和第一个类型不一致的数据会被忽略,用到了反射。

package main

import (
    "fmt"
    "reflect"
)

func main() {
    fmt.Println(sum(1.2, 2.0))
    fmt.Println(sum(1, 2))
    fmt.Println(sum("hello", " ", "world"))
}

func sum(vars ...interface{}) interface{} {
    var result interface{}
    if len(vars) == 0 {
        return result
    }
    inferType := reflect.TypeOf(vars[0])
    for i, v := range vars {
        if i == 0 {
            result = v
            continue
        }
        if reflect.TypeOf(v) == inferType {
            switch inferType.Name() {
            case "int":
                result = result.(int) + v.(int)
            case "float64":
                result = result.(float64) + v.(float64)
            case "string":
                result = result.(string) + v.(string)
            }
        }
    }
    return result
}

输出

3.2
3
hello world

今天关于《如何实现多态变参数的求和?》的内容介绍就到此结束,如果有什么疑问或者建议,可以在golang学习网公众号下多多回复交流;文中若有不正之处,也希望回复留言以告知!

声明:本文转载于:SegmentFault 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>
评论列表