/
server.go
93 lines (81 loc) · 1.82 KB
/
server.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
package main
import (
"encoding/json"
"fmt"
"github.com/boltdb/bolt"
"github.com/goji/param"
"github.com/zenazn/goji"
"github.com/zenazn/goji/web"
"log"
"net/http"
"strconv"
)
func insert(db *bolt.DB, contents string) {
_ = db.Update(func(tx *bolt.Tx) error {
bucket := tx.Bucket([]byte("MyBucket"))
c := bucket.Cursor()
k, _ := c.Last()
i, _ := strconv.Atoi(string(k[:]))
str := fmt.Sprintf("%06d", i+1)
bucket.Put([]byte(str), []byte(contents))
return nil
})
}
type Post struct {
PostId string
Contents string
}
func insertPost(w http.ResponseWriter, r *http.Request) {
db, err := bolt.Open("my.db", 0600, nil)
if err != nil {
log.Fatal(err)
}
var p Post
r.ParseForm()
err = param.Parse(r.Form, &p)
if err != nil {
log.Fatal(err)
}
insert(db, p.Contents)
w.Header().Set("Content-Type", "application/json")
fmt.Fprintf(w, "{\"status\": \"success\"}")
defer db.Close()
}
func readPosts(c web.C, w http.ResponseWriter, r *http.Request) {
db, err := bolt.Open("my.db", 0600, nil)
if err != nil {
log.Fatal(err)
}
var posts = make([]Post, 0, 0)
db.View(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte("MyBucket"))
b.ForEach(func(k, v []byte) error {
var p = Post{PostId: string(k[:]), Contents: string(v[:])}
posts = append(posts, p)
return nil
})
return nil
})
defer db.Close()
w.Header().Set("Content-Type", "application/json")
encoder := json.NewEncoder(w)
encoder.Encode(posts)
}
func createDB() {
db, err := bolt.Open("my.db", 0600, nil)
if err != nil {
log.Fatal(err)
}
db.Update(func(tx *bolt.Tx) error {
tx.CreateBucketIfNotExists([]byte("MyBucket"))
return nil
})
defer db.Close()
}
func main() {
createDB()
goji.Get("/api/posts/", readPosts)
goji.Post("/api/posts/", insertPost)
goji.Get("/*", http.FileServer(http.Dir("./static/")))
goji.Serve()
}