如何在for循环中实现立即触发的计时器机制？

Golang不知道大家是否熟悉？今天我将给大家介绍《如何在for循环中实现立即触发的计时器机制？》，这篇文章主要会讲到等等知识点，如果你在看完本篇文章后，有更好的建议或者发现哪里有问题，希望大家都能积极评论指出，谢谢！希望我们能一起加油进步！

我想编写一个 golang 程序，它作为 kubernetes 容器中的服务运行。程序应该一直运行并且不会自行终止 - 除非发生错误。如果由于 pod 应被删除而导致 sigterm 来自 kubelet / kubernetes，则程序应退出。

在我的第一种方法中，我实现了一个每分钟运行一次的 for 循环。在此循环中，程序从另一个服务查询资源并对其进行处理。然后，该服务应“休眠”一分钟，然后再次执行这些步骤。在“睡眠”阶段，如果有 sigterm，程序应立即响应。

func main() {
  
  c := make(chan os.Signal, 1)
  signal.Notify(c, os.Interrupt, syscall.SIGTERM)

  //restart loop every minute 
  ticker := time.NewTicker(60 * time.Second)
  defer ticker.Stop()

  // while true loop
  for {
    select {
    case <-c:
      fmt.Println("Break the loop")
      return
    case <-ticker.C:
      fmt.Println("Hello in a loop")
    }
  }
}

我当前的方法有两个问题。首先，第一次运行 for 循环时我必须等待一分钟。我可以以某种方式配置它，以便计时器仅在第一次运行后运行吗？

第二个问题是程序在运行之间不应执行任何操作 - 除了对 sigterm 做出反应之外。

我不确定我解决问题的方法是否正确。我刚刚开始使用 go 语言。也许有更好的方法来解决我的问题。

正确答案

您可以在 select 语句之外完成工作，然后在票证通道案例内继续。这将立即执行您的工作一次，然后在每次股票报价时执行。

您的函数看起来不错，但在可以的情况下使用 context 是一个很好的做法。

很多包（数据库、io 等）都可以选择指定上下文，这通常用于定义超时。在您的情况下，将相同（或子）上下文传递给这些包意味着它们也尊重 sigterm。

package main

import (
    "context"
    "fmt"
    "os"
    "os/signal"
    "syscall"
    "time"
)

func main() {
    ctx := cancelCtxOnSigterm(context.Background())
    startWork(ctx)
}

// cancelCtxOnSigterm returns a Context that will be cancelled when the program receives a sigterm.
func cancelCtxOnSigterm(ctx context.Context) context.Context {
    exitCh := make(chan os.Signal, 1)
    signal.Notify(exitCh, os.Interrupt, syscall.SIGTERM)

    ctx, cancel := context.WithCancel(ctx)
    go func() {
        <-exitCh
        cancel()
    }()
    return ctx
}

// startWork performs a task every 60 seconds until the context is done.
func startWork(ctx context.Context) {
    ticker := time.NewTicker(60 * time.Second)
    defer ticker.Stop()
    for {
        // Do work here so we don't need duplicate calls. It will run immediately, and again every minute as the loop continues.
        if err := work(ctx); err != nil {
            fmt.Printf("failed to do work: %s", err)
        }
        select {
        case <-ticker.C:
            continue
        case <-ctx.Done():
            return
        }
    }
}

func work(ctx context.Context) error {
    fmt.Println("doing work")
    return nil
}

好了，本文到此结束，带大家了解了《如何在for循环中实现立即触发的计时器机制？》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！