在Go中编写不同频道的正确方法是什么?

时间:2014-08-13 14:24:21

标签: go

我是初学者。

我试图找出一种简单的方法来实现只输出不同值的通道。

我想做的是:

package example

import (
    "fmt"
    "testing"
)

func TestShouldReturnDistinctValues(t *testing.T) {

    var c := make([]chan int)

    c <- 1
    c <- 1
    c <- 2
    c <- 2
    c <- 3

    for e := range c {
        // only print 1, 2 and 3.
        fmt.println(e)      
    }
}

如果我使用地图记住之前的值,我是否应该关注内存泄漏?

感谢。

2 个答案:

答案 0 :(得分:5)

你真的不能这样做,你必须以某种方式跟踪价值观,map[int]struct{}可能是最有效的记忆方式。

一个简单的example

func UniqueGen(min, max int) <-chan int {
    m := make(map[int]struct{}, max-min)
    ch := make(chan int)
    go func() {
        for i := 0; i < 1000; i++ {
            v := min + rand.Intn(max)
            if _, ok := m[v]; !ok {
                ch <- v
                m[v] = struct{}{}
            }
        }
        close(ch)
    }()

    return ch
}

答案 1 :(得分:1)

之前我做过类似的事情,除了我的问题是输出输入按升序排列。您可以通过添加中间例程来完成此操作。这是一个example

package main

func main() {
    input, output := distinct()

    go func() {
        input <- 1
        input <- 1
        input <- 2
        input <- 2
        input <- 3
        close(input)
    }()

    for i := range output {
        println(i)
    }
}

func distinct() (input chan int, output chan int) {
    input = make(chan int)
    output = make(chan int)

    go func() {
        set := make(map[int]struct{})
        for i := range input {
            if _, ok := set[i]; !ok {
                set[i] = struct{}{}
                output <- i
            }
        }
        close(output)
    }()
    return
}