| 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 | 
|---|