| OLD | NEW |
| 1 // Copyright 2016 The LUCI Authors. All rights reserved. | 1 // Copyright 2016 The LUCI Authors. All rights reserved. |
| 2 // Use of this source code is governed under the Apache License, Version 2.0 | 2 // Use of this source code is governed under the Apache License, Version 2.0 |
| 3 // that can be found in the LICENSE file. | 3 // that can be found in the LICENSE file. |
| 4 | 4 |
| 5 package main | 5 package main |
| 6 | 6 |
| 7 import ( | 7 import ( |
| 8 "bytes" | 8 "bytes" |
| 9 "compress/gzip" | 9 "compress/gzip" |
| 10 "fmt" | 10 "fmt" |
| 11 "io/ioutil" | 11 "io/ioutil" |
| 12 "os" | 12 "os" |
| 13 "os/exec" | 13 "os/exec" |
| 14 "strings" | 14 "strings" |
| 15 "text/template" | 15 "text/template" |
| 16 | 16 |
| 17 "github.com/golang/protobuf/proto" | 17 "github.com/golang/protobuf/proto" |
| 18 "github.com/luci/luci-go/common/proto/google/descriptor" | |
| 19 "golang.org/x/net/context" | 18 "golang.org/x/net/context" |
| 19 "google.golang.org/genproto/protobuf" |
| 20 ) | 20 ) |
| 21 | 21 |
| 22 const ( | 22 const ( |
| 23 discoveryPackagePath = "github.com/luci/luci-go/grpc/discovery" | 23 discoveryPackagePath = "github.com/luci/luci-go/grpc/discovery" |
| 24 ) | 24 ) |
| 25 | 25 |
| 26 // discoveryTmpl is template for generated Go discovery file. | 26 // discoveryTmpl is template for generated Go discovery file. |
| 27 // The result of execution will also be passed through gofmt. | 27 // The result of execution will also be passed through gofmt. |
| 28 var discoveryTmpl = template.Must(template.New("").Parse(strings.TrimSpace(` | 28 var discoveryTmpl = template.Must(template.New("").Parse(strings.TrimSpace(` |
| 29 // AUTOGENERATED. DO NOT EDIT. | 29 // AUTOGENERATED. DO NOT EDIT. |
| 30 | 30 |
| 31 package {{.GoPkg}}; | 31 package {{.GoPkg}}; |
| 32 | 32 |
| 33 {{if .ImportDiscovery}} | 33 {{if .ImportDiscovery}} |
| 34 import discovery "github.com/luci/luci-go/grpc/discovery" | 34 import discovery "github.com/luci/luci-go/grpc/discovery" |
| 35 {{end}} | 35 {{end}} |
| 36 import "github.com/luci/luci-go/common/proto/google/descriptor" | 36 import "google.golang.org/genproto/protobuf" |
| 37 | 37 |
| 38 func init() { | 38 func init() { |
| 39 {{if .ImportDiscovery}}discovery.{{end}}RegisterDescriptorSetCompressed( | 39 {{if .ImportDiscovery}}discovery.{{end}}RegisterDescriptorSetCompressed( |
| 40 []string{ | 40 []string{ |
| 41 {{range .ServiceNames}}"{{.}}",{{end}} | 41 {{range .ServiceNames}}"{{.}}",{{end}} |
| 42 }, | 42 }, |
| 43 {{.CompressedBytes}}, | 43 {{.CompressedBytes}}, |
| 44 ) | 44 ) |
| 45 } | 45 } |
| 46 | 46 |
| (...skipping 100 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 147 var buf bytes.Buffer | 147 var buf bytes.Buffer |
| 148 w := gzip.NewWriter(&buf) | 148 w := gzip.NewWriter(&buf) |
| 149 if _, err := w.Write(data); err != nil { | 149 if _, err := w.Write(data); err != nil { |
| 150 return nil, err | 150 return nil, err |
| 151 } | 151 } |
| 152 if err := w.Close(); err != nil { | 152 if err := w.Close(); err != nil { |
| 153 return nil, err | 153 return nil, err |
| 154 } | 154 } |
| 155 return buf.Bytes(), nil | 155 return buf.Bytes(), nil |
| 156 } | 156 } |
| OLD | NEW |