Skip to content

Commit ee1a3dc

Browse files
committed
fixed implementation generation for imported RequestTypes
1 parent 85d135b commit ee1a3dc

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

cmd/protoc-gen-goclay/genhandler/handler.go

+6-6
Original file line numberDiff line numberDiff line change
@@ -353,22 +353,22 @@ func (g *Generator) getServiceImpl(f *descriptor.File, s *descriptor.Service) (s
353353

354354
func (g *Generator) getMethodImpl(s *descriptor.Service, m *descriptor.Method) (string, error) {
355355
// restore orig GoPkg
356-
savedPkg := m.RequestType.File.GoPkg
356+
savedPkg := s.File.GoPkg
357357
defer func() {
358-
m.RequestType.File.GoPkg = savedPkg
358+
s.File.GoPkg = savedPkg
359359
}()
360360

361-
return applyImplTemplate(g.getImplParam(m.RequestType.File, s, m, []string{"context", "github.com/pkg/errors"}))
361+
return applyImplTemplate(g.getImplParam(s.File, s, m, []string{"context", "github.com/pkg/errors"}))
362362
}
363363

364364
func (g *Generator) getTestImpl(s *descriptor.Service, m *descriptor.Method) (string, error) {
365365
// restore orig GoPkg
366-
savedPkg := m.RequestType.File.GoPkg
366+
savedPkg := s.File.GoPkg
367367
defer func() {
368-
m.RequestType.File.GoPkg = savedPkg
368+
s.File.GoPkg = savedPkg
369369
}()
370370

371-
return applyTestTemplate(g.getImplParam(m.RequestType.File, s, m, []string{"context", "testing", "github.com/stretchr/testify/require"}))
371+
return applyTestTemplate(g.getImplParam(s.File, s, m, []string{"context", "testing", "github.com/stretchr/testify/require"}))
372372
}
373373

374374
func (g *Generator) getImplParam(f *descriptor.File, s *descriptor.Service, m *descriptor.Method, deps []string) implParam {

0 commit comments

Comments
 (0)