1// Go support for Protocol Buffers - Google's data interchange format 2// 3// Copyright 2010 The Go Authors. All rights reserved. 4// https://github.com/golang/protobuf 5// 6// Redistribution and use in source and binary forms, with or without 7// modification, are permitted provided that the following conditions are 8// met: 9// 10// * Redistributions of source code must retain the above copyright 11// notice, this list of conditions and the following disclaimer. 12// * Redistributions in binary form must reproduce the above 13// copyright notice, this list of conditions and the following disclaimer 14// in the documentation and/or other materials provided with the 15// distribution. 16// * Neither the name of Google Inc. nor the names of its 17// contributors may be used to endorse or promote products derived from 18// this software without specific prior written permission. 19// 20// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 21// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 22// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 23// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 24// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 25// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 26// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 27// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 28// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 29// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 30// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 31 32// protoc-gen-go is a plugin for the Google protocol buffer compiler to generate 33// Go code. Run it by building this program and putting it in your path with 34// the name 35// protoc-gen-go 36// That word 'go' at the end becomes part of the option string set for the 37// protocol compiler, so once the protocol compiler (protoc) is installed 38// you can run 39// protoc --go_out=output_directory input_directory/file.proto 40// to generate Go bindings for the protocol defined by file.proto. 41// With that input, the output will be written to 42// output_directory/file.pb.go 43// 44// The generated code is documented in the package comment for 45// the library. 46// 47// See the README and documentation for protocol buffers to learn more: 48// https://developers.google.com/protocol-buffers/ 49package main 50 51import ( 52 "io/ioutil" 53 "os" 54 55 "github.com/golang/protobuf/proto" 56 "github.com/golang/protobuf/protoc-gen-go/generator" 57) 58 59func main() { 60 // Begin by allocating a generator. The request and response structures are stored there 61 // so we can do error handling easily - the response structure contains the field to 62 // report failure. 63 g := generator.New() 64 65 data, err := ioutil.ReadAll(os.Stdin) 66 if err != nil { 67 g.Error(err, "reading input") 68 } 69 70 if err := proto.Unmarshal(data, g.Request); err != nil { 71 g.Error(err, "parsing input proto") 72 } 73 74 if len(g.Request.FileToGenerate) == 0 { 75 g.Fail("no files to generate") 76 } 77 78 g.CommandLineParameters(g.Request.GetParameter()) 79 80 // Create a wrapped version of the Descriptors and EnumDescriptors that 81 // point to the file that defines them. 82 g.WrapTypes() 83 84 g.SetPackageNames() 85 g.BuildTypeNameMap() 86 87 g.GenerateAllFiles() 88 89 // Send back the results. 90 data, err = proto.Marshal(g.Response) 91 if err != nil { 92 g.Error(err, "failed to marshal output proto") 93 } 94 _, err = os.Stdout.Write(data) 95 if err != nil { 96 g.Error(err, "failed to write output proto") 97 } 98} 99