| // Copyright 2016 Google 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. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License 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. |
| |
| // AUTO-GENERATED CODE. DO NOT EDIT. |
| |
| package logging |
| |
| import ( |
| "fmt" |
| "math" |
| "runtime" |
| "time" |
| |
| gax "github.com/googleapis/gax-go" |
| "golang.org/x/net/context" |
| "google.golang.org/api/iterator" |
| "google.golang.org/api/option" |
| "google.golang.org/api/transport" |
| loggingpb "google.golang.org/genproto/googleapis/logging/v2" |
| "google.golang.org/grpc" |
| "google.golang.org/grpc/codes" |
| "google.golang.org/grpc/metadata" |
| ) |
| |
| var ( |
| configParentPathTemplate = gax.MustCompilePathTemplate("projects/{project}") |
| configSinkPathTemplate = gax.MustCompilePathTemplate("projects/{project}/sinks/{sink}") |
| ) |
| |
| // ConfigCallOptions contains the retry settings for each method of ConfigClient. |
| type ConfigCallOptions struct { |
| ListSinks []gax.CallOption |
| GetSink []gax.CallOption |
| CreateSink []gax.CallOption |
| UpdateSink []gax.CallOption |
| DeleteSink []gax.CallOption |
| } |
| |
| func defaultConfigClientOptions() []option.ClientOption { |
| return []option.ClientOption{ |
| option.WithEndpoint("logging.googleapis.com:443"), |
| option.WithScopes( |
| "https://www.googleapis.com/auth/cloud-platform", |
| "https://www.googleapis.com/auth/cloud-platform.read-only", |
| "https://www.googleapis.com/auth/logging.admin", |
| "https://www.googleapis.com/auth/logging.read", |
| "https://www.googleapis.com/auth/logging.write", |
| ), |
| } |
| } |
| |
| func defaultConfigCallOptions() *ConfigCallOptions { |
| retry := map[[2]string][]gax.CallOption{ |
| {"default", "idempotent"}: { |
| gax.WithRetry(func() gax.Retryer { |
| return gax.OnCodes([]codes.Code{ |
| codes.DeadlineExceeded, |
| codes.Unavailable, |
| }, gax.Backoff{ |
| Initial: 100 * time.Millisecond, |
| Max: 1000 * time.Millisecond, |
| Multiplier: 1.2, |
| }) |
| }), |
| }, |
| } |
| return &ConfigCallOptions{ |
| ListSinks: retry[[2]string{"default", "idempotent"}], |
| GetSink: retry[[2]string{"default", "idempotent"}], |
| CreateSink: retry[[2]string{"default", "non_idempotent"}], |
| UpdateSink: retry[[2]string{"default", "non_idempotent"}], |
| DeleteSink: retry[[2]string{"default", "idempotent"}], |
| } |
| } |
| |
| // ConfigClient is a client for interacting with Stackdriver Logging API. |
| type ConfigClient struct { |
| // The connection to the service. |
| conn *grpc.ClientConn |
| |
| // The gRPC API client. |
| configClient loggingpb.ConfigServiceV2Client |
| |
| // The call options for this service. |
| CallOptions *ConfigCallOptions |
| |
| // The metadata to be sent with each request. |
| metadata map[string][]string |
| } |
| |
| // NewConfigClient creates a new config service v2 client. |
| // |
| // Service for configuring sinks used to export log entries outside Stackdriver |
| // Logging. |
| func NewConfigClient(ctx context.Context, opts ...option.ClientOption) (*ConfigClient, error) { |
| conn, err := transport.DialGRPC(ctx, append(defaultConfigClientOptions(), opts...)...) |
| if err != nil { |
| return nil, err |
| } |
| c := &ConfigClient{ |
| conn: conn, |
| CallOptions: defaultConfigCallOptions(), |
| |
| configClient: loggingpb.NewConfigServiceV2Client(conn), |
| } |
| c.SetGoogleClientInfo("gax", gax.Version) |
| return c, nil |
| } |
| |
| // Connection returns the client's connection to the API service. |
| func (c *ConfigClient) Connection() *grpc.ClientConn { |
| return c.conn |
| } |
| |
| // Close closes the connection to the API service. The user should invoke this when |
| // the client is no longer required. |
| func (c *ConfigClient) Close() error { |
| return c.conn.Close() |
| } |
| |
| // SetGoogleClientInfo sets the name and version of the application in |
| // the `x-goog-api-client` header passed on each request. Intended for |
| // use by Google-written clients. |
| func (c *ConfigClient) SetGoogleClientInfo(name, version string) { |
| c.metadata = map[string][]string{ |
| "x-goog-api-client": {fmt.Sprintf("%s/%s %s gax/%s go/%s", name, version, gapicNameVersion, gax.Version, runtime.Version())}, |
| } |
| } |
| |
| // ConfigParentPath returns the path for the parent resource. |
| func ConfigParentPath(project string) string { |
| path, err := configParentPathTemplate.Render(map[string]string{ |
| "project": project, |
| }) |
| if err != nil { |
| panic(err) |
| } |
| return path |
| } |
| |
| // ConfigSinkPath returns the path for the sink resource. |
| func ConfigSinkPath(project, sink string) string { |
| path, err := configSinkPathTemplate.Render(map[string]string{ |
| "project": project, |
| "sink": sink, |
| }) |
| if err != nil { |
| panic(err) |
| } |
| return path |
| } |
| |
| // ListSinks lists sinks. |
| func (c *ConfigClient) ListSinks(ctx context.Context, req *loggingpb.ListSinksRequest) *LogSinkIterator { |
| md, _ := metadata.FromContext(ctx) |
| ctx = metadata.NewContext(ctx, metadata.Join(md, c.metadata)) |
| it := &LogSinkIterator{} |
| |
| fetch := func(pageSize int, pageToken string) (string, error) { |
| var resp *loggingpb.ListSinksResponse |
| req.PageToken = pageToken |
| if pageSize > math.MaxInt32 { |
| req.PageSize = math.MaxInt32 |
| } else { |
| req.PageSize = int32(pageSize) |
| } |
| err := gax.Invoke(ctx, func(ctx context.Context) error { |
| var err error |
| resp, err = c.configClient.ListSinks(ctx, req) |
| return err |
| }, c.CallOptions.ListSinks...) |
| if err != nil { |
| return "", err |
| } |
| it.items = append(it.items, resp.Sinks...) |
| return resp.NextPageToken, nil |
| } |
| bufLen := func() int { return len(it.items) } |
| takeBuf := func() interface{} { |
| b := it.items |
| it.items = nil |
| return b |
| } |
| |
| it.pageInfo, it.nextFunc = iterator.NewPageInfo(fetch, bufLen, takeBuf) |
| return it |
| } |
| |
| // GetSink gets a sink. |
| func (c *ConfigClient) GetSink(ctx context.Context, req *loggingpb.GetSinkRequest) (*loggingpb.LogSink, error) { |
| md, _ := metadata.FromContext(ctx) |
| ctx = metadata.NewContext(ctx, metadata.Join(md, c.metadata)) |
| var resp *loggingpb.LogSink |
| err := gax.Invoke(ctx, func(ctx context.Context) error { |
| var err error |
| resp, err = c.configClient.GetSink(ctx, req) |
| return err |
| }, c.CallOptions.GetSink...) |
| if err != nil { |
| return nil, err |
| } |
| return resp, nil |
| } |
| |
| // CreateSink creates a sink. |
| func (c *ConfigClient) CreateSink(ctx context.Context, req *loggingpb.CreateSinkRequest) (*loggingpb.LogSink, error) { |
| md, _ := metadata.FromContext(ctx) |
| ctx = metadata.NewContext(ctx, metadata.Join(md, c.metadata)) |
| var resp *loggingpb.LogSink |
| err := gax.Invoke(ctx, func(ctx context.Context) error { |
| var err error |
| resp, err = c.configClient.CreateSink(ctx, req) |
| return err |
| }, c.CallOptions.CreateSink...) |
| if err != nil { |
| return nil, err |
| } |
| return resp, nil |
| } |
| |
| // UpdateSink updates or creates a sink. |
| func (c *ConfigClient) UpdateSink(ctx context.Context, req *loggingpb.UpdateSinkRequest) (*loggingpb.LogSink, error) { |
| md, _ := metadata.FromContext(ctx) |
| ctx = metadata.NewContext(ctx, metadata.Join(md, c.metadata)) |
| var resp *loggingpb.LogSink |
| err := gax.Invoke(ctx, func(ctx context.Context) error { |
| var err error |
| resp, err = c.configClient.UpdateSink(ctx, req) |
| return err |
| }, c.CallOptions.UpdateSink...) |
| if err != nil { |
| return nil, err |
| } |
| return resp, nil |
| } |
| |
| // DeleteSink deletes a sink. |
| func (c *ConfigClient) DeleteSink(ctx context.Context, req *loggingpb.DeleteSinkRequest) error { |
| md, _ := metadata.FromContext(ctx) |
| ctx = metadata.NewContext(ctx, metadata.Join(md, c.metadata)) |
| err := gax.Invoke(ctx, func(ctx context.Context) error { |
| var err error |
| _, err = c.configClient.DeleteSink(ctx, req) |
| return err |
| }, c.CallOptions.DeleteSink...) |
| return err |
| } |
| |
| // LogSinkIterator manages a stream of *loggingpb.LogSink. |
| type LogSinkIterator struct { |
| items []*loggingpb.LogSink |
| pageInfo *iterator.PageInfo |
| nextFunc func() error |
| } |
| |
| // PageInfo supports pagination. See the google.golang.org/api/iterator package for details. |
| func (it *LogSinkIterator) PageInfo() *iterator.PageInfo { |
| return it.pageInfo |
| } |
| |
| // Next returns the next result. Its second return value is iterator.Done if there are no more |
| // results. Once Next returns Done, all subsequent calls will return Done. |
| func (it *LogSinkIterator) Next() (*loggingpb.LogSink, error) { |
| if err := it.nextFunc(); err != nil { |
| return nil, err |
| } |
| item := it.items[0] |
| it.items = it.items[1:] |
| return item, nil |
| } |