main.go 3.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798
  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. // protoc-gen-go is a plugin for the Google protocol buffer compiler to generate
  32. // Go code. Run it by building this program and putting it in your path with
  33. // the name
  34. // protoc-gen-go
  35. // That word 'go' at the end becomes part of the option string set for the
  36. // protocol compiler, so once the protocol compiler (protoc) is installed
  37. // you can run
  38. // protoc --go_out=output_directory input_directory/file.proto
  39. // to generate Go bindings for the protocol defined by file.proto.
  40. // With that input, the output will be written to
  41. // output_directory/file.pb.go
  42. //
  43. // The generated code is documented in the package comment for
  44. // the library.
  45. //
  46. // See the README and documentation for protocol buffers to learn more:
  47. // https://developers.google.com/protocol-buffers/
  48. package main
  49. import (
  50. "io/ioutil"
  51. "os"
  52. "github.com/golang/protobuf/proto"
  53. "github.com/golang/protobuf/protoc-gen-go/generator"
  54. )
  55. func main() {
  56. // Begin by allocating a generator. The request and response structures are stored there
  57. // so we can do error handling easily - the response structure contains the field to
  58. // report failure.
  59. g := generator.New()
  60. data, err := ioutil.ReadAll(os.Stdin)
  61. if err != nil {
  62. g.Error(err, "reading input")
  63. }
  64. if err := proto.Unmarshal(data, g.Request); err != nil {
  65. g.Error(err, "parsing input proto")
  66. }
  67. if len(g.Request.FileToGenerate) == 0 {
  68. g.Fail("no files to generate")
  69. }
  70. g.CommandLineParameters(g.Request.GetParameter())
  71. // Create a wrapped version of the Descriptors and EnumDescriptors that
  72. // point to the file that defines them.
  73. g.WrapTypes()
  74. g.SetPackageNames()
  75. g.BuildTypeNameMap()
  76. g.GenerateAllFiles()
  77. // Send back the results.
  78. data, err = proto.Marshal(g.Response)
  79. if err != nil {
  80. g.Error(err, "failed to marshal output proto")
  81. }
  82. _, err = os.Stdout.Write(data)
  83. if err != nil {
  84. g.Error(err, "failed to write output proto")
  85. }
  86. }