context

context.Context

• struct에 컨텍스트를 저장하지 않습니다
• 일반적으로 ctx context.Context를 첫번째 파라미터로 선언합니다
• nil 컨텍스트를 전달하지 않습니다
• 절달해야할 컨텍스트가 없거나 확실치 않다면 context.TODO()를 전달합니다

func DoSomething(ctx context.Context, arg Arg) error {
	// ctx 사용
	return nil
}

context.WithValue

func WithValue(parent Context, key, val any) Context {
	if parent == nil {
		panic("cannot create context from nil parent")
	}
	if key == nil {
		panic("nil key")
	}
	if !reflectlite.TypeOf(key).Comparable() {
		panic("key is not comparable")
	}
	return &valueCtx{parent, key, val}
}

type valueCtx struct {
	Context
	key, val any
}

context.WithValue는 부모 컨텍스트를 멤버 변수로 갖는 컨텍스트를 생성합니다.

func (c *valueCtx) Value(key any) any {
	// 자신(n 세대)이 key에 맞는 Context인 경우 자신의 값을 반환
	if c.key == key {
		return c.val
	}

	// 아닌 경우 부모(n-1 세대) Context를 확인
	return value(c.Context, key)
}

func value(c Context, key any) any {
	for {
		switch ctx := c.(type) {
		case *valueCtx:
			// 부모(n-1 세대)가 key에 맞는 Context인 경우 부모의 값을 반환
			if key == ctx.key {
				return ctx.val
			}

			// 아닌 경우 부모의 부모(n-2 세대) Context를 탐색하기 위해 준비
			c = ctx.Context
		case *cancelCtx:
			// 내부 코드를 위한 처리
			if key == &cancelCtxKey {
				return c
			}
			c = ctx.Context
		case *timerCtx:
			// 내부 코드를 위한 처리
			if key == &cancelCtxKey {
				return ctx.cancelCtx
			}
			c = ctx.Context
		case *emptyCtx:
			// n-1 세대 == 1 세대 Context인 경우 더 이상 찾을 수 없으므로 nil 반환
			return nil
		default:
			// n-2 세대 Context 확인
			return c.Value(key)
		}
	}
}

warning

Context를 키-값 저장소로 너무 많이 사용하면 성능에 영향을 줄 수 있습니다

context.WithCancel

func WithCancel(parent Context) (ctx Context, cancel CancelFunc) {
	c := withCancel(parent)
	return c, func() { c.cancel(true, Canceled, nil) }
}

func withCancel(parent Context) *cancelCtx {
	if parent == nil {
		panic("cannot create context from nil parent")
	}
	c := newCancelCtx(parent)
	propagateCancel(parent, c)
	return c
}

type cancelCtx struct {
	Context

	mu       sync.Mutex            // protects following fields
	done     atomic.Value          // of chan struct{}, created lazily, closed by first cancel call
	children map[canceler]struct{} // set to nil by the first cancel call
	err      error                 // set to non-nil by the first cancel call
	cause    error                 // set to non-nil by the first cancel call
}

context.WithCancel은 부모 컨텍스트를 멤버 변수로 갖는 컨텍스트를 생성하고, 부모 컨텍스트가 취소되면 자신도 취소되도록 설계되어 있습니다.

WithCancelCause(parent Context) (ctx Context, cancel CancelCauseFunc)
WithDeadline(parent Context, d time.Time) (Context, CancelFunc)
WithTimeout(parent Context, timeout time.Duration) (Context, CancelFunc)