Est-il possible de capturer un signal Ctrl+C (SIGINT) et d'exécuter une fonction de nettoyage, de manière "différée" ?

Question

Je veux capturer le Ctrl+C ( SIGINT ) envoyé par la console et d'imprimer des totaux partiels d'exécution.

Answer 1

Vous pouvez utiliser le os/signal pour traiter les signaux entrants. Ctrl + C es SIGINT Vous pouvez donc l'utiliser pour piéger os.Interrupt .

c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt)
go func(){
    for sig := range c {
        // sig is a ^C, handle it
    }
}()

La manière dont vous faites en sorte que votre programme se termine et imprime des informations vous appartient entièrement.

Answer 2

Ça marche :

package main

import (
    "fmt"
    "os"
    "os/signal"
    "syscall"
    "time" // or "runtime"
)

func cleanup() {
    fmt.Println("cleanup")
}

func main() {
    c := make(chan os.Signal)
    signal.Notify(c, os.Interrupt, syscall.SIGTERM)
    go func() {
        <-c
        cleanup()
        os.Exit(1)
    }()

    for {
        fmt.Println("sleeping...")
        time.Sleep(10 * time.Second) // or runtime.Gosched() or similar per @misterbee
    }
}

Passage en caisse dans la Terrain de jeux

Answer 3

Pour compléter légèrement les autres réponses, si vous voulez réellement attraper SIGTERM (le signal par défaut envoyé par la commande kill), vous pouvez utiliser syscall.SIGTERM à la place de os.Interrupt. Attention, l'interface syscall est spécifique au système et peut ne pas fonctionner partout (par exemple sous Windows). Mais elle fonctionne bien pour attraper les deux :

c := make(chan os.Signal, 2)
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
....

Answer 4

Il y avait (au moment de la publication) une ou deux petites fautes de frappe dans la réponse acceptée ci-dessus, voici donc la version corrigée. Dans cet exemple, j'arrête le profileur de CPU lorsque je reçois le message suivant Ctrl + C .

// capture ctrl+c and stop CPU profiler                            
c := make(chan os.Signal, 1)                                       
signal.Notify(c, os.Interrupt)                                     
go func() {                                                        
  for sig := range c {                                             
    log.Printf("captured %v, stopping profiler and exiting..", sig)
    pprof.StopCPUProfile()                                         
    os.Exit(1)                                                     
  }                                                                
}()

Answer 5

Tout ce qui précède semble fonctionner lorsqu'il est épissé, mais Page des signaux de gobyexample a un exemple vraiment propre et complet de capture de signal. Il mérite d'être ajouté à cette liste.

package main

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

func main() {
    sigs := make(chan os.Signal, 1)
    done := make(chan bool, 1)

    signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)

    go func() {
        sig := <-sigs
        fmt.Println()
        fmt.Println(sig)
        done <- true
    }()

    fmt.Println("awaiting signal")
    <-done
    fmt.Println("exiting")
}

Source : gobyexample.com/signals