从 Context 看 Go 设计模式：接口、封装和并发控制

本文介绍: 首先，需要明确的是，context.Context 是一个接口，而不是一个具体的数据结构。然而，当你传递一个指针时，你传递的是指针的副本，这个副本指向原始数据。如果 context 通过指针传递，暴露内部实现，那么在并发访问时，可能有就方式修改实际数据的内部状态。通过接口隐藏实现细节，context 的设计者可以确保内部状态的同步和一致性，而不需要用户介入。包的设计和实现，着重解析其非指针传值的原因，从而揭示 Go 并发模型背后的设计智慧。在 Go 中，context 的实现是通过结构体和指针的组合完成。

在这里插入图片描述

在 Go 语言中，
context 包是并发编程的核心，用于传递取消信号和请求范围的值。但其传值机制，特别是为什么不通过指针传递，而是通过接口，引发了我的一些思考。

考虑以下典型的代码片段：

package main

import "context"

func main() {
    ctx, cancel := context.WithCancel(context.Background())
    // ... call cancel() when specified signals are triggered

    handle(ctx)
}

func handle(ctx context.Context) error {
    return nil
}

这段代码展示了在 Go 中创建和传递 context 的简单用法。但背后的设计理念和实现细节却值得研究。

为什么 context 是以接口的形式传递，而非指针？这不仅涉及 Go 的并发哲学，还关系到封装性、并发安全性和接口的灵活性。

本文将简要探讨 context 包的设计和实现，着重解析其非指针传值的原因，从而揭示 Go 并发模型背后的设计智慧。

首先，如上的代码中，通过 context.WithCancel(context.Background()) 返回的是一个 context.Context 类型，而需要明确的是，context.Context 是一个接口，而不是一个具体的数据结构。

type Context interface {
    Deadline() (deadline time.Time, ok bool)
    Done() <-chan struct{}
    Err() error
    Value(key interface{}) interface{}
}

在 Go 中，context 的实现是通过结构体和指针的组合完成。例如，WithCancel 函数返回的 context.Context 类型实际上是一个指向 cancelCtx 结构体的指针。

func WithCancel(parent Context) (ctx Context, cancel CancelFunc) {
    c := newCancelCtx(parent)
    propagateCancel(parent, &c)
    return &c, func() { c.cancel(true, Canceled) }
}

在调用方，我们拿到 WithCancel 返回的指针，因为它的内部实现，满足 context.Context 的接口约束，能成功转为 Context 接口类型。

虽然 context 的某些实现（如 cancelCtx）在内部使用指针，但 context.Context 接口本身并不暴露任何指针。

通过将具体结构隐藏在接口后面，context 包确保了用户不能直接访问或修改内部状态，这是良好封装的标志。如其中的 timerCtx 保存了时间信息，而 valueCtx 保存了请求范围了的上下文信息，这些数据保证一致性和并发安全。

context.Context 是一个接口，这意味着你可以有多种实现。

如果 context 通过指针传递，那么所有的实现都必须是具体的结构体，如 handle 函数指定传递 cancelCtx 的话，那就不能传递 timerCtx、valueCtx 等其他类型 Context 实现类。而通过使用接口，Go 语言允许更多的灵活性和实现多样性。

我们已经完成了 context 包设计理念的探讨，尤其是它如何通过接口和封装来保证并发安全性，同时提供清晰的抽象。

我们要实现一个 datastore 实现存储数据，要求同时提供两种版本的实现：并发安全和无并发安全版本。

代码片段如下所示，它展示了 DataStore 接口的两种不同实现：safeDataStore 和 inMemoryDataStore。

DataStore 是一个接口，定义了对数据的操作。

DataStore 是一个接口的具体代码，如下所示：

type DataStore interface {
	ReadData() string
	WriteData(data string)
}

safeDataStore 是一个实现了 DataStore 接口的结构体，它使用 sync.Mutex 来保证并发安全.

type safeDataStore struct {
	mu   sync.Mutex
	data string
}

func (ds *safeDataStore) ReadData() string {
	ds.mu.Lock()
	defer ds.mu.Unlock()
	return ds.data
}

func (ds *safeDataStore) WriteData(data string) {
	ds.mu.Lock()
	defer ds.mu.Unlock()
	ds.data = data
}

inMemoryDataStore 是另一个实现了 DataStore 接口的结构体，它假设只在单个 goroutine 中使用，因此不需要同步机制


type inMemoryDataStore struct {
	data string
}

func (ds *inMemoryDataStore) ReadData() string {
	return ds.data
}

func (ds *inMemoryDataStore) WriteData(data string) {
	ds.data = data
}

如上的代码所示，DataStore 接口定义了数据存储的基本操作。同时，我们提供了两种实现：

func main() {
	var store DataStore

	// 使用 safeDataStore
	store = &safeDataStore{}
	store.WriteData("safe data")
	fmt.Println(store.ReadData())

	// 使用 inMemoryDataStore
	store = &inMemoryDataStore{}
	store.WriteData("in-memory data")
	fmt.Println(store.ReadData())
}