在Go中编写独特渠道的正确方法是什么？

Question

I am a beginner in go.

I am trying to figure out an easy way to implement a channel that only output distinct values.

What I want to do is this:

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)      
    }
}

Should I be concern about memory leak here if I were to use a map to remember previous values?

Thanks.

Answer 1

You really can't do that, you'd have to keep a track of the values somehow, a map[int]struct{} is probably the most memory efficient way.

A simple 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
}