# Go
### app.go
```go=
package mainimport (
"net/http"
"mux-practice/routes"
)
func main() {
r := routes.NewRouter()
http.ListenAndServe(":8080", r)
}
```
- `net/http`: 提供前後端 http methods
- `mux-practice/routes`: 提供路由規則
- `HandleFunc`: API, 根據 URL 提供對應的 function
- `http.ListenAndServe` 根據 `mux.NewRouter()` 提供服務
### routes.go
```go=
package routes
import (
"net/http"
"mux-practice/controllers"
"github.com/gorilla/mux"
)
type Route struct {
Method string
Pattern string
Handler http.HandlerFunc
Middleware mux.MiddlewareFunc
}
var routes []Route
func init() {
register("GET", "/movies", controllers.AllMovies, nil)
register("GET", "/movies/{id}", controllers.FindMovie, nil)
register("POST", "/movies", controllers.CreateMovie, nil)
register("PUT", "/movies", controllers.UpdateMovie, nil)
register("DELETE", "/movies", controllers.DeleteMovie, nil)
}
func NewRouter() *mux.Router {
r := mux.NewRouter()
for _, route := range routes {
r.HandleFunc(route.Pattern, route.Handler).Methods(route.Method)
if route.Middleware != nil {
r.Use(route.Middleware)
}
}
return r
}
func register(method, pattern string, handler http.HandlerFunc, middleware mux.MiddlewareFunc) {
routes = append(routes, Route{method, pattern, handler, middleware})
}
```
- `routes` 是一個沒有限制長度的 slice (可變長度陣列)
- `register()` 的功能則是用 append 的方式新增 route 到 `routes` 中
---
#### 很好,讀一半才發現,這邊以下是 mangoDB
### models/db.go
```go=
package models
import (
"log"
"github.com/globalsign/mgo"
)
const (
host = "127.0.0.1:27017"
source = "admin"
user = "user"
pass = "1234"
)
var globalS *mgo.Session
func init() {
dialInfo := &mgo.DialInfo{
Addrs: []string{host},
Source: source,
Username: user,
Password: pass,
}
s, err := mgo.DialWithInfo(dialInfo)
if err != nil {
log.Fatalln("create session error", err)
}
globalS = s
}
func connect(db, collection string) (*mgo.Session, *mgo.Collection) {
s := globalS.Copy()
c := s.DB(db).C(collection)
return s, c
}
func Insert(db, collection string, docs ...interface{}) error {
ms, c := connect(db, collection)
defer ms.Close()
return c.Insert(docs...)
}
func FindOne(db, collection string, query, selector, result interface{}) error {
ms, c := connect(db, collection)
defer ms.Close()
return c.Find(query).Select(selector).One(result)
}
func FindAll(db, collection string, query, selector, result interface{}) error {
ms, c := connect(db, collection)
defer ms.Close()
return c.Find(query).Select(selector).All(result)
}
func Update(db, collection string, query, update interface{}) error {
ms, c := connect(db, collection)
defer ms.Close()
return c.Update(query, update)
}
func Remove(db, collection string, query interface{}) error {
ms, c := connect(db, collection)
defer ms.Close()
return c.Remove(query)
}
```
### models/movies.go
```go=
package models
import "github.com/globalsign/mgo/bson"
type Movies struct {
Id bson.ObjectId `bson:"_id" json:"id"`
Name string `bson:"name" json:"name"`
CoverImage string `bson:"cover_image" json:"cover_image"`
Description string `bson:"description" json:"description"`
}
const (
db = "Movies"
collection = "MovieModel"
)
func (m *Movies) InsertMovie(movie Movies) error {
return Insert(db, collection, movie)
}
func (m *Movies) FindAllMovies() ([]Movies, error) {
var result []Movies
err := FindAll(db, collection, nil, nil, &result)
return result, err
}
func (m *Movies) FindMovieById(id string) (Movies, error) {
var result Movies
err := FindOne(db, collection, bson.M{"_id": bson.ObjectIdHex(id)}, nil, &result)
return result, err
}
func (m *Movies) UpdateMovie(movie Movies) error {
return Update(db, collection, bson.M{"_id": movie.Id}, movie)
}
func (m *Movies) RemoveMovie(id string) error {
return Remove(db, collection, bson.M{"_id": bson.ObjectIdHex(id)})
}
```
### controllers/movies.go
```go=
package controllers
import (
"fmt"
"net/http"
)
func responseWithJson(w http.ResponseWriter, code int, payload interface{}) {
response, _ := json.Marshal(payload)
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(code)
w.Write(response)
}
func CreateMovie(w http.ResponseWriter, r *http.Request) {
defer r.Body.Close()
var movie models.Movies
if err := json.NewDecoder(r.Body).Decode(&movie); err != nil {
responseWithJson(w, http.StatusBadRequest, "Invalid request payload")
return
}
movie.Id = bson.NewObjectId()
if err := dao.InsertMovie(movie); err != nil {
responseWithJson(w, http.StatusInternalServerError, err.Error())
return
}
responseWithJson(w, http.StatusCreated, movie)
}
func AllMovies(w http.ResponseWriter, r *http.Request) {
defer r.Body.Close()
var movies []models.Movies
movies, err := dao.FindAllMovies()
if err != nil {
responseWithJson(w, http.StatusInternalServerError, err.Error())
return
}
responseWithJson(w, http.StatusOK, movies)
}
func FindMovie(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
id := vars["id"]
result, err := dao.FindMovieById(id)
if err != nil {
responseWithJson(w, http.StatusInternalServerError, err.Error())
return
}
responseWithJson(w, http.StatusOK, result)
}
func UpdateMovie(w http.ResponseWriter, r *http.Request) {
defer r.Body.Close()
var params models.Movies
if err := json.NewDecoder(r.Body).Decode(¶ms); err != nil {
responseWithJson(w, http.StatusBadRequest, "Invalid request payload")
return
}
if err := dao.UpdateMovie(params); err != nil {
responseWithJson(w, http.StatusInternalServerError, err.Error())
return
}
responseWithJson(w, http.StatusOK, map[string]string{"result": "success"})
}
func DeleteMovie(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
id := vars["id"]
if err := dao.RemoveMovie(id); err != nil {
responseWithJson(w, http.StatusBadRequest, "Invalid request payload")
return
}
responseWithJson(w, http.StatusOK, map[string]string{"result": "success"})
}
```
---
### Reference
[從這邊整理 ](https://medium.com/@intheblackworld/%E5%89%8D%E7%AB%AF%E6%94%BB%E5%9F%8E%E8%A9%A9-let-s-golang-6-golang-%E9%96%8B-api-%E4%B8%8A-5e82cde62853)
##### tags: `tutorial`