启动和取消多个 UDP 消息上的定时器

积累知识，胜过积蓄金银！毕竟在Golang开发的过程中，会遇到各种各样的问题，往往都是一些细节知识点还没有掌握好而导致的，因此基础知识点的积累是很重要的。下面本文《启动和取消多个 UDP 消息上的定时器》，就带大家讲解一下知识点，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

我正在侦听 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) 我有一个网络服务器，可以将此数据库提供给我的 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)
}

解决方案

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

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，func调用后，启动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 上运行演示：https://play.golang.org/p/bGWzTaWe3ZU

以上就是《启动和取消多个 UDP 消息上的定时器》的详细内容，更多关于的资料请关注golang学习网公众号！