Add goctl kotlin support
stevenzack
4 years ago
committed by
Kevin Wan
parent
4b636cd293
commit
926d746df5
@ -0,0 +1,36 @@
|
|||||||
|
package ktgen
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"github.com/tal-tech/go-zero/core/lang"
|
||||||
|
"github.com/tal-tech/go-zero/tools/goctl/api/parser"
|
||||||
|
"github.com/urfave/cli"
|
||||||
|
)
|
||||||
|
|
||||||
|
func KtCommand(c *cli.Context) error {
|
||||||
|
apiFile := c.String("api")
|
||||||
|
if apiFile == "" {
|
||||||
|
return errors.New("missing -api")
|
||||||
|
}
|
||||||
|
dir := c.String("dir")
|
||||||
|
if dir == "" {
|
||||||
|
return errors.New("missing -dir")
|
||||||
|
}
|
||||||
|
pkg := c.String("pkg")
|
||||||
|
if pkg == "" {
|
||||||
|
return errors.New("missing -pkg")
|
||||||
|
}
|
||||||
|
|
||||||
|
p, e := parser.NewParser(apiFile)
|
||||||
|
if e != nil {
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
api,e:=p.Parse()
|
||||||
|
if e!=nil {
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
|
||||||
|
lang.Must(genBase(dir,pkg,api))
|
||||||
|
lang.Must(genApi(dir,pkg, api))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@ -0,0 +1,67 @@
|
|||||||
|
package ktgen
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/tal-tech/go-zero/tools/goctl/api/util"
|
||||||
|
"log"
|
||||||
|
"strings"
|
||||||
|
"text/template"
|
||||||
|
)
|
||||||
|
var funcsMap=template.FuncMap{
|
||||||
|
"lowCamelCase":lowCamelCase,
|
||||||
|
"pathToFuncName":pathToFuncName,
|
||||||
|
"parseType":parseType,
|
||||||
|
"add":add,
|
||||||
|
}
|
||||||
|
func lowCamelCase(s string) string {
|
||||||
|
if len(s) < 1 {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
s = util.ToCamelCase(util.ToSnakeCase(s))
|
||||||
|
return util.ToLower(s[:1]) + s[1:]
|
||||||
|
}
|
||||||
|
|
||||||
|
func pathToFuncName(path string) string {
|
||||||
|
if !strings.HasPrefix(path, "/") {
|
||||||
|
path = "/" + path
|
||||||
|
}
|
||||||
|
|
||||||
|
path = strings.Replace(path, "/", "_", -1)
|
||||||
|
path = strings.Replace(path, "-", "_", -1)
|
||||||
|
|
||||||
|
camel := util.ToCamelCase(path)
|
||||||
|
return util.ToLower(camel[:1]) + camel[1:]
|
||||||
|
}
|
||||||
|
func parseType(t string) string {
|
||||||
|
t=strings.Replace(t,"*","",-1)
|
||||||
|
if strings.HasPrefix(t,"[]"){
|
||||||
|
return "List<"+parseType(t[2:])+ ">"
|
||||||
|
}
|
||||||
|
|
||||||
|
if strings.HasPrefix(t,"map"){
|
||||||
|
tys,e:=util.DecomposeType(t)
|
||||||
|
if e!=nil{
|
||||||
|
log.Fatal(e)
|
||||||
|
}
|
||||||
|
if len(tys)!=2{
|
||||||
|
log.Fatal("Map type number !=2")
|
||||||
|
}
|
||||||
|
return "Map<String,"+parseType(tys[1])+">"
|
||||||
|
}
|
||||||
|
|
||||||
|
switch t {
|
||||||
|
case "string":
|
||||||
|
return "String"
|
||||||
|
case "int", "int32", "int64":
|
||||||
|
return "Int"
|
||||||
|
case "float", "float32", "float64":
|
||||||
|
return "Double"
|
||||||
|
case "bool":
|
||||||
|
return "Boolean"
|
||||||
|
default:
|
||||||
|
return t
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func add(a,i int)int{
|
||||||
|
return a+i
|
||||||
|
}
|
||||||
@ -0,0 +1,172 @@
|
|||||||
|
package ktgen
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/tal-tech/go-zero/core/logx"
|
||||||
|
"github.com/tal-tech/go-zero/tools/goctl/api/spec"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"text/template"
|
||||||
|
"github.com/iancoleman/strcase"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
apiBaseTemplate = `package {{.}}
|
||||||
|
|
||||||
|
import com.google.gson.Gson
|
||||||
|
import kotlinx.coroutines.Dispatchers
|
||||||
|
import kotlinx.coroutines.withContext
|
||||||
|
import java.io.BufferedReader
|
||||||
|
import java.io.InputStreamReader
|
||||||
|
import java.io.OutputStreamWriter
|
||||||
|
import java.net.HttpURLConnection
|
||||||
|
import java.net.URL
|
||||||
|
|
||||||
|
const val SERVER = "http://localhost:8080"
|
||||||
|
|
||||||
|
suspend fun apiPost(
|
||||||
|
uri: String,
|
||||||
|
body: Any,
|
||||||
|
onOk: ((String) -> Unit)? = null,
|
||||||
|
onFail: ((String) -> Unit)? = null,
|
||||||
|
eventually: (() -> Unit)? = null
|
||||||
|
) = withContext(Dispatchers.IO) {
|
||||||
|
val url = URL(SERVER + uri)
|
||||||
|
with(url.openConnection() as HttpURLConnection) {
|
||||||
|
requestMethod = "POST"
|
||||||
|
headerFields["Content-Type"] = listOf("Application/json")
|
||||||
|
|
||||||
|
val data = when (body) {
|
||||||
|
is String -> {
|
||||||
|
body
|
||||||
|
}
|
||||||
|
else -> {
|
||||||
|
Gson().toJson(body)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
val wr = OutputStreamWriter(outputStream)
|
||||||
|
wr.write(data)
|
||||||
|
wr.flush()
|
||||||
|
|
||||||
|
//response
|
||||||
|
BufferedReader(InputStreamReader(inputStream)).use {
|
||||||
|
val response = it.readText()
|
||||||
|
if (responseCode == 200) {
|
||||||
|
onOk?.invoke(response)
|
||||||
|
} else {
|
||||||
|
onFail?.invoke(response)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
eventually?.invoke()
|
||||||
|
}
|
||||||
|
|
||||||
|
suspend fun apiGet(
|
||||||
|
uri: String,
|
||||||
|
onOk: ((String) -> Unit)? = null,
|
||||||
|
onFail: ((String) -> Unit)? = null,
|
||||||
|
eventually: (() -> Unit)? = null
|
||||||
|
) = withContext(Dispatchers.IO) {
|
||||||
|
val url = URL(SERVER + uri)
|
||||||
|
with(url.openConnection() as HttpURLConnection) {
|
||||||
|
requestMethod = "POST"
|
||||||
|
headerFields["Content-Type"] = listOf("Application/json")
|
||||||
|
|
||||||
|
val wr = OutputStreamWriter(outputStream)
|
||||||
|
wr.flush()
|
||||||
|
|
||||||
|
//response
|
||||||
|
BufferedReader(InputStreamReader(inputStream)).use {
|
||||||
|
val response = it.readText()
|
||||||
|
if (responseCode == 200) {
|
||||||
|
onOk?.invoke(response)
|
||||||
|
} else {
|
||||||
|
onFail?.invoke(response)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
eventually?.invoke()
|
||||||
|
}
|
||||||
|
`
|
||||||
|
apiTemplate = `package {{with .Info}}{{.Title}}{{end}}
|
||||||
|
|
||||||
|
import com.google.gson.Gson
|
||||||
|
|
||||||
|
object Api{
|
||||||
|
{{range .Types}}
|
||||||
|
data class {{.Name}}({{$length := (len .Members)}}{{range $i,$item := .Members}}
|
||||||
|
val {{with $item}}{{lowCamelCase .Name}}: {{parseType .Type}}{{end}}{{if ne $i (add $length -1)}},{{end}}{{end}}
|
||||||
|
){{end}}
|
||||||
|
{{with .Service}}
|
||||||
|
{{range .Routes}}suspend fun {{pathToFuncName .Path}}({{if ne .Method "get"}}
|
||||||
|
req:{{with .RequestType}}{{.Name}},{{end}}{{end}}
|
||||||
|
onOk: (({{with .ResponseType}}{{.Name}}{{end}}) -> Unit)? = null,
|
||||||
|
onFail: ((String) -> Unit)? = null,
|
||||||
|
eventually: (() -> Unit)? = null
|
||||||
|
){
|
||||||
|
api{{if eq .Method "get"}}Get{{else}}Post{{end}}("{{.Path}}",{{if ne .Method "get"}}req,{{end}} onOk = {
|
||||||
|
onOk?.invoke(Gson().fromJson(it,{{with .ResponseType}}{{.Name}}{{end}}::class.java))
|
||||||
|
}, onFail = onFail, eventually =eventually)
|
||||||
|
}
|
||||||
|
{{end}}{{end}}
|
||||||
|
}`
|
||||||
|
)
|
||||||
|
|
||||||
|
func genBase(dir, pkg string, api *spec.ApiSpec) error {
|
||||||
|
e := os.MkdirAll(dir, 0755)
|
||||||
|
if e != nil {
|
||||||
|
logx.Error(e)
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
path := filepath.Join(dir, "BaseApi.kt")
|
||||||
|
if _, e := os.Stat(path); e == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
file, e := os.OpenFile(path, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644)
|
||||||
|
if e != nil {
|
||||||
|
logx.Error(e)
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
t, e := template.New("n").Parse(apiBaseTemplate)
|
||||||
|
if e != nil {
|
||||||
|
logx.Error(e)
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
e = t.Execute(file, pkg)
|
||||||
|
if e != nil {
|
||||||
|
logx.Error(e)
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func genApi(dir, pkg string, api *spec.ApiSpec) error {
|
||||||
|
path := filepath.Join(dir, strcase.ToCamel(api.Info.Title+"Api")+".kt")
|
||||||
|
api.Info.Title= pkg
|
||||||
|
|
||||||
|
e:=os.MkdirAll(dir,0755)
|
||||||
|
if e!=nil {
|
||||||
|
logx.Error(e)
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
|
||||||
|
file,e:=os.OpenFile(path,os.O_WRONLY|os.O_TRUNC|os.O_CREATE,0644)
|
||||||
|
if e!=nil {
|
||||||
|
logx.Error(e)
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
t,e:=template.New("api").Funcs(funcsMap).Parse(apiTemplate)
|
||||||
|
if e!=nil{
|
||||||
|
log.Fatal(e)
|
||||||
|
}
|
||||||
|
e=t.Execute(file,api)
|
||||||
|
if e!=nil{
|
||||||
|
log.Fatal(e)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue