GF框架中自定义HTTP状态码的实现方法

在GF框架开发过程中，开发者经常需要根据业务逻辑返回特定的HTTP状态码。例如在创建资源时返回201 Created状态码，或者在实现SSO单点登录时需要使用301/302/307等重定向状态码。本文将详细介绍在GF框架中如何灵活控制HTTP响应状态码。

获取HTTP请求响应对象

GF框架提供了便捷的方法来访问底层的HTTP请求和响应对象。通过g.RequestFromCtx函数可以从上下文(context)中获取请求对象，进而操作响应状态码：

// 从上下文中获取请求对象
request := g.RequestFromCtx(ctx)

// 获取请求头信息
headers := request.Request.Header

// 获取响应对象
response := request.Response

设置HTTP状态码

获取到响应对象后，可以直接设置状态码：

// 设置201 Created状态码
g.RequestFromCtx(ctx).Response.WriteStatus(http.StatusCreated)

// 设置301永久重定向
g.RequestFromCtx(ctx).Response.WriteStatus(http.StatusMovedPermanently)

// 设置302临时重定向
g.RequestFromCtx(ctx).Response.WriteStatus(http.StatusFound)

实际应用场景示例

1. 创建资源返回201状态码

func CreateUser(ctx context.Context, req *CreateUserReq) (*CreateUserRes, error) {
    // 创建用户逻辑...
    if err := service.User().Create(ctx, req); err != nil {
        return nil, err
    }
    
    // 设置201 Created状态码
    g.RequestFromCtx(ctx).Response.WriteStatus(http.StatusCreated)
    return &CreateUserRes{ID: newUserID}, nil
}

2. 实现SSO重定向

func SSOLogin(ctx context.Context) error {
    // 验证逻辑...
    
    // 设置302重定向到SSO服务
    g.RequestFromCtx(ctx).Response.WriteStatus(http.StatusFound)
    g.RequestFromCtx(ctx).Response.Header().Set("Location", ssoURL)
    return nil
}

注意事项

1. 设置状态码应该在返回响应体之前完成
2. 重定向时除了设置状态码，还需要设置Location头
3. 某些状态码需要配合特定的响应头使用
4. 在中间件中也可以使用这种方式统一处理状态码

通过这种方式，GF框架开发者可以完全控制HTTP响应的状态码，满足各种业务场景的需求。这种设计既保持了框架的简洁性，又提供了足够的灵活性。