当前位置： > > > > 如何在golang中根据ID取消通道作业
来源：stackoverflow
2024-04-20 13:27:32
0浏览
收藏
一分耕耘，一分收获！既然打开了这篇文章《如何在golang中根据ID取消通道作业》，就坚持看下去吧！文中内容包含等等知识点…希望你能在阅读本文后，能真真实实学到知识或者帮你解决心中的疑惑，也欢迎大佬或者新人朋友们多留言评论，多给建议！谢谢！
问题内容
所以我有一个 post 端点，它创建一个作业并将它们添加到一个 chan 中。 workerjobschan = make(chan job, maxqueuesize)
这就是我在频道中执行作业的方式（main.go）：
for i := 1; i <= maxworkers; i++ {
go func(i int) {
for job := range workerjobschan {
ctx, cancel := context.withcancel(context.background())
storejob(job.search.id, cancel)
job.execute(ctx, c.db, i)
}
}(i)
}
我将取消函数存储在地图中：canceljobfuncs = make(map[int]context.cancelfunc)。
这是工作职能：
func (j *job) execute(ctx context.context, db *sql.db, workerid int) error {
for {
select {
// check for cancellation signal
case <-ctx.done():
if err := ctx.err(); err != nil {
fmt.println("worker", workerid, "error", err)
}
fmt.println("worker", workerid, "cancelled")
return nil
default:
fmt.printf("worker%d: processing %s\n", workerid, j.search.query)
time.sleep(2 * time.second)
fmt.printf("worker%d: active %s\n", workerid, j.search.query)
time.sleep(5 * time.second)
fmt.printf("worker%d: completed %s!\n", workerid, j.search.query)
}
}
}
我取消这样的上下文（在http处理程序中）：
cancelJob(search.ID)
但该作业仍在继续运行。我尝试了很多东西，但似乎无法让它发挥作用。
正确答案
这里有一种方法可以说明一个明显的观点：如果您的代码不检查 ctx.done()，它就无法知道它已被取消。
（顺便说一句，这是@jimb 在对您的问题的评论中所写内容的另一种改写）。
因此，当 .execute(...) 方法中的代码开始执行此块时：
fmt.printf("worker%d: processing %s\n", workerid, j.search.query)
time.sleep(2 * time.second)
fmt.printf("worker%d: active %s\n", workerid, j.search.query)
time.sleep(5 * time.second)
fmt.printf("worker%d: completed %s!\n", workerid, j.search.query)
它将到达该块的末尾（7 秒）。没有任何指令告诉它在取消时停止。
如果您希望函数能够在“睡眠”指令期间检测取消，则必须更改代码。
以下是如何使用您的示例执行此操作的一个示例：
func (j *job) execute(ctx context.context, db *sql.db, workerid int) error {
for {
fmt.printf("worker%d: processing %s\n", workerid, j.search.query)
// rewrite time.sleep() with time.after() so that it can be composed
// in a select statement:
select {
case <-ctx.done():
fmt.println("worker", workerid, "cancelled")
return nil
case <-time.after(2 * time.second):
// keep going
}
fmt.printf("worker%d: active %s\n", workerid, j.search.query)
select {
case <-ctx.done():
fmt.println("worker", workerid, "cancelled")
return nil
case <-time.after(5 * time.second):
// keep going
}
fmt.printf("worker%d: completed %s!\n", workerid, j.search.query)
}
}
我猜你的实际代码没有 time.sleep() 指令，而是 processsearch(...) 或 doquery(...) 或 …
如果您需要这些函数在执行期间可取消，则需要以某种方式将取消上下文传递给它们，并让它们以某种方式检查取消情况。
“传递上下文”的一种方法显然是将其添加到所述函数的参数中：
processsearch(ctx, ...) doquery(ctx, ...)
但根据您现有的代码，某些参数可能已经有内置的方法可以取消。
举例来说：
// an http.Request carries a context:
func doQuery(req *http.Request, ....) {
...
}
// at call site:
...
req := http.NewRequestWithContext(ctx, "GET", "https://some.other.service/", nil)
doQuery(req, ...)
到这里，我们也就讲完了《如何在golang中根据ID取消通道作业》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注公众号，带你了解更多关于的知识点！
