Skip to content

Commit 6956d79

Browse files
authored
collect messages from services if reflection enabled (#89)
1 parent 5b6a5aa commit 6956d79

File tree

2 files changed

+14
-1
lines changed

2 files changed

+14
-1
lines changed

entity/env.go

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,10 +81,23 @@ func NewEnv(pkgs []*Package, config *config.Config) *Env {
8181
// NewEnvFromServices is called if the target server has enabled gRPC reflection.
8282
// gRPC reflection has no packages, so Evans creates pseudo package "default".
8383
func NewEnvFromServices(svcs []Service, config *config.Config) *Env {
84+
mmsgs := map[string]Message{}
85+
for _, svc := range svcs {
86+
for _, rpc := range svc.RPCs() {
87+
mmsgs[rpc.RequestMessage().Name()] = rpc.RequestMessage()
88+
mmsgs[rpc.ResponseMessage().Name()] = rpc.ResponseMessage()
89+
}
90+
}
91+
var msgs []Message
92+
for _, msg := range mmsgs {
93+
msgs = append(msgs, msg)
94+
}
95+
8496
env := NewEnv([]*Package{
8597
{
8698
Name: "default",
8799
Services: svcs,
100+
Messages: msgs,
88101
},
89102
}, config)
90103

meta/meta.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,5 +7,5 @@ import (
77
const AppName = "evans"
88

99
var (
10-
Version = semver.MustParse("0.6.0")
10+
Version = semver.MustParse("0.6.1")
1111
)

0 commit comments

Comments
 (0)