如何实现带有在第一个循环中立即运行的定时器的for循环机制？

Question

我想编写一个 golang 程序，运行作为 kubernetes 容器中的服务。该程序应该始终运行并且不会自行终止 - 除非出现错误。如果一个 SIGTERM 来自 kubelet / kubernetes 因为 pod 应该被删除，程序应该退出。

在我的第一种方法中，我实施了一个 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 语言。也许有更好的方法来解决我的问题。

Answer 1

您可以在 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循环机制？

How to implement a for loop mechanism with a timer that runs immediately in the first loop?

go