-
Notifications
You must be signed in to change notification settings - Fork 0
/
fugu.go
72 lines (64 loc) · 1.62 KB
/
fugu.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
package router
import (
"encoding/json"
"github.com/go-chi/chi"
"github.com/lnquy/fugu/languages"
"github.com/lnquy/fugu/languages/base"
"github.com/lnquy/fugu/modules/global"
"io/ioutil"
"net/http"
)
func CalcStructSizeOf(w http.ResponseWriter, r *http.Request) {
lang, arch, data, err := getParams(r)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
ret, err := languages.CalcSizeOf(string(data), lang, arch)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.Header().Set("content-type", "application/json")
w.Write([]byte(ret))
}
func OptimizeStructMem(w http.ResponseWriter, r *http.Request) {
lang, arch, data, err := getParams(r)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
s := &base.Struct{}
if err = json.Unmarshal(data, s); err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
ret, err := languages.OptimizeMem(s, lang, arch)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.Header().Set("content-type", "application/json")
w.Write([]byte(ret))
}
func getParams(r *http.Request) (global.Language, global.Architecture, []byte, error) {
var lang global.Language
parm := chi.URLParam(r, "lang")
if parm != "" {
lang = global.LanguageEnum(parm)
} else {
lang = global.Go
}
var arch global.Architecture
parm = chi.URLParam(r, "arch")
if parm != "" {
arch = global.ArchitectureEnum(parm)
} else {
arch = global.Amd64
}
data, err := ioutil.ReadAll(r.Body)
if err != nil {
return 0, 0, nil, err
}
return lang, arch, data, nil
}