mirror of
https://github.com/hanzoai/psrpc.git
synced 2026-09-30 23:47:22 +00:00
David ColburnandGitHub
3a3a2a935f
* checkpoint * cleaned up streams * clean up multi rpcs * clean up rpc * service definition, fix stream bug * fix #54 * updates * version bump
79 lines
1.9 KiB
Go
79 lines
1.9 KiB
Go
// Copyright 2018 Twitch Interactive, Inc. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License"). You may not
|
|
// use this file except in compliance with the License. A copy of the License is
|
|
// located at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// or in the "license" file accompanying this file. This file is distributed on
|
|
// an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
|
|
// express or implied. See the License for the specific language governing
|
|
// permissions and limitations under the License.
|
|
|
|
package gen
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
|
|
"google.golang.org/protobuf/proto"
|
|
descriptor "google.golang.org/protobuf/types/descriptorpb"
|
|
plugin "google.golang.org/protobuf/types/pluginpb"
|
|
)
|
|
|
|
type Generator interface {
|
|
Generate(in *plugin.CodeGeneratorRequest) *plugin.CodeGeneratorResponse
|
|
}
|
|
|
|
func Main(g Generator) {
|
|
req := readGenRequest(os.Stdin)
|
|
resp := g.Generate(req)
|
|
writeResponse(os.Stdout, resp)
|
|
}
|
|
|
|
func FilesToGenerate(req *plugin.CodeGeneratorRequest) []*descriptor.FileDescriptorProto {
|
|
genFiles := make([]*descriptor.FileDescriptorProto, 0)
|
|
Outer:
|
|
for _, name := range req.FileToGenerate {
|
|
for _, f := range req.ProtoFile {
|
|
if f.GetName() == name {
|
|
genFiles = append(genFiles, f)
|
|
continue Outer
|
|
}
|
|
}
|
|
Fail("could not find file named", name)
|
|
}
|
|
|
|
return genFiles
|
|
}
|
|
|
|
func readGenRequest(r io.Reader) *plugin.CodeGeneratorRequest {
|
|
data, err := io.ReadAll(r)
|
|
if err != nil {
|
|
Error(err, "reading input")
|
|
}
|
|
|
|
req := new(plugin.CodeGeneratorRequest)
|
|
if err = proto.Unmarshal(data, req); err != nil {
|
|
Error(err, "parsing input proto")
|
|
}
|
|
|
|
if len(req.FileToGenerate) == 0 {
|
|
Fail("no files to generate")
|
|
}
|
|
|
|
return req
|
|
}
|
|
|
|
func writeResponse(w io.Writer, resp *plugin.CodeGeneratorResponse) {
|
|
data, err := proto.Marshal(resp)
|
|
if err != nil {
|
|
Error(err, "marshaling response")
|
|
}
|
|
_, err = w.Write(data)
|
|
if err != nil {
|
|
Error(err, "writing response")
|
|
}
|
|
}
|