我想对golang封装一个切片过滤的方法 filter
func Filter(arr []interface{}, fun fn ) []interface{}{
var result []interface{}
if len(arr)==0 {
result = make([]interface{},0)
} else {
for _,v := range arr {
if fun(v) {
result = append(result, v)
}
}
}
return result
}
使用方法是传入任意类型的切片[]interface{},传入一个过滤的方法。
但是调用的时候就出现问题了
var list = []int{1,2,3,4,5}
var list2 = Filter(list, func(item interface{})bool{
if item > 3 {
return true
}
return false
})
log.Println("filter", list2)
这时候会报错。报错信息是
Cannot use 'list' (type []int) as type []interface{}
意思是[]int
类型不能用作[]interface{}
类型。
那么就需要将list转化为[]interface{}
类型。怎么转呢?只能通过for循环方法
var _list []interface{}
for _,v := range list {
_list = append(_list, v)
}
转完之后将_list传入Filter第一个参数,这时候出现新的问题了。
var list2 = Filter(_list, func(item interface{})bool{
if item > 3 {
return true
}
return false
})
报错信息是
Invalid operation: item > 3 (operator > not defined on interface{})
因为这时候item是interface{}类型,没有这个>操作符。
应该要把item强制转化为int类型。方法是item.(int)
var list2 = Filter(_list, func(item interface{})bool{
if item.(int) > 3 {
return true
}
return false
})
同理,如果要处理的数组是如下类型
type Stu struct {
Id int
Name string
}
var list3 = []Stu{{1,"a"},{2,"b"}, {3, "c"}}
var list3_1= Filter(_list3, func(item interface{})bool{
if item.(Stu).Id > 1 {
return true
}
return false
})
那么在比较的地方也要做类型转化。 item.(Stu)
先转化为Stu
类型,再取Id做比较
网友评论