Context: บอกให้ทุกคนหยุดพร้อมกัน

1 · Context ส่งอะไรได้บ้าง

2 · context.WithCancel — กดกริ่งเอง

func main() {
    ctx, cancel := context.WithCancel(context.Background())

    go worker(ctx, "A")               // ตัว worker เขียนยังไง ดูหัวข้อ 3

    time.Sleep(2 * time.Second)
    fmt.Println("สั่งยกเลิก...")
    cancel()                              // ← กดกริ่ง

    time.Sleep(1 * time.Second)
}

   context.Background()          ← ต้นทาง ไม่มีวันถูกยกเลิก
          │
          ├── WithCancel  ──► ctx  ────────┬──────────┬──────────┐
          │                   │            │          │          │
       cancel()               ▼            ▼          ▼          ▼
          │              Handler       Service   Repository   Logger
          │                                │
          └──► ปิด ctx.Done() ──► ทุกคนที่ถือ ctx รู้พร้อมกัน

   cancel ที่ต้นทาง = ทุกกิ่งที่แตกออกไปได้สัญญาณหมด

3 · ctx.Done() — ช่องรับสัญญาณ

ctx.Done() คืน channel ที่จะถูกปิด เมื่อถึงเวลายกเลิก — เอามาใช้กับ select ตามที่ฝึกในบทเรียน 23:

func worker(ctx context.Context, name string) {
    for {
        select {
        case <-ctx.Done():
            fmt.Printf("[%s] ได้รับสัญญาณ หยุด: %v\n", name, ctx.Err())
            return                      // ← ต้อง return จริง ๆ

        default:
            fmt.Printf("[%s] ทำงาน...\n", name)
            time.Sleep(400 * time.Millisecond)
        }
    }
}

4 · context.WithTimeout — ตั้งเวลาไว้ล่วงหน้า

ctx, cancel := context.WithTimeout(
    context.Background(),
    2 * time.Second,
)
defer cancel()      // ← ต้องมีเสมอ แม้จะปล่อยให้หมดเวลาเอง

the cancel function returned by context.WithTimeout should be called,
not discarded, to avoid a context leak

5 · ctx.Err() — ถามว่าหยุดเพราะอะไร

if errors.Is(ctx.Err(), context.Canceled) {
    fmt.Println("มีคนกด cancel!")
}

if errors.Is(ctx.Err(), context.DeadlineExceeded) {
    fmt.Println("หมดเวลาทำงานแล้ว!")
}

สังเกตว่าใช้ errors.Is ไม่ใช่ == — หลักการเดิมจาก บทเรียน 12 เพราะ error อาจถูกห่อด้วย %w ระหว่างทาง

6 · การยกเลิกเป็นเรื่อง "ขอความร่วมมือ" ไม่ใช่ "บังคับ"

ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
defer cancel()

go worker(ctx, "A")      // worker วน select + Sleep 400ms
go worker(ctx, "B")

<-ctx.Done()               // รอจนหมดเวลา
time.Sleep(100 * time.Millisecond)
fmt.Println("DeadlineExceeded?", errors.Is(ctx.Err(), context.DeadlineExceeded))

[B] ทำงาน...
[A] ทำงาน...
[A] ทำงาน...
[B] ทำงาน...
[B] ทำงาน...
[A] ทำงาน...
DeadlineExceeded? true

รอให้ worker หยุดจริง ๆ ต้องใช้ WaitGroup คู่กับ context — worker จึงต้องรับ wg เพิ่มอีกตัว เพิ่มจากเวอร์ชันหัวข้อ 3 แค่สองบรรทัด ที่เหลือเหมือนเดิมทุกอย่าง:

func worker(ctx context.Context, name string, wg *sync.WaitGroup) {
    defer wg.Done()              // ← เพิ่มบรรทัดนี้
    for {
        select {
        case <-ctx.Done():
            fmt.Printf("[%s] ได้รับสัญญาณ หยุด: %v\n", name, ctx.Err())
            return

        default:
            fmt.Printf("[%s] ทำงาน...\n", name)
            time.Sleep(400 * time.Millisecond)
        }
    }
}

var wg sync.WaitGroup
wg.Add(2)
go worker(ctx, "A", &wg)
go worker(ctx, "B", &wg)

wg.Wait()      // ← รอให้เขาจบ "เอง" จริง ๆ
fmt.Println("ทุก worker หยุดเรียบร้อย")

คราวนี้ผลลัพธ์เปลี่ยนไป — ผมรันแล้วได้:

[A] ทำงาน...
[B] ทำงาน...
...
[A] ได้รับสัญญาณ หยุด: context deadline exceeded
[B] ได้รับสัญญาณ หยุด: context deadline exceeded
ทุก worker หยุดเรียบร้อย

7 · ctx ต้องเป็น parameter ตัวแรกเสมอ

func Handler(ctx context.Context, r *http.Request) { Service(ctx) }

func Service(ctx context.Context)                    { Repository(ctx) }

func Repository(ctx context.Context) {
    db.QueryRowContext(ctx, query, id)   // ส่งต่อจนถึง DB
}

8 · Background() กับ TODO() ต่างกันตรงไหน

ctx := context.Background()

ctx := context.TODO()

ทั้งสองตัวทำงานเหมือนกันเป๊ะ — ต่างกันแค่เจตนา ที่สื่อให้คนอ่านโค้ด (และเครื่องมือวิเคราะห์บางตัวใช้แยกได้)

9 · Graceful Shutdown — ปิดโดยไม่ตัดงานกลางคัน

ตอน deploy หรือ scale down (ลดจำนวนเครื่องที่รันอยู่ลง) ระบบจะส่ง SIGTERM มาบอกให้ปิด
SIGTERM คือสัญญาณมาตรฐานของระบบปฏิบัติการ แปลว่า “ขอให้ปิดตัวเองนะ” ไม่ใช่การฆ่าทิ้งทันที — โปรแกรมมีสิทธิ์เก็บของก่อนจบ แต่มีสิทธิ์ก็ต่อเมื่อเราดักไว้เอง (signal.NotifyContext ข้างล่างนี่แหละ) ถ้าไม่ดัก OS ก็ปิดให้ทันทีอยู่ดี · ตัวที่ดักไม่ได้เลยคือ SIGKILL — โดนแล้วจบทันที ไม่มีโอกาสเก็บของ
ถ้าปิดทันที = request ที่ค้างอยู่ถูกตัดกลางคัน = ลูกค้าจ่ายเงินแล้วแต่ order ไม่ถูกบันทึก

ctx, stop := signal.NotifyContext(
    context.Background(),
    os.Interrupt,        // Ctrl+C
    syscall.SIGTERM,     // docker stop / kubernetes
)
defer stop()

// ... เปิด server แล้วส่ง ctx ต่อไปเรื่อย ๆ ...

<-ctx.Done()                        // ค้างตรงนี้จนกว่าจะมีสัญญาณ
fmt.Println("ได้รับสัญญาณปิด:", ctx.Err())

// เก็บงานให้เรียบร้อยก่อนจบ
// context ตัวที่สอง: ให้เวลาเก็บงานค้าง 30 วินาที ไม่เสร็จก็ตัดจบ
shutdownCtx, cancelShutdown := context.WithTimeout(
    context.Background(), 30*time.Second)
defer cancelShutdown()

srv.Shutdown(shutdownCtx)   // srv คือ *http.Server ที่เปิดไว้ข้างบน — ชั้น HTTP เรียนสัปดาห์ที่ 3

signal.NotifyContext ทำให้สัญญาณจากระบบปฏิบัติการ กลายเป็น context ธรรมดา — ทุกอย่างที่คุณเรียนมาในบทนี้ใช้ได้ทันที

ผมทดลองส่ง SIGTERM ให้โปรแกรมจริง ๆ ได้แบบนี้:

server กำลังรับ request...
server กำลังรับ request...

ได้รับสัญญาณปิด: context canceled
กำลังปิดงานค้าง (drain)...
ปิดเรียบร้อย ไม่มี request ไหนถูกตัดกลางคัน

ทดสอบความเข้าใจ

parent 5s, child 1s → child จบที่ 1s
parent 1s, child 5s → child จบที่ 1s   ← ไม่ได้ 5s

อ้างอิง

• Go Concurrency Patterns: Context — Go Blog · ต้นทางของบทนี้
• pkg.go.dev — context · signal.NotifyContext
• Go Concurrency Patterns: Pipelines and cancellation
• Go Code Review Comments — Contexts — ธรรมเนียมการวาง ctx
• pkg.go.dev — http.Server.Shutdown