|
종종 프로그램이 데이터 컬렉션 위에서 작업을 수행해야하는 경우가 있습니다. 가령, 주어진 조건을 만족하는 모든 항목을 선택하거나 모든 항목을 커스텀 함수를 가지고 새 컬렉션으로 매핑하는 경우가 있습니다. |
|
|
몇몇 언어에서는 generic 자료 구조와 알고리즘을 사용하는게 일반적입니다. Go는 제네릭을 지원하지 않습니다. Go에선 여러분의 프로그램과 데이터 타입에 특별히 필요한 경우 컬렉션 함수를 제공하는게 일반적입니다. |
|
|
여기에 |
|
package main
|
|
import "strings"
import "fmt"
|
|
|
문자열 |
func Index(vs []string, t string) int {
for i, v := range vs {
if v == t {
return i
}
}
return -1
}
|
|
문자열 t가 슬라이스에 존재하면 |
func Include(vs []string, t string) bool {
return Index(vs, t) >= 0
}
|
|
슬라이스의 문자열중 하나가 조건 |
func Any(vs []string, f func(string) bool) bool {
for _, v := range vs {
if f(v) {
return true
}
}
return false
}
|
|
슬라이스의 문자열 모두가 조건 |
func All(vs []string, f func(string) bool) bool {
for _, v := range vs {
if !f(v) {
return false
}
}
return true
}
|
|
슬라이스에서 조건 |
func Filter(vs []string, f func(string) bool) []string {
vsf := make([]string, 0)
for _, v := range vs {
if f(v) {
vsf = append(vsf, v)
}
}
return vsf
}
|
|
기존 슬라이스에 있는 각각의 문자열에 함수 |
func Map(vs []string, f func(string) string) []string {
vsm := make([]string, len(vs))
for i, v := range vs {
vsm[i] = f(v)
}
return vsm
}
|
func main() {
|
|
|
다양한 컬렉션 함수를 사용해봅시다. |
var strs = []string{"peach", "apple", "pear", "plum"}
|
fmt.Println(Index(strs, "pear"))
|
|
fmt.Println(Include(strs, "grape"))
|
|
fmt.Println(Any(strs, func(v string) bool {
return strings.HasPrefix(v, "p")
}))
|
|
fmt.Println(All(strs, func(v string) bool {
return strings.HasPrefix(v, "p")
}))
|
|
fmt.Println(Filter(strs, func(v string) bool {
return strings.Contains(v, "e")
}))
|
|
|
위의 에시 모두 익명함수를 사용했지만, 올바른 타입을 가진 명명된 함수를 사용할 수도 있습니다. |
fmt.Println(Map(strs, strings.ToUpper))
|
}
|
$ go run collection-functions.go
2
false
true
false
[peach apple pear]
[PEACH APPLE PEAR PLUM]
|
다음 예제: 문자열 함수.