Golang应用程序在终端退出时如何发送邮件

你尝试过使用 nohup、disown 或 screen 来运行你的应用吗？

@samfrmd : 我想在终端被意外关闭时执行一个特定的操作。你认为 defer 在这里能起作用吗？

我不明白这样做有什么意义，如果终端关闭了，应用程序也会随之关闭。通过文件、进程ID或其他任何方式监控它们，都可以触发事件并发送邮件…

package main

import (
    "fmt"
    "log"
    "net/smtp"
    "os"
    "os/signal"
    "syscall"
    "time"

    "github.com/jordan-wright/email"
)

func main() {
    // 创建信号通道
    sigChan := make(chan os.Signal, 1)
    done := make(chan bool, 1)
    
    // 注册需要捕获的信号
    signal.Notify(sigChan, 
        syscall.SIGHUP,  // 终端关闭
        syscall.SIGINT,  // Ctrl+C
        syscall.SIGTERM, // 终止信号
        syscall.SIGQUIT, // Ctrl+\
    )

    // 启动信号处理goroutine
    go func() {
        sig := <-sigChan
        fmt.Printf("接收到信号: %v\n", sig)
        
        // 发送邮件
        if err := sendTerminationEmail(); err != nil {
            log.Printf("发送邮件失败: %v", err)
        } else {
            fmt.Println("邮件发送成功")
        }
        
        done <- true
    }()

    // 模拟应用程序运行
    fmt.Println("应用程序运行中...")
    fmt.Println("尝试关闭终端或按Ctrl+C触发邮件发送")
    
    // 等待信号
    <-done
    fmt.Println("应用程序退出")
}

func sendTerminationEmail() error {
    // 创建邮件
    e := email.NewEmail()
    e.From = "应用程序 <sender@example.com>"
    e.To = []string{"recipient@example.com"}
    e.Subject = "应用程序终端关闭通知"
    e.Text = []byte(fmt.Sprintf("应用程序于 %s 因终端关闭而退出", time.Now().Format(time.RFC3339)))
    e.HTML = []byte(fmt.Sprintf(`
        <h1>应用程序终止通知</h1>
        <p>应用程序于 <strong>%s</strong> 因终端关闭而退出</p>
        <p>请检查应用程序状态</p>
    `, time.Now().Format(time.RFC3339)))

    // 配置SMTP认证（根据实际情况修改）
    auth := smtp.PlainAuth("", "username", "password", "smtp.example.com")
    
    // 设置超时并发送邮件
    return e.Send("smtp.example.com:587", auth)
}

对于终端关闭时邮件发送失败的问题，可以添加以下改进：

package main

import (
    "context"
    "fmt"
    "log"
    "net/smtp"
    "os"
    "os/signal"
    "syscall"
    "time"

    "github.com/jordan-wright/email"
)

func main() {
    // 使用带缓冲的通道防止阻塞
    sigChan := make(chan os.Signal, 1)
    signal.Notify(sigChan, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM)

    // 创建带超时的context
    ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
    defer cancel()

    // 等待信号
    go func() {
        sig := <-sigChan
        log.Printf("接收到终止信号: %v", sig)
        
        // 在goroutine中发送邮件，避免阻塞信号处理
        go func() {
            if err := sendEmailWithRetry(ctx); err != nil {
                log.Printf("最终发送邮件失败: %v", err)
            }
            os.Exit(0)
        }()
        
        // 给邮件发送留出时间
        time.Sleep(2 * time.Second)
    }()

    // 主程序逻辑
    for {
        select {
        case <-ctx.Done():
            return
        default:
            time.Sleep(1 * time.Second)
        }
    }
}

func sendEmailWithRetry(ctx context.Context) error {
    maxRetries := 3
    for i := 0; i < maxRetries; i++ {
        select {
        case <-ctx.Done():
            return ctx.Err()
        default:
            if err := sendEmail(); err == nil {
                return nil
            }
            time.Sleep(time.Duration(i+1) * time.Second)
        }
    }
    return fmt.Errorf("发送邮件重试%d次后失败", maxRetries)
}

func sendEmail() error {
    e := email.NewEmail()
    e.From = "监控系统 <monitor@example.com>"
    e.To = []string{"admin@example.com"}
    e.Subject = "终端关闭警报"
    e.Text = []byte(fmt.Sprintf("终端于 %s 关闭", time.Now().Format("2006-01-02 15:04:05")))
    
    auth := smtp.PlainAuth("", "user", "pass", "smtp.example.com")
    return e.Send("smtp.example.com:587", auth)
}

关键改进点：

1. 使用带缓冲的信号通道
2. 添加超时控制
3. 实现重试机制
4. 在独立的goroutine中发送邮件避免阻塞
5. 确保有足够的时间完成邮件发送