小旭
5 years ago
parent
094ee714bd
commit
21b7b19008
@ -0,0 +1,78 @@ |
|||||||
|
package main |
||||||
|
|
||||||
|
import ( |
||||||
|
"bytes" |
||||||
|
"fmt" |
||||||
|
"io/ioutil" |
||||||
|
"os" |
||||||
|
"path/filepath" |
||||||
|
"sync" |
||||||
|
) |
||||||
|
|
||||||
|
var envCache struct { |
||||||
|
once sync.Once |
||||||
|
m map[string]string |
||||||
|
} |
||||||
|
|
||||||
|
// EnvFile returns the name of the Go environment configuration file.
|
||||||
|
func EnvFile() (string, error) { |
||||||
|
if file := os.Getenv("GOENV"); file != "" { |
||||||
|
if file == "off" { |
||||||
|
return "", fmt.Errorf("GOENV=off") |
||||||
|
} |
||||||
|
return file, nil |
||||||
|
} |
||||||
|
dir, err := os.UserConfigDir() |
||||||
|
if err != nil { |
||||||
|
return "", err |
||||||
|
} |
||||||
|
if dir == "" { |
||||||
|
return "", fmt.Errorf("missing user-config dir") |
||||||
|
} |
||||||
|
return filepath.Join(dir, "go/env"), nil |
||||||
|
} |
||||||
|
|
||||||
|
func initEnvCache() { |
||||||
|
envCache.m = make(map[string]string) |
||||||
|
file, _ := EnvFile() |
||||||
|
if file == "" { |
||||||
|
return |
||||||
|
} |
||||||
|
data, err := ioutil.ReadFile(file) |
||||||
|
if err != nil { |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
for len(data) > 0 { |
||||||
|
// Get next line.
|
||||||
|
line := data |
||||||
|
i := bytes.IndexByte(data, '\n') |
||||||
|
if i >= 0 { |
||||||
|
line, data = line[:i], data[i+1:] |
||||||
|
} else { |
||||||
|
data = nil |
||||||
|
} |
||||||
|
|
||||||
|
i = bytes.IndexByte(line, '=') |
||||||
|
if i < 0 || line[0] < 'A' || 'Z' < line[0] { |
||||||
|
// Line is missing = (or empty) or a comment or not a valid env name. Ignore.
|
||||||
|
// (This should not happen, since the file should be maintained almost
|
||||||
|
// exclusively by "go env -w", but better to silently ignore than to make
|
||||||
|
// the go command unusable just because somehow the env file has
|
||||||
|
// gotten corrupted.)
|
||||||
|
continue |
||||||
|
} |
||||||
|
key, val := line[:i], line[i+1:] |
||||||
|
envCache.m[string(key)] = string(val) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Getenv gets the value from env or configuration.
|
||||||
|
func Getenv(key string) string { |
||||||
|
val := os.Getenv(key) |
||||||
|
if val != "" { |
||||||
|
return val |
||||||
|
} |
||||||
|
envCache.once.Do(initEnvCache) |
||||||
|
return envCache.m[key] |
||||||
|
} |
||||||
@ -1,592 +0,0 @@ |
|||||||
package main |
|
||||||
|
|
||||||
const ( |
|
||||||
_tplAppToml = ` |
|
||||||
# This is a TOML document. Boom~ |
|
||||||
` |
|
||||||
|
|
||||||
_tplMySQLToml = ` |
|
||||||
[demo] |
|
||||||
addr = "127.0.0.1:3306" |
|
||||||
dsn = "{user}:{password}@tcp(127.0.0.1:3306)/{database}?timeout=1s&readTimeout=1s&writeTimeout=1s&parseTime=true&loc=Local&charset=utf8mb4,utf8" |
|
||||||
readDSN = ["{user}:{password}@tcp(127.0.0.2:3306)/{database}?timeout=1s&readTimeout=1s&writeTimeout=1s&parseTime=true&loc=Local&charset=utf8mb4,utf8","{user}:{password}@tcp(127.0.0.3:3306)/{database}?timeout=1s&readTimeout=1s&writeTimeout=1s&parseTime=true&loc=Local&charset=utf8,utf8mb4"] |
|
||||||
active = 20 |
|
||||||
idle = 10 |
|
||||||
idleTimeout ="4h" |
|
||||||
queryTimeout = "200ms" |
|
||||||
execTimeout = "300ms" |
|
||||||
tranTimeout = "400ms" |
|
||||||
` |
|
||||||
_tplMCToml = ` |
|
||||||
demoExpire = "24h" |
|
||||||
|
|
||||||
[demo] |
|
||||||
name = "{{.Name}}" |
|
||||||
proto = "tcp" |
|
||||||
addr = "127.0.0.1:11211" |
|
||||||
active = 50 |
|
||||||
idle = 10 |
|
||||||
dialTimeout = "100ms" |
|
||||||
readTimeout = "200ms" |
|
||||||
writeTimeout = "300ms" |
|
||||||
idleTimeout = "80s" |
|
||||||
` |
|
||||||
_tplRedisToml = ` |
|
||||||
demoExpire = "24h" |
|
||||||
|
|
||||||
[demo] |
|
||||||
name = "{{.Name}}" |
|
||||||
proto = "tcp" |
|
||||||
addr = "127.0.0.1:6389" |
|
||||||
idle = 10 |
|
||||||
active = 10 |
|
||||||
dialTimeout = "1s" |
|
||||||
readTimeout = "1s" |
|
||||||
writeTimeout = "1s" |
|
||||||
idleTimeout = "10s" |
|
||||||
` |
|
||||||
|
|
||||||
_tplHTTPToml = ` |
|
||||||
[server] |
|
||||||
addr = "0.0.0.0:8000" |
|
||||||
timeout = "1s" |
|
||||||
` |
|
||||||
_tplGRPCToml = ` |
|
||||||
[server] |
|
||||||
addr = "0.0.0.0:9000" |
|
||||||
timeout = "1s" |
|
||||||
` |
|
||||||
|
|
||||||
_tplChangeLog = `## {{.Name}} |
|
||||||
|
|
||||||
### v1.0.0 |
|
||||||
1. 上线功能xxx |
|
||||||
` |
|
||||||
_tplMain = `package main |
|
||||||
|
|
||||||
import ( |
|
||||||
"context" |
|
||||||
"flag" |
|
||||||
"os" |
|
||||||
"os/signal" |
|
||||||
"syscall" |
|
||||||
"time" |
|
||||||
|
|
||||||
"{{.ModuleName}}/internal/server/http" |
|
||||||
"{{.ModuleName}}/internal/service" |
|
||||||
"github.com/bilibili/kratos/pkg/conf/paladin" |
|
||||||
"github.com/bilibili/kratos/pkg/log" |
|
||||||
|
|
||||||
_ "go.uber.org/automaxprocs" |
|
||||||
) |
|
||||||
|
|
||||||
func main() { |
|
||||||
flag.Parse() |
|
||||||
if err := paladin.Init(); err != nil { |
|
||||||
panic(err) |
|
||||||
} |
|
||||||
log.Init(nil) // debug flag: log.dir={path}
|
|
||||||
defer log.Close() |
|
||||||
log.Info("{{.Name}} start") |
|
||||||
svc := service.New() |
|
||||||
httpSrv := http.New(svc) |
|
||||||
c := make(chan os.Signal, 1) |
|
||||||
signal.Notify(c, syscall.SIGHUP, syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT) |
|
||||||
for { |
|
||||||
s := <-c |
|
||||||
log.Info("get a signal %s", s.String()) |
|
||||||
switch s { |
|
||||||
case syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT: |
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), 35*time.Second) |
|
||||||
if err := httpSrv.Shutdown(ctx); err != nil { |
|
||||||
log.Error("httpSrv.Shutdown error(%v)", err) |
|
||||||
} |
|
||||||
log.Info("{{.Name}} exit") |
|
||||||
svc.Close() |
|
||||||
cancel() |
|
||||||
time.Sleep(time.Second) |
|
||||||
return |
|
||||||
case syscall.SIGHUP: |
|
||||||
default: |
|
||||||
return |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
` |
|
||||||
|
|
||||||
_tplGRPCMain = `package main |
|
||||||
|
|
||||||
import ( |
|
||||||
"context" |
|
||||||
"flag" |
|
||||||
"os" |
|
||||||
"os/signal" |
|
||||||
"syscall" |
|
||||||
"time" |
|
||||||
|
|
||||||
"{{.ModuleName}}/internal/server/grpc" |
|
||||||
"{{.ModuleName}}/internal/server/http" |
|
||||||
"{{.ModuleName}}/internal/service" |
|
||||||
"github.com/bilibili/kratos/pkg/conf/paladin" |
|
||||||
"github.com/bilibili/kratos/pkg/log" |
|
||||||
|
|
||||||
_ "go.uber.org/automaxprocs" |
|
||||||
) |
|
||||||
|
|
||||||
func main() { |
|
||||||
flag.Parse() |
|
||||||
if err := paladin.Init(); err != nil { |
|
||||||
panic(err) |
|
||||||
} |
|
||||||
log.Init(nil) // debug flag: log.dir={path}
|
|
||||||
defer log.Close() |
|
||||||
log.Info("{{.Name}} start") |
|
||||||
svc := service.New() |
|
||||||
grpcSrv := grpc.New(svc) |
|
||||||
httpSrv := http.New(svc) |
|
||||||
c := make(chan os.Signal, 1) |
|
||||||
signal.Notify(c, syscall.SIGHUP, syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT) |
|
||||||
for s := range c { |
|
||||||
log.Info("get a signal %s", s.String()) |
|
||||||
switch s { |
|
||||||
case syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT: |
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), 35*time.Second) |
|
||||||
if err := grpcSrv.Shutdown(ctx); err != nil { |
|
||||||
log.Error("grpcSrv.Shutdown error(%v)", err) |
|
||||||
} |
|
||||||
if err := httpSrv.Shutdown(ctx); err != nil { |
|
||||||
log.Error("httpSrv.Shutdown error(%v)", err) |
|
||||||
} |
|
||||||
log.Info("{{.Name}} exit") |
|
||||||
svc.Close() |
|
||||||
cancel() |
|
||||||
time.Sleep(time.Second) |
|
||||||
return |
|
||||||
case syscall.SIGHUP: |
|
||||||
default: |
|
||||||
return |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
` |
|
||||||
|
|
||||||
_tplContributors = `# Owner |
|
||||||
{{.Owner}} |
|
||||||
|
|
||||||
# Author |
|
||||||
|
|
||||||
# Reviewer |
|
||||||
` |
|
||||||
|
|
||||||
_tplDao = `package dao |
|
||||||
|
|
||||||
import ( |
|
||||||
"context" |
|
||||||
"time" |
|
||||||
|
|
||||||
"github.com/bilibili/kratos/pkg/cache/memcache" |
|
||||||
"github.com/bilibili/kratos/pkg/cache/redis" |
|
||||||
"github.com/bilibili/kratos/pkg/conf/paladin" |
|
||||||
"github.com/bilibili/kratos/pkg/database/sql" |
|
||||||
"github.com/bilibili/kratos/pkg/log" |
|
||||||
xtime "github.com/bilibili/kratos/pkg/time" |
|
||||||
) |
|
||||||
|
|
||||||
// Dao dao interface
|
|
||||||
type Dao interface { |
|
||||||
Close() |
|
||||||
Ping(ctx context.Context) (err error) |
|
||||||
} |
|
||||||
|
|
||||||
// dao dao.
|
|
||||||
type dao struct { |
|
||||||
db *sql.DB |
|
||||||
redis *redis.Pool |
|
||||||
redisExpire int32 |
|
||||||
mc *memcache.Memcache |
|
||||||
mcExpire int32 |
|
||||||
} |
|
||||||
|
|
||||||
func checkErr(err error) { |
|
||||||
if err != nil { |
|
||||||
panic(err) |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
// New new a dao and return.
|
|
||||||
func New() (Dao) { |
|
||||||
var ( |
|
||||||
dc struct { |
|
||||||
Demo *sql.Config |
|
||||||
} |
|
||||||
rc struct { |
|
||||||
Demo *redis.Config |
|
||||||
DemoExpire xtime.Duration |
|
||||||
} |
|
||||||
mc struct { |
|
||||||
Demo *memcache.Config |
|
||||||
DemoExpire xtime.Duration |
|
||||||
} |
|
||||||
) |
|
||||||
checkErr(paladin.Get("mysql.toml").UnmarshalTOML(&dc)) |
|
||||||
checkErr(paladin.Get("redis.toml").UnmarshalTOML(&rc)) |
|
||||||
checkErr(paladin.Get("memcache.toml").UnmarshalTOML(&mc)) |
|
||||||
return &dao{ |
|
||||||
// mysql
|
|
||||||
db: sql.NewMySQL(dc.Demo), |
|
||||||
// redis
|
|
||||||
redis: redis.NewPool(rc.Demo), |
|
||||||
redisExpire: int32(time.Duration(rc.DemoExpire) / time.Second), |
|
||||||
// memcache
|
|
||||||
mc: memcache.New(mc.Demo), |
|
||||||
mcExpire: int32(time.Duration(mc.DemoExpire) / time.Second), |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
// Close close the resource.
|
|
||||||
func (d *dao) Close() { |
|
||||||
d.mc.Close() |
|
||||||
d.redis.Close() |
|
||||||
d.db.Close() |
|
||||||
} |
|
||||||
|
|
||||||
// Ping ping the resource.
|
|
||||||
func (d *dao) Ping(ctx context.Context) (err error) { |
|
||||||
if err = d.pingMC(ctx); err != nil { |
|
||||||
return |
|
||||||
} |
|
||||||
if err = d.pingRedis(ctx); err != nil { |
|
||||||
return |
|
||||||
} |
|
||||||
return d.db.Ping(ctx) |
|
||||||
} |
|
||||||
|
|
||||||
func (d *dao) pingMC(ctx context.Context) (err error) { |
|
||||||
if err = d.mc.Set(ctx, &memcache.Item{Key: "ping", Value: []byte("pong"), Expiration: 0}); err != nil { |
|
||||||
log.Error("conn.Set(PING) error(%v)", err) |
|
||||||
} |
|
||||||
return |
|
||||||
} |
|
||||||
|
|
||||||
func (d *dao) pingRedis(ctx context.Context) (err error) { |
|
||||||
conn := d.redis.Get(ctx) |
|
||||||
defer conn.Close() |
|
||||||
if _, err = conn.Do("SET", "ping", "pong"); err != nil { |
|
||||||
log.Error("conn.Set(PING) error(%v)", err) |
|
||||||
} |
|
||||||
return |
|
||||||
} |
|
||||||
` |
|
||||||
_tplReadme = `# {{.Name}} |
|
||||||
|
|
||||||
## 项目简介 |
|
||||||
1. |
|
||||||
` |
|
||||||
_tplService = `package service |
|
||||||
|
|
||||||
import ( |
|
||||||
"context" |
|
||||||
|
|
||||||
"{{.ModuleName}}/internal/dao" |
|
||||||
"github.com/bilibili/kratos/pkg/conf/paladin" |
|
||||||
) |
|
||||||
|
|
||||||
// Service service.
|
|
||||||
type Service struct { |
|
||||||
ac *paladin.Map |
|
||||||
dao dao.Dao |
|
||||||
} |
|
||||||
|
|
||||||
// New new a service and return.
|
|
||||||
func New() (s *Service) { |
|
||||||
var ac = new(paladin.TOML) |
|
||||||
if err := paladin.Watch("application.toml", ac); err != nil { |
|
||||||
panic(err) |
|
||||||
} |
|
||||||
s = &Service{ |
|
||||||
ac: ac, |
|
||||||
dao: dao.New(), |
|
||||||
} |
|
||||||
return s |
|
||||||
} |
|
||||||
|
|
||||||
// Ping ping the resource.
|
|
||||||
func (s *Service) Ping(ctx context.Context) (err error) { |
|
||||||
return s.dao.Ping(ctx) |
|
||||||
} |
|
||||||
|
|
||||||
// Close close the resource.
|
|
||||||
func (s *Service) Close() { |
|
||||||
s.dao.Close() |
|
||||||
} |
|
||||||
` |
|
||||||
|
|
||||||
_tplGPRCService = `package service |
|
||||||
|
|
||||||
import ( |
|
||||||
"context" |
|
||||||
"fmt" |
|
||||||
|
|
||||||
pb "{{.ModuleName}}/api" |
|
||||||
"{{.ModuleName}}/internal/dao" |
|
||||||
"github.com/bilibili/kratos/pkg/conf/paladin" |
|
||||||
|
|
||||||
"github.com/golang/protobuf/ptypes/empty" |
|
||||||
) |
|
||||||
|
|
||||||
// Service service.
|
|
||||||
type Service struct { |
|
||||||
ac *paladin.Map |
|
||||||
dao dao.Dao |
|
||||||
} |
|
||||||
|
|
||||||
// New new a service and return.
|
|
||||||
func New() (s *Service) { |
|
||||||
var ac = new(paladin.TOML) |
|
||||||
if err := paladin.Watch("application.toml", ac); err != nil { |
|
||||||
panic(err) |
|
||||||
} |
|
||||||
s = &Service{ |
|
||||||
ac: ac, |
|
||||||
dao: dao.New(), |
|
||||||
} |
|
||||||
return s |
|
||||||
} |
|
||||||
|
|
||||||
// SayHello grpc demo func.
|
|
||||||
func (s *Service) SayHello(ctx context.Context, req *pb.HelloReq) (reply *empty.Empty, err error) { |
|
||||||
reply = new(empty.Empty) |
|
||||||
fmt.Printf("hello %s", req.Name) |
|
||||||
return |
|
||||||
} |
|
||||||
|
|
||||||
// SayHelloURL bm demo func.
|
|
||||||
func (s *Service) SayHelloURL(ctx context.Context, req *pb.HelloReq) (reply *pb.HelloResp, err error) { |
|
||||||
reply = &pb.HelloResp{ |
|
||||||
Content: "hello " + req.Name, |
|
||||||
} |
|
||||||
fmt.Printf("hello url %s", req.Name) |
|
||||||
return |
|
||||||
} |
|
||||||
|
|
||||||
// Ping ping the resource.
|
|
||||||
func (s *Service) Ping(ctx context.Context) (err error) { |
|
||||||
return s.dao.Ping(ctx) |
|
||||||
} |
|
||||||
|
|
||||||
// Close close the resource.
|
|
||||||
func (s *Service) Close() { |
|
||||||
s.dao.Close() |
|
||||||
} |
|
||||||
` |
|
||||||
_tplHTTPServer = `package http |
|
||||||
|
|
||||||
import ( |
|
||||||
"net/http" |
|
||||||
|
|
||||||
"{{.ModuleName}}/internal/model" |
|
||||||
"{{.ModuleName}}/internal/service" |
|
||||||
|
|
||||||
"github.com/bilibili/kratos/pkg/conf/paladin" |
|
||||||
"github.com/bilibili/kratos/pkg/log" |
|
||||||
bm "github.com/bilibili/kratos/pkg/net/http/blademaster" |
|
||||||
) |
|
||||||
|
|
||||||
var ( |
|
||||||
svc *service.Service |
|
||||||
) |
|
||||||
|
|
||||||
// New new a bm server.
|
|
||||||
func New(s *service.Service) (engine *bm.Engine) { |
|
||||||
var ( |
|
||||||
hc struct { |
|
||||||
Server *bm.ServerConfig |
|
||||||
} |
|
||||||
) |
|
||||||
if err := paladin.Get("http.toml").UnmarshalTOML(&hc); err != nil { |
|
||||||
if err != paladin.ErrNotExist { |
|
||||||
panic(err) |
|
||||||
} |
|
||||||
} |
|
||||||
svc = s |
|
||||||
engine = bm.DefaultServer(hc.Server) |
|
||||||
initRouter(engine) |
|
||||||
if err := engine.Start(); err != nil { |
|
||||||
panic(err) |
|
||||||
} |
|
||||||
return |
|
||||||
} |
|
||||||
|
|
||||||
func initRouter(e *bm.Engine) { |
|
||||||
e.Ping(ping) |
|
||||||
g := e.Group("/{{.Name}}") |
|
||||||
{ |
|
||||||
g.GET("/start", howToStart) |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
func ping(ctx *bm.Context) { |
|
||||||
if err := svc.Ping(ctx); err != nil { |
|
||||||
log.Error("ping error(%v)", err) |
|
||||||
ctx.AbortWithStatus(http.StatusServiceUnavailable) |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
// example for http request handler.
|
|
||||||
func howToStart(c *bm.Context) { |
|
||||||
k := &model.Kratos{ |
|
||||||
Hello: "Golang 大法好 !!!", |
|
||||||
} |
|
||||||
c.JSON(k, nil) |
|
||||||
} |
|
||||||
` |
|
||||||
_tplPBHTTPServer = `package http |
|
||||||
|
|
||||||
import ( |
|
||||||
"net/http" |
|
||||||
|
|
||||||
pb "{{.ModuleName}}/api" |
|
||||||
"{{.ModuleName}}/internal/model" |
|
||||||
"{{.ModuleName}}/internal/service" |
|
||||||
|
|
||||||
"github.com/bilibili/kratos/pkg/conf/paladin" |
|
||||||
"github.com/bilibili/kratos/pkg/log" |
|
||||||
bm "github.com/bilibili/kratos/pkg/net/http/blademaster" |
|
||||||
) |
|
||||||
|
|
||||||
var ( |
|
||||||
svc *service.Service |
|
||||||
) |
|
||||||
|
|
||||||
// New new a bm server.
|
|
||||||
func New(s *service.Service) (engine *bm.Engine) { |
|
||||||
var ( |
|
||||||
hc struct { |
|
||||||
Server *bm.ServerConfig |
|
||||||
} |
|
||||||
) |
|
||||||
if err := paladin.Get("http.toml").UnmarshalTOML(&hc); err != nil { |
|
||||||
if err != paladin.ErrNotExist { |
|
||||||
panic(err) |
|
||||||
} |
|
||||||
} |
|
||||||
svc = s |
|
||||||
engine = bm.DefaultServer(hc.Server) |
|
||||||
pb.RegisterDemoBMServer(engine, svc) |
|
||||||
initRouter(engine) |
|
||||||
if err := engine.Start(); err != nil { |
|
||||||
panic(err) |
|
||||||
} |
|
||||||
return |
|
||||||
} |
|
||||||
|
|
||||||
func initRouter(e *bm.Engine) { |
|
||||||
e.Ping(ping) |
|
||||||
g := e.Group("/{{.Name}}") |
|
||||||
{ |
|
||||||
g.GET("/start", howToStart) |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
func ping(ctx *bm.Context) { |
|
||||||
if err := svc.Ping(ctx); err != nil { |
|
||||||
log.Error("ping error(%v)", err) |
|
||||||
ctx.AbortWithStatus(http.StatusServiceUnavailable) |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
// example for http request handler.
|
|
||||||
func howToStart(c *bm.Context) { |
|
||||||
k := &model.Kratos{ |
|
||||||
Hello: "Golang 大法好 !!!", |
|
||||||
} |
|
||||||
c.JSON(k, nil) |
|
||||||
} |
|
||||||
|
|
||||||
` |
|
||||||
|
|
||||||
_tplAPIProto = `// 定义项目 API 的 proto 文件 可以同时描述 gRPC 和 HTTP API
|
|
||||||
// protobuf 文件参考:
|
|
||||||
// - https://developers.google.com/protocol-buffers/
|
|
||||||
syntax = "proto3"; |
|
||||||
|
|
||||||
import "github.com/gogo/protobuf/gogoproto/gogo.proto"; |
|
||||||
import "google/protobuf/empty.proto"; |
|
||||||
import "google/api/annotations.proto"; |
|
||||||
|
|
||||||
// package 命名使用 {appid}.{version} 的方式, version 形如 v1, v2 ..
|
|
||||||
package demo.service.v1; |
|
||||||
|
|
||||||
// NOTE: 最后请删除这些无用的注释 (゜-゜)つロ
|
|
||||||
|
|
||||||
option go_package = "api"; |
|
||||||
option (gogoproto.goproto_getters_all) = false; |
|
||||||
|
|
||||||
service Demo { |
|
||||||
rpc SayHello (HelloReq) returns (.google.protobuf.Empty); |
|
||||||
rpc SayHelloURL(HelloReq) returns (HelloResp) { |
|
||||||
option (google.api.http) = { |
|
||||||
get:"/{{.Name}}/say_hello" |
|
||||||
}; |
|
||||||
}; |
|
||||||
} |
|
||||||
|
|
||||||
message HelloReq { |
|
||||||
string name = 1 [(gogoproto.moretags)='form:"name" validate:"required"']; |
|
||||||
} |
|
||||||
|
|
||||||
message HelloResp { |
|
||||||
string Content = 1 [(gogoproto.jsontag) = 'content']; |
|
||||||
} |
|
||||||
` |
|
||||||
_tplModel = `package model |
|
||||||
|
|
||||||
// Kratos hello kratos.
|
|
||||||
type Kratos struct { |
|
||||||
Hello string |
|
||||||
}` |
|
||||||
_tplGoMod = `module {{.ModuleName}} |
|
||||||
|
|
||||||
go 1.12 |
|
||||||
|
|
||||||
require ( |
|
||||||
github.com/bilibili/kratos master |
|
||||||
github.com/gogo/protobuf v1.2.1 |
|
||||||
github.com/golang/protobuf v1.3.2 |
|
||||||
golang.org/x/net v0.0.0-20190628185345-da137c7871d7 |
|
||||||
google.golang.org/grpc v1.22.0 |
|
||||||
) |
|
||||||
` |
|
||||||
_tplGRPCServer = `package grpc |
|
||||||
|
|
||||||
import ( |
|
||||||
pb "{{.ModuleName}}/api" |
|
||||||
"{{.ModuleName}}/internal/service" |
|
||||||
"github.com/bilibili/kratos/pkg/conf/paladin" |
|
||||||
"github.com/bilibili/kratos/pkg/net/rpc/warden" |
|
||||||
) |
|
||||||
|
|
||||||
// New new a grpc server.
|
|
||||||
func New(svc *service.Service) *warden.Server { |
|
||||||
var rc struct { |
|
||||||
Server *warden.ServerConfig |
|
||||||
} |
|
||||||
if err := paladin.Get("grpc.toml").UnmarshalTOML(&rc); err != nil { |
|
||||||
if err != paladin.ErrNotExist { |
|
||||||
panic(err) |
|
||||||
} |
|
||||||
} |
|
||||||
ws := warden.NewServer(rc.Server) |
|
||||||
pb.RegisterDemoServer(ws.Server(), svc) |
|
||||||
ws, err := ws.Start() |
|
||||||
if err != nil { |
|
||||||
panic(err) |
|
||||||
} |
|
||||||
return ws |
|
||||||
} |
|
||||||
` |
|
||||||
_tplGogen = `package api |
|
||||||
|
|
||||||
//go:generate kratos tool protoc --swagger --grpc --bm api.proto
|
|
||||||
` |
|
||||||
) |
|
||||||
Loading…
Reference in new issue