Як вийти з програми go на честь відкладених дзвінків?

Мені потрібно використовувати deferбезкоштовні розподіли, створені вручну за допомогою Cбібліотеки, але os.Exitв якийсь момент мені також потрібно зі статусом не 0. Хитра частина полягає в тому, що os.Exitпропускає будь-яку відкладену інструкцію:

package main

import "fmt"
import "os"

func main() {

    // `defer`s will _not_ be run when using `os.Exit`, so
    // this `fmt.Println` will never be called.
    defer fmt.Println("!")
    // sometimes ones might use defer to do critical operations
    // like close a database, remove a lock or free memory

    // Exit with status code.
    os.Exit(3)
}

Дитячий майданчик: http://play.golang.org/p/CDiAh9SXRM, викрадене з https://gobyexample.com/exit

Тож як вийти із програми go, що вшановує заявлені deferдзвінки? Чи існує якась альтернатива os.Exit?

runtime.Goexit() це найпростіший спосіб досягти цього.

Goexit припиняє програму, яка його викликає. Жоден інший горутин не впливає. Goexit запускає всі відкладені дзвінки перед тим, як завершити роботу програми. Оскільки Goexit не панікує, проте будь-які виклики відновлення в цих відкладених функціях повернуть нуль.

Однак:

Виклик Goexit з головної програми припиняє цю програму без повернення функції main. Оскільки func main не повернувся, програма продовжує виконувати інші програми. Якщо всі інші програми виходять, програма аварійно завершує роботу.

Отже, якщо ви називаєте це з основного горутину, у верхній частині mainпотрібно додати

defer os.Exit(0)

Нижче, можливо, ви захочете додати деякі інші deferзаяви, які повідомляють інші програми про зупинку та очищення.

Просто перемістіть програму вниз за рівнем і поверніть код виходу:

package main

import "fmt"
import "os"

func doTheStuff() int {
    defer fmt.Println("!")

    return 3
}

func main() {
    os.Exit(doTheStuff())
}

Після деяких досліджень, зверніться до цього , я знайшов альтернативу:

• Не нав'язує певну архітектуру, як у https://stackoverflow.com/a/27629493/438563
• Не вимагає жодного глобального значення, як у https://stackoverflow.com/a/24601700/438563

Ми можемо скористатися panicі recover. Виявляється, що panic, за своєю природою, буде виконувати deferвиклики, але також завжди виходитиме з 0кодом не стану і скидатиме трасування стека. Фокус у тому, що ми можемо замінити останній аспект панічної поведінки за допомогою:

package main

import "fmt"
import "os"

type Exit struct{ Code int }

// exit code handler
func handleExit() {
    if e := recover(); e != nil {
        if exit, ok := e.(Exit); ok == true {
            os.Exit(exit.Code)
        }
        panic(e) // not an Exit, bubble up
    }
}

Тепер, щоб вийти з програми в будь-який момент і зберегти будь-яку заявлену deferінструкцію, нам просто потрібно видати Exitтип:

func main() {
    defer handleExit() // plug the exit handler
    defer fmt.Println("cleaning...")
    panic(Exit{3}) // 3 is the exit code
}

Це не вимагає ніякого рефакторингу, крім підключення рядка всередині func main:

func main() {
    defer handleExit()
    // ready to go
}

Це досить добре масштабується з більшими базами коду, тому я залишу його доступним для вивчення. Сподіваюся, це допоможе.

Дитячий майданчик: http://play.golang.org/p/4tyWwhcX0-

Для нащадків це було більш елегантне рішення:

func main() { 
    retcode := 0
    defer func() { os.Exit(retcode) }()
    defer defer1()
    defer defer2()

    [...]

    if err != nil {
        retcode = 1
        return
    }
}

Я б запропонував використовувати горутин

func main(){
     defer fmt.Println("Hello world")
     go func(){
          os.Exit(0)
     }
}