Question

我正在侦听UDP上的消息。我们有一些以这种方式宣布自己的设备。他们还说何时发送下一个公告。如果这没有发生，我们将假定设备已消失。

我想列出网络中当前的设备。我想添加新设备并删除我从未听到过的设备。

这是我到目前为止所掌握的。

1）我有一个存储所有设备的内存数据库。

func NewDB() *DB {
    return &DB{
        table: make(map[string]Announcement),
    }
}

type DB struct {
    mutex sync.Mutex
    table map[string]Announcement
}

func (db *DB) Set(ip string, ann Announcement) {
    db.mutex.Lock()
    defer db.mutex.Unlock()
    db.table[ip] = ann
}

func (db *DB) Delete(ip string) {
    db.mutex.Lock()
    defer db.mutex.Unlock()
    delete(db.table, ip)
}

func (db *DB) Snapshot() map[string]Announcement {
    db.mutex.Lock()
    defer db.mutex.Unlock()
    return db.table
}

2）我有一个Web服务器，可以将此数据库提供给我的JavaScript前端

http.HandleFunc("/json", func(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Content-Type", "application/json")
    json.NewEncoder(w).Encode(db.Snapshot())
})

// start server
go func() {
    log.Fatal(http.ListenAndServe(":8085", nil))
}()

3）最后，我正在侦听UDP消息。每当将新设备添加到数据库时，我都会使用提供的超时时间创建一个新计时器（此处我将其设置为10秒）。当收到新消息时，我会检查现有的计时器，将其停下来再重新启动以清除设备（如果它不再发送消息的话）。

但是我实际上并没有工作。 AfterFunc被称为经常使用的方式。尽管该设备仍在网络中，但仍将其从我的db中删除了。有什么想法吗？

// some global variable
var (
    timers = map[string]*time.Timer{}
)

for {
    // create new buffer
    b := make([]byte, 1500)

    // read message from udp into buffer
    n, src, err := conn.ReadFromUDP(b)
    if err != nil {
        panic(err)
    }

    // convert raw json bytes to struct
    var ann Announcement
    if err := json.Unmarshal(b[:n], &ann); err != nil {
        panic(err)
    }

    // add announcement to db
    ip := src.IP.String()
    db.Set(ip, ann)

    // check for existing timer
    timer, ok := timers[ip]
    if ok {
        log.Println("stopping timer", ip)
        // stop existing timer
        timer.Stop()
    }

    // start new timer for device
    timer = time.AfterFunc(time.Second*10, func() {
        log.Println("time after func", ip)
        delete(timers, ip)
        db.Delete(ip)
    })

    // store timer in timers db
    timers[ip] = timer

    time.Sleep(250 * time.Millisecond)
}

Answer 1

我认为您遇到的问题可能与您在ip AfterFunc中捕获的func变量的值有关。

timer = time.AfterFunc(time.Second*10, func() {
        log.Println("time after func", ip)
        delete(timers, ip)
        db.Delete(ip)
    })

根据此代码，此计时器到期后，将使用ip变量的值调用ip上的删除。因此，如果与此同时您从另一个具有不同IP的设备接收到一个数据包，则该数据包将被删除。

正在发生的事的例子：

秒1 ：IP为1.2.3.4的设备发送UDP数据包。 ip = 1.2.3.4，AfterFunc被调用，启动10秒计时器

秒3 ：IP 4.5.6.7的设备发送UDP数据包。现在调用ip = 4.5.6.7，AfterFunc，启动10秒计时器

第二个10 ：调用删除ip变量当前值的函数，删除设备4.5.6.7

秒13 ：秒计时器超时，我们尝试再次删除4.5.6.7

结果，永远不会删除IP为1.2.3.4的设备。

您可以通过创建一个接受参数并返回具有当前参数值的func（）的函数来解决此问题。

timer = time.AfterFunc(time.Second*10, func(ip string) func() { return func() { log.Println("time after func", ip) delete(timers, ip) db.Delete(ip) } }(ip))

简单的工作示例：

package main import ( "fmt" "time" ) func main() { fmt.Println("Capturing value of i at the moment of execution of func()") for i := 0; i < 5; i++ { afterFuncTimer := time.AfterFunc(time.Second*2, func() { fmt.Printf("AfterFunc() with %v\n", i) }) defer afterFuncTimer.Stop() } time.Sleep(5 * time.Second) fmt.Println("Capturing value of i from the loop") for i := 0; i < 5; i++ { afterFuncTimer := time.AfterFunc(time.Second*2, func(i int) func() { return func() { fmt.Printf("AfterFunc() with %v\n", i) } }(i)) defer afterFuncTimer.Stop() } time.Sleep(5 * time.Second) }

在Go Playground上运行它：https://play.golang.org/p/bGWzTaWe3ZU

在UDP消息上启动多个具有取消功能的计时器

1 个答案: