|
45
|
1 /*
|
|
47
|
2 Last Change: 2020-04-21 火 13:59:52.
|
|
45
|
3 */
|
|
|
4
|
|
|
5 package main
|
|
|
6
|
|
|
7 import (
|
|
47
|
8 "bytes"
|
|
|
9 "compress/gzip"
|
|
45
|
10 "encoding/csv"
|
|
|
11 "fmt"
|
|
|
12 "io"
|
|
|
13 "io/ioutil"
|
|
|
14 "log"
|
|
|
15 "net"
|
|
|
16 "net/http"
|
|
|
17 "os"
|
|
|
18 "path/filepath"
|
|
|
19 "strconv"
|
|
|
20 "strings"
|
|
|
21 )
|
|
|
22
|
|
|
23 type hhs struct {
|
|
|
24 No string
|
|
|
25 //Birth string
|
|
|
26 Name string
|
|
|
27 //Kana string
|
|
|
28 Addr string
|
|
|
29 //Sex string
|
|
|
30 Ccn []string
|
|
|
31 }
|
|
|
32
|
|
|
33 func (h *hhs) GetData() string {
|
|
|
34 s := strings.Join(h.Ccn, "#")
|
|
|
35 s = strings.Join([]string{h.Name, h.Addr, s}, ":")
|
|
|
36 return s
|
|
|
37 }
|
|
|
38
|
|
|
39 func (h *hhs) GetRecent() string {
|
|
|
40 ccn := ""
|
|
|
41 if len(h.Ccn) > 0 {
|
|
|
42 ccn = h.Ccn[0]
|
|
|
43 }
|
|
|
44 return strings.Join([]string{h.No, h.Name, ccn}, ",")
|
|
|
45 }
|
|
|
46
|
|
|
47 var (
|
|
|
48 server string
|
|
|
49 port string
|
|
|
50 hhsdb string
|
|
|
51 img_root string
|
|
|
52 hhash map[string]hhs
|
|
|
53 iymdhash map[string]string
|
|
|
54 iyhash map[string]string
|
|
|
55 )
|
|
|
56
|
|
|
57 func init() {
|
|
|
58 port = ":3910"
|
|
|
59 hhsdb = "hhsdb.csv"
|
|
|
60 img_root = "./images"
|
|
|
61 hhash = make(map[string]hhs)
|
|
|
62 iymdhash = make(map[string]string)
|
|
|
63 iyhash = make(map[string]string)
|
|
|
64 }
|
|
|
65
|
|
|
66 func main() {
|
|
|
67
|
|
|
68 // setting IP-Address & Port
|
|
|
69 addrs, err := net.InterfaceAddrs()
|
|
|
70 if err != nil {
|
|
|
71 log.Fatal(err)
|
|
|
72 }
|
|
|
73 for _, a := range addrs {
|
|
|
74 if ipnet, ok := a.(*net.IPNet); ok && !ipnet.IP.IsLoopback() {
|
|
|
75 if strings.HasPrefix(ipnet.IP.String(), "169.254") {
|
|
|
76 continue
|
|
|
77 }
|
|
|
78 if ipnet.IP.To4() != nil {
|
|
|
79 server = ipnet.IP.String() + port
|
|
|
80 }
|
|
|
81 }
|
|
|
82 }
|
|
|
83
|
|
|
84 if err := loadDB(); err != nil {
|
|
|
85 log.Fatal(err)
|
|
|
86 }
|
|
|
87
|
|
|
88 // Http-Handler
|
|
46
|
89 http.HandleFunc("/h/", hhs_handler) // Get /h/0800012345 -> name:addr:20200101#20210701#...
|
|
|
90 http.HandleFunc("/hn/", hhsnm_handler) // Get /h/0800012345:0800098765:... -> name1:name2:...
|
|
47
|
91 http.HandleFunc("/ht/", hhstm_handler) // Get /ht/ -> 2020-03-14 12:34
|
|
46
|
92 http.HandleFunc("/ha/", hhsdb_handler) // Get /ha/ -> hhsdb.csv for Mover
|
|
|
93 http.HandleFunc("/i/", image_handler) // Get /i/20200110/0800012345.tgz
|
|
|
94 http.HandleFunc("/r/", recent_handler) // Get /r/0800012345:0800067890:0800099999:... -> 0800012345,name1,20200101:0800067890,name2,20210405:...
|
|
|
95 http.HandleFunc("/d/", index_handler) // Get /d/20xx -> 20xx0401:2020xx0408:... , /d/20xx0401 -> 0800012345:0800098765:...
|
|
47
|
96 http.HandleFunc("/u/", uphhsdb_handler) // POST /u/
|
|
|
97 http.HandleFunc("/ui/", upimage_handler) // POST /ui/20200401/0800012345.tgz
|
|
45
|
98
|
|
|
99 log.Fatal(http.ListenAndServe(server, nil))
|
|
|
100 }
|
|
|
101
|
|
|
102 func loadDB() error {
|
|
|
103 b, err := ioutil.ReadFile(hhsdb)
|
|
|
104 if err != nil {
|
|
|
105 return err
|
|
|
106 }
|
|
|
107 r := csv.NewReader(strings.NewReader(string(b)))
|
|
|
108 for {
|
|
|
109 record, err := r.Read()
|
|
|
110 if err == io.EOF {
|
|
|
111 break
|
|
|
112 }
|
|
|
113 if err != nil {
|
|
|
114 return err
|
|
|
115 }
|
|
|
116 h := hhs{
|
|
|
117 No: record[0],
|
|
|
118 //Birth: record[1],
|
|
|
119 Name: record[2],
|
|
|
120 //Kana: record[3],
|
|
|
121 Addr: record[4],
|
|
|
122 //Sex: record[5],
|
|
|
123 }
|
|
|
124 hhash[record[0]] = h
|
|
|
125 }
|
|
|
126
|
|
|
127 b, err = ioutil.ReadFile("index.csv")
|
|
|
128 if err != nil {
|
|
|
129 return err
|
|
|
130 }
|
|
|
131 r = csv.NewReader(strings.NewReader(string(b)))
|
|
|
132 for {
|
|
|
133 record, err := r.Read()
|
|
|
134 if err == io.EOF {
|
|
|
135 break
|
|
|
136 }
|
|
|
137 if err != nil {
|
|
|
138 return err
|
|
|
139 }
|
|
|
140 h := hhash[record[0]]
|
|
|
141 ccn := h.Ccn
|
|
|
142 h.Ccn = append(ccn, record[1])
|
|
|
143 hhash[record[0]] = h
|
|
|
144
|
|
|
145 iymdhash[record[1]] += ":" + record[0]
|
|
|
146 }
|
|
|
147
|
|
|
148 for ymd, _ := range iymdhash {
|
|
|
149 y := ymd[0:4]
|
|
|
150 if ymd[4:6] == "01" || ymd[4:6] == "02" || ymd[4:6] == "03" {
|
|
|
151 yy, _ := strconv.Atoi(y)
|
|
|
152 y = fmt.Sprintf("%d", yy - 1)
|
|
|
153 }
|
|
|
154 iyhash[y] += ":" + ymd
|
|
|
155 }
|
|
|
156
|
|
|
157 return nil
|
|
|
158 }
|
|
|
159
|
|
46
|
160 /* Get /h/0800012345 -> name:addr:20200101#20210701#... */
|
|
45
|
161 func hhs_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
162 hno := r.URL.Path[len("/h/"):]
|
|
|
163 s := ""
|
|
|
164 if h, ok := hhash[hno]; ok {
|
|
|
165 s = h.GetData()
|
|
|
166 }
|
|
|
167 w.Write([]byte(s))
|
|
|
168 }
|
|
|
169
|
|
46
|
170 /* Get /hn/0800012345:0800098765:... -> name1:name2:... */
|
|
|
171 func hhsnm_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
172 hnolist := strings.Split(r.URL.Path[len("/hn/"):], ":")
|
|
|
173 var buf []string
|
|
|
174 for _, hno := range hnolist {
|
|
|
175 var n string
|
|
|
176 if h, ok := hhash[hno]; ok {
|
|
|
177 n = h.Name
|
|
|
178 }
|
|
|
179 buf = append(buf, n)
|
|
|
180 }
|
|
|
181 w.Write([]byte(strings.Join(buf, ":")))
|
|
|
182 }
|
|
|
183
|
|
47
|
184 /* Get /ht/ -> 2020-03-14 12:34 */
|
|
46
|
185 func hhstm_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
186 date := ""
|
|
47
|
187 if fi, err := os.Stat(hhsdb); err == nil {
|
|
|
188 t := fi.ModTime()
|
|
|
189 date = t.Format("2006-01-02 15:04")
|
|
|
190 }
|
|
46
|
191 w.Write([]byte(date))
|
|
|
192 }
|
|
|
193
|
|
45
|
194 /* Get /ha/ -> hhsdb.csv for Mover */
|
|
|
195 func hhsdb_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
196 b, _ := ioutil.ReadFile(hhsdb)
|
|
|
197 w.Write(b)
|
|
|
198 }
|
|
|
199
|
|
|
200 /* Get /i/20200110/0800012345.tgz */
|
|
|
201 func image_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
202 file := r.URL.Path[len("/i/"):]
|
|
|
203 file = filepath.Join(img_root, filepath.FromSlash(file))
|
|
|
204
|
|
|
205 f, err := os.Open(file)
|
|
|
206 if err != nil {
|
|
|
207 http.NotFound(w, r)
|
|
|
208 return
|
|
|
209 }
|
|
|
210 defer f.Close()
|
|
|
211
|
|
|
212 fi, _ := f.Stat()
|
|
|
213
|
|
|
214 w.Header().Set("Content-Type", "rsearcher/octet-stream")
|
|
|
215 w.Header().Set("Content-Length", fmt.Sprintf("%d", fi.Size()))
|
|
|
216 io.Copy(w, f)
|
|
|
217 }
|
|
|
218
|
|
46
|
219 /* Get /r/0800012345:0800067890:0800099999:... -> 0800012345,name1,20200101:0800067890,name2,20210405:... */
|
|
45
|
220 func recent_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
221 hnolist := strings.Split(r.URL.Path[len("/r/"):], ":")
|
|
|
222 var buf []string
|
|
|
223 for _, hno := range hnolist {
|
|
|
224 s := hno + ",,"
|
|
|
225 if h, ok := hhash[hno]; ok {
|
|
|
226 s = h.GetRecent()
|
|
|
227 if h.No == "" {
|
|
|
228 s = hno + s
|
|
|
229 }
|
|
|
230 }
|
|
|
231 buf = append(buf, s)
|
|
|
232 }
|
|
|
233 w.Write([]byte(strings.Join(buf, ":")))
|
|
|
234 }
|
|
|
235
|
|
|
236 /* Get /d/20xx -> 20xx0401:2020xx0408:... , /d/20xx0401 -> 0800012345:0800098765:... */
|
|
|
237 func index_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
238 var buf string
|
|
|
239 ymd := r.URL.Path[len("/d/"):]
|
|
|
240 if len(ymd) == 4 {
|
|
|
241 buf = iyhash[ymd]
|
|
|
242 }
|
|
|
243 if len(ymd) == 8 {
|
|
|
244 buf = iymdhash[ymd]
|
|
|
245 }
|
|
|
246 w.Write([]byte(buf[1:]))
|
|
|
247 }
|
|
|
248
|
|
47
|
249 /* POST /u/ */
|
|
45
|
250 func uphhsdb_handler(w http.ResponseWriter, r *http.Request) {
|
|
47
|
251 if r.Method != http.MethodPost {
|
|
|
252 http.NotFound(w, r)
|
|
|
253 return
|
|
|
254 }
|
|
|
255
|
|
45
|
256 server_root := filepath.Dir(os.Args[0])
|
|
47
|
257 file := filepath.Join(server_root, hhsdb)
|
|
45
|
258 f, err := os.Create(file)
|
|
|
259 if err != nil {
|
|
|
260 http.NotFound(w, r)
|
|
|
261 return
|
|
|
262 }
|
|
47
|
263 defer f.Close()
|
|
|
264
|
|
|
265 b, err := ioutil.ReadAll(r.Body)
|
|
|
266 r.Body.Close()
|
|
|
267 if err != nil {
|
|
|
268 http.NotFound(w, r)
|
|
|
269 return
|
|
|
270 }
|
|
|
271
|
|
|
272 br := bytes.NewReader(b)
|
|
|
273 zr, err := gzip.NewReader(br)
|
|
|
274 if err != nil {
|
|
|
275 http.NotFound(w, r)
|
|
|
276 return
|
|
|
277 }
|
|
|
278 n, err := io.Copy(f, zr)
|
|
|
279
|
|
|
280 if err := zr.Close(); err != nil {
|
|
|
281 http.NotFound(w, r)
|
|
|
282 return
|
|
|
283 }
|
|
|
284
|
|
|
285 w.Write([]byte(fmt.Sprintf("%d bytes are recieved.\n", n)))
|
|
|
286
|
|
|
287 loadDB();
|
|
|
288 }
|
|
|
289
|
|
|
290 /* POST /ui/20200401/0800012345.tgz */
|
|
|
291 func upimage_handler(w http.ResponseWriter, r *http.Request) {
|
|
|
292 if r.Method != http.MethodPost {
|
|
|
293 http.NotFound(w, r)
|
|
|
294 return
|
|
|
295 }
|
|
|
296
|
|
|
297 uri := r.URL.Path[len("/ui/"):]
|
|
|
298 ymd := uri[0:8]
|
|
|
299 tgz := uri[9:]
|
|
|
300
|
|
|
301 server_root := filepath.Dir(os.Args[0])
|
|
|
302 dir := filepath.Join(server_root, "images", ymd)
|
|
|
303
|
|
|
304 if _, err := os.Stat(dir); os.IsNotExist(err) {
|
|
|
305 os.Mkdir(dir, 0644)
|
|
|
306 }
|
|
|
307
|
|
|
308 file := filepath.Join(server_root, "images", ymd, tgz)
|
|
|
309 f, err := os.Create(file)
|
|
|
310 if err != nil {
|
|
|
311 http.NotFound(w, r)
|
|
|
312 return
|
|
|
313 }
|
|
|
314 defer f.Close()
|
|
|
315
|
|
45
|
316 n, err := io.Copy(f, r.Body)
|
|
|
317 if err != nil {
|
|
|
318 http.NotFound(w, r)
|
|
|
319 return
|
|
|
320 }
|
|
|
321 w.Write([]byte(fmt.Sprintf("%d bytes are recieved.\n", n)))
|
|
|
322 }
|
|
|
323
|
