Golangのコンテキストパッケージ：効果的な活用方法ガイド

package main

import (
    "context"
    "fmt"
    "time"
)

func main() {
    // 空のコンテキストを作成
    ctx := context.Background()

    // キャンセル可能なコンテキストを作成
    cancelCtx, cancel := context.WithCancel(ctx)
    defer cancel() // 使用後は必ずキャンセル関数を呼び出す

    // タイムアウト付きのコンテキストを作成
    timeoutCtx, cancel := context.WithTimeout(ctx, 2*time.Second)
    defer cancel()

    fmt.Println("Contexts created")
}

package main

import (
    "context"
    "fmt"
)

func main() {
    ctx := context.Background()

    // コンテキストに値を設定
    ctx = context.WithValue(ctx, "key", "value")

    // コンテキストから値を取得
    value := ctx.Value("key")
    fmt.Println("Value:", value)
}

package main

import (
    "context"
    "fmt"
    "time"
)

func worker(ctx context.Context) {
    for {
        select {
        case <-ctx.Done():
            fmt.Println("Worker: Stopping due to cancellation")
            return
        default:
            fmt.Println("Worker: Working...")
            time.Sleep(1 * time.Second)
        }
    }
}

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

    // 3秒後にキャンセル
    time.Sleep(3 * time.Second)
    cancel()

    // ワーカーが停止するのを待つ
    time.Sleep(1 * time.Second)
    fmt.Println("Main: Finished")
}

package main

import (
    "context"
    "fmt"
    "time"
)

func slowOperation(ctx context.Context) {
    select {
    case <-time.After(5 * time.Second):
        fmt.Println("Slow operation completed")
    case <-ctx.Done():
        fmt.Println("Slow operation canceled")
    }
}

func main() {
    ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
    defer cancel()

    go slowOperation(ctx)

    select {
    case <-ctx.Done():
        fmt.Println("Main: Operation timed out")
    }

    time.Sleep(1 * time.Second) // Wait for goroutine to print
}

package main

import (
    "context"
    "fmt"
    "net/http"
    "time"
)

func slowHandler(w http.ResponseWriter, r *http.Request) {
    ctx := r.Context()

    select {
    case <-time.After(5 * time.Second):
        fmt.Fprintln(w, "Slow operation completed")
    case <-ctx.Done():
        fmt.Println("Slow operation canceled by client")
        return
    }
}

func main() {
    http.HandleFunc("/slow", slowHandler)
    http.ListenAndServe(":8080", nil)
}

package main

import (
    "context"
    "database/sql"
    "fmt"
    "time"

    _ "github.com/go-sql-driver/mysql"
)

func queryWithTimeout(db *sql.DB) {
    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
    defer cancel()

    var result string
    err := db.QueryRowContext(ctx, "SELECT SLEEP(10)").Scan(&result)
    if err != nil {
        fmt.Println("Query error:", err)
        return
    }
    fmt.Println("Query result:", result)
}

func main() {
    db, err := sql.Open("mysql", "user:password@/dbname")
    if err != nil {
        panic(err)
    }
    defer db.Close()

    queryWithTimeout(db)
}

type key int

const userIDKey key = 0

func setUserID(ctx context.Context, userID string) context.Context {
    return context.WithValue(ctx, userIDKey, userID)
}

func getUserID(ctx context.Context) (string, bool) {
    userID, ok := ctx.Value(userIDKey).(string)
    return userID, ok
}