Go-RESTful-创建RESTful API服务（三）

package main

import (
    "encoding/json"
    "log"
    "net/http"

    "github.com/emicklei/go-restful"
)

// 用户结构体
type User struct {
    ID    int    `json:"id"`
    Name  string `json:"name"`
    Email string `json:"email"`
}

// 用户列表
var users = []User{
    {ID: 1, Name: "Alice", Email: "alice@example.com"},
    {ID: 2, Name: "Bob", Email: "bob@example.com"},
    {ID: 3, Name: "Charlie", Email: "charlie@example.com"},
}

// 获取所有用户
func getUsers(request *restful.Request, response *restful.Response) {
    // 将用户列表转换为JSON格式，并写入HTTP响应
    response.WriteEntity(users)
}

// 创建一个新用户
func createUser(request *restful.Request, response *restful.Response) {
    // 从HTTP请求中读取用户信息
    var user User
    err := request.ReadEntity(&user)
    if err != nil {
        response.WriteErrorString(http.StatusBadRequest, err.Error())
        return
    }

    // 生成新的用户ID
    maxID := 0
    for _, u := range users {
        if u.ID > maxID {
            maxID = u.ID
        }
    }
    user.ID = maxID + 1

    // 将新用户添加到用户列表
    users = append(users, user)

    // 返回新用户信息
    response.WriteHeaderAndEntity(http.StatusCreated, user)
}

func main() {
    // 创建一个新的WebService对象
    ws := new(restful.WebService)

    // 定义路由
    ws.Route(ws.GET("/users").To(getUsers))
    ws.Route(ws.POST("/users").To(createUser))

    // 将WebService对象注册到RESTful框架中
    restful.Add(ws)

    // 启动HTTP服务器
    log.Fatal(http.ListenAndServe(":8080", nil))
}