Golang 进字节求职攻略大全

package main

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

    "github.com/gorilla/mux"
)

type Person struct {
    ID   int    `json:"id"`
    Name string `json:"name"`
}

var people []Person

func main() {
    r := mux.NewRouter()
    r.HandleFunc("/people", getPeople).Methods("GET")
    r.HandleFunc("/people/{id}", getPerson).Methods("GET")
    r.HandleFunc("/people", createPerson).Methods("POST")
    r.HandleFunc("/people/{id}", updatePerson).Methods("PUT")
    r.HandleFunc("/people/{id}", deletePerson).Methods("DELETE")

    http.Handle("/", r)
    http.ListenAndServe(":8080", nil)
}

func getPeople(w http.ResponseWriter, r *http.Request) {
    json.NewEncoder(w).Encode(people)
}

func getPerson(w http.ResponseWriter, r *http.Request) {
    vars := mux.Vars(r)
    id := vars["id"]

    for _, p := range people {
        if p.ID == id {
            json.NewEncoder(w).Encode(p)
            return
        }
    }

    http.Error(w, "Person not found", http.StatusNotFound)
}

func createPerson(w http.ResponseWriter, r *http.Request) {
    var p Person
    json.NewDecoder(r.Body).Decode(&p)
    p.ID = len(people) + 1
    people = append(people, p)

    json.NewEncoder(w).Encode(p)
}

func updatePerson(w http.ResponseWriter, r *http.Request) {
    vars := mux.Vars(r)
    id := vars["id"]

    for i, p := range people {
        if p.ID == id {
            json.NewDecoder(r.Body).Decode(&p)
            people[i] = p
            json.NewEncoder(w).Encode(p)
            return
        }
    }

    http.Error(w, "Person not found", http.StatusNotFound)
}

func deletePerson(w http.ResponseWriter, r *http.Request) {
    vars := mux.Vars(r)
    id := vars["id"]

    for i, p := range people {
        if p.ID == id {
            people = append(people[:i], people[i+1:]...)
            w.WriteHeader(http.StatusNoContent)
            return
        }
    }

    http.Error(w, "Person not found", http.StatusNotFound)
}