Go实现多租户示例

package main
import (  "context"  "fmt"  "net/http")
// 租户上下文键type contextKey stringvar tenantKey contextKey = "tenant"
// 中间件以从请求中提取租户ID，并将租户ID设置到context中func tenantMiddleware(next http.Handler) http.Handler {  return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {    tenantID := r.Header.Get("X-Tenant-ID") // 假设租户ID是通过HTTP Header传入的    if tenantID == "" {      http.Error(w, "Tenant ID is required", http.StatusBadRequest)      return    }    ctxWithTenantID := context.WithValue(r.Context(), tenantKey, tenantID)    next.ServeHTTP(w, r.WithContext(ctxWithTenantID))  })}
// 数据库查询函数可以这样使用上下文来获取租户IDfunc queryDatabase(ctx context.Context) {  tenantID, ok := ctx.Value(tenantKey).(string)  if !ok {    // 处理错误情况    fmt.Println("Tenant ID not found in context")    return  }  // 使用租户ID来进行查询...  fmt.Printf("Querying database for tenant ID: %s\n", tenantID)}
// 一个示例HTTP处理函数func myHandler(w http.ResponseWriter, r *http.Request) {  // 使用之前存储在context中的租户ID  queryDatabase(r.Context())  fmt.Fprintf(w, "Handled request for tenant\n")}
func main() {  http.Handle("/", tenantMiddleware(http.HandlerFunc(myHandler)))  http.ListenAndServe(":8080", nil)}

// 假设我们有一个函数来验证请求是否已认证，并且返回已认证的用户信息func authenticateRequest(r *http.Request) (*UserInfo, error) {    // 实现身份认证逻辑，失败时返回错误}
func tenantMiddleware(next http.Handler) http.Handler {    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {        userInfo, err := authenticateRequest(r)        if err != nil {            http.Error(w, "Authentication failed", http.StatusUnauthorized)            return        }
        tenantID := r.Header.Get("X-Tenant-ID")        if tenantID == "" || userInfo.TenantID != tenantID {            http.Error(w, "Invalid tenant ID", http.StatusBadRequest)            return        }
        ctxWithTenantID := context.WithValue(r.Context(), tenantKey, tenantID)        next.ServeHTTP(w, r.WithContext(ctxWithTenantID))    })}
func queryDatabase(ctx context.Context) {    tenantID, ok := ctx.Value(tenantKey).(string)    if !ok {        // 处理错误情况        fmt.Println("Tenant ID not found in context")        return    }    // 在执行数据库操作之前，确保所有请求都使用tenantID过滤    fmt.Printf("Querying database for tenant ID: %s\n", tenantID)    // 执行实际的数据库查询，确保使用tenantID作为查询条件    // ... 查询逻辑 ...}