You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
1.0 KiB
39 lines
1.0 KiB
package codegenerator |
|
|
|
import ( |
|
"fmt" |
|
"io" |
|
"io/ioutil" |
|
|
|
"github.com/golang/protobuf/proto" |
|
plugin "github.com/golang/protobuf/protoc-gen-go/plugin" |
|
) |
|
|
|
// ParseRequest parses a code generator request from a proto Message. |
|
func ParseRequest(r io.Reader) (*plugin.CodeGeneratorRequest, error) { |
|
input, err := ioutil.ReadAll(r) |
|
if err != nil { |
|
return nil, fmt.Errorf("failed to read code generator request: %v", err) |
|
} |
|
req := new(plugin.CodeGeneratorRequest) |
|
if err = proto.Unmarshal(input, req); err != nil { |
|
return nil, fmt.Errorf("failed to unmarshal code generator request: %v", err) |
|
} |
|
return req, nil |
|
} |
|
|
|
// WriteResponse write a code generator response |
|
func WriteResponse(w io.Writer, files []*plugin.CodeGeneratorResponse_File, inErr error) error { |
|
var perrMsg *string |
|
if inErr != nil { |
|
errMsg := inErr.Error() |
|
perrMsg = &errMsg |
|
} |
|
resp := &plugin.CodeGeneratorResponse{Error: perrMsg, File: files} |
|
buf, err := proto.Marshal(resp) |
|
if err != nil { |
|
return err |
|
} |
|
_, err = w.Write(buf) |
|
return err |
|
}
|
|
|