Ist es möglich, ein Ctrl + C-Signal zu erfassen und eine Bereinigungsfunktion in einer "Defer" -Methode auszuführen?

Dies funktioniert:

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
    }
}

InformationsquelleAutor der Antwort Barry

Gab es (zum Zeitpunkt der Buchung) ein oder zwei kleine Tippfehler in der akzeptierten Antwort oben, also hier ist die gereinigte version. In diesem Beispiel bin ich stoppen des CPU-profiler beim Empfang von 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)                                                     
  }                                                                
}()    

InformationsquelleAutor der Antwort gravitron

Dies ist eine weitere version, die in den Fall, Sie haben einige Aufgaben zu bereinigen. Code hinterlässt clean-up-Prozess in Ihrer Methode.

package main

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

)



func main() {

    _,done1:=doSomething1()
    _,done2:=doSomething2()

    //do main thread


    println("wait for finish")
    <-done1
    <-done2
    fmt.Print("clean up done, can exit safely")

}

func doSomething1() (error, chan bool) {
    //do something
    done:=make(chan bool)
    c := make(chan os.Signal, 2)
    signal.Notify(c, os.Interrupt, syscall.SIGTERM)
    go func() {
        <-c
        //cleanup of something1
        done<-true
    }()
    return nil,done
}


func doSomething2() (error, chan bool) {
    //do something
    done:=make(chan bool)
    c := make(chan os.Signal, 2)
    signal.Notify(c, os.Interrupt, syscall.SIGTERM)
    go func() {
        <-c
        //cleanup of something2
        done<-true
    }()
    return nil,done
}

in den Fall, Sie brauchen, um saubere main-Funktion, die Sie erfassen müssen-signal in den Haupt-thread mit go func() sowie.

InformationsquelleAutor der Antwort Hlex

Können Sie eine andere goroutine, der erkennt, syscall.SIGINT und syscall.SIGTERM-Signale und leiten Sie dann an einen Kanal mit signal.Informieren. Sie können senden Sie einen Haken an, die goroutine mit einem Kanal und speichern Sie es in eine Funktion slice. Wenn das shutdown-signal erkannt wird, auf den Kanal, können Sie diese Funktionen ausführen, die in die Scheibe. Dies kann verwendet werden, zu bereinigen, die Ressourcen, warten Sie für die Ausführung von goroutines auf fertig stellen, speichern von Daten oder drucken von Teil-Summen.

Schrieb ich ein kleines und einfaches Dienstprogramm zum hinzufügen und führen Sie die Haken beim Herunterfahren. Hoffe, es kann helfen.

https://github.com/ankit-arora/go-utils/blob/master/go-shutdown-hook/shutdown-hook.go

Können Sie dies tun, in ein 'defer' Mode.

Beispiel für das Herunterfahren eines Servers anmutig :

srv := &http.Server{}

go_shutdown_hook.ADD(func() {
    log.Println("shutting down server")
    srv.Shutdown(nil)
    log.Println("shutting down server-done")
})

l, err := net.Listen("tcp", ":3090")

log.Println(srv.Serve(l))

go_shutdown_hook.Wait()

InformationsquelleAutor der Antwort Ankit Arora