Go言語でシンプルな JSON API サーバー

概要

HTTP で JSON を POST すると、JSON で応答を返すシンプルなサーバーを Go言語で作ってみたものです。

日本語文章を POST すると mecab でパースして、結果を返してくれます。

関連記事

インターフェース

リクエスト

HTTP で JSON形式のコマンドを POST します。
{"Token": "01234", "Sentence": "なにか文章"}

• Token: future reserve
• Sentence: mecab で解析したい一文(UTF-8)

リザルト

JSON形式で戻ってきます。
{"Status":0,"Code":"OK","Result":["A","B","C"]}

• Status: 0 で成功、それ以外で失敗
• Code: 成功時は OK、それ以外は失敗理由
• Result: 成功時 mecab の解析結果が array で戻る(UTF-8)

インストール

go install bitbucket.org/rerofumi/webmecab
でどぞー

実行

install で入れたのならば $GOPATH/bin に webmecab が居るはず。

usage: webmecab [-p port]

• p もしくは port でポート指定、無指定時 80

ソースコード

mecab-web-api.go

package main

import (
    "bitbucket.org/rerofumi/mecab"
    "fmt"
    "net/http"
    "encoding/json"
    "os"
    "flag"
    "strconv"
    "io"
    "bufio"
)

/* --- JSON define */
type MecabOutput struct {
    Status int
    Code string
    Result []string
}

type MecabInput struct {
    Token string
    Sentence string
}


/* ----------------------- */
/* --- controller          */
/* ----------------------- */
func apiRequest(w http.ResponseWriter, r *http.Request) {
    list := make([]string,0)
    ret := MecabOutput{0,"OK",list}
    request := ""

    // JSON return
    defer func() {
        // result
        outjson,err := json.Marshal(ret)
        if err != nil {
            fmt.Println(err) //TODO: change to log
        }
        w.Header().Set("Content-Type", "application/json")
        fmt.Fprint(w, string(outjson))
    }()

    // type check
    if r.Method != "POST" {
        ret.Status = 1
        ret.Code = "Not POST method"
        return
    }

    // request body
    rb := bufio.NewReader(r.Body)
    for {
        s, err := rb.ReadString('\n')
        request = request + s
        if err == io.EOF { break }
    }

    // JSON parse
    var dec MecabInput
    b := []byte(request)
    err := json.Unmarshal(b, &dec)
    if err != nil {
        ret.Status = 2
        ret.Code = "JSON parse error."
        return
    }

    // mecab parse
    result, err := mecab.Parse(dec.Sentence)
    if err == nil {
        for _, n := range result {
            ret.Result = append(ret.Result, n)
        }
    }

}


/* ----------------------- */
/* --- main                */
/* ----------------------- */
func main() {
    // switch
    var (portNum int)
    flag.IntVar(&portNum, "port", 80, "int flag")
    flag.IntVar(&portNum, "p", 80, "int flag")
    flag.Parse()

    var port string
    port = ":"+strconv.Itoa(portNum)
    fmt.Println("listen port =", port)

    // route handler
    http.HandleFunc("/", apiRequest)

    // do serve
    err := http.ListenAndServe(port, nil)

    // error abort
    if err != nil {
        fmt.Println(err)
        os.Exit(1)
    }

}