mirror of
https://github.com/schollz/croc.git
synced 2025-10-11 13:21:00 +02:00
added signal for programme termination
This commit is contained in:
parent
defee4b4f6
commit
95de3790d7
1 changed files with 23 additions and 3 deletions
20
main.go
20
main.go
|
@ -6,6 +6,9 @@ package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"log"
|
"log"
|
||||||
|
"os"
|
||||||
|
"os/signal"
|
||||||
|
"syscall"
|
||||||
|
|
||||||
"github.com/schollz/croc/v10/src/cli"
|
"github.com/schollz/croc/v10/src/cli"
|
||||||
)
|
)
|
||||||
|
@ -27,7 +30,24 @@ func main() {
|
||||||
// fmt.Println("wrote profile")
|
// fmt.Println("wrote profile")
|
||||||
// }
|
// }
|
||||||
// }()
|
// }()
|
||||||
|
|
||||||
|
// Create a channel to receive OS signals
|
||||||
|
sigs := make(chan os.Signal, 1)
|
||||||
|
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
|
||||||
|
|
||||||
|
go func() {
|
||||||
if err := cli.Run(); err != nil {
|
if err := cli.Run(); err != nil {
|
||||||
log.Fatalln(err)
|
log.Fatalln(err)
|
||||||
}
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
// Wait for a termination signal
|
||||||
|
sig := <-sigs
|
||||||
|
log.Println("Received signal:", sig)
|
||||||
|
|
||||||
|
// Perform any necessary cleanup here
|
||||||
|
log.Println("Performing cleanup...")
|
||||||
|
|
||||||
|
// Exit the program gracefully
|
||||||
|
os.Exit(0)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue