1 /* 2 * 3 * Copyright 2019 gRPC authors. 4 * 5 * Licensed under the Apache License, Version 2.0 (the "License"); 6 * you may not use this file except in compliance with the License. 7 * You may obtain a copy of the License at 8 * 9 * http://www.apache.org/licenses/LICENSE-2.0 10 * 11 * Unless required by applicable law or agreed to in writing, software 12 * distributed under the License is distributed on an "AS IS" BASIS, 13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14 * See the License for the specific language governing permissions and 15 * limitations under the License. 16 */ 17 18 package xdsclient 19 20 import ( 21 "google.golang.org/grpc/internal/xds/bootstrap" 22 "google.golang.org/grpc/xds/internal/xdsclient/load" 23 ) 24 25 // ReportLoad starts a load reporting stream to the given server. All load 26 // reports to the same server share the LRS stream. 27 // 28 // It returns a Store for the user to report loads, a function to cancel the 29 // load reporting stream. 30 func (c *clientImpl) ReportLoad(server *bootstrap.ServerConfig) (*load.Store, func()) { 31 c.authorityMu.Lock() 32 a, err := c.newAuthorityLocked(server) 33 if err != nil { 34 c.authorityMu.Unlock() 35 c.logger.Infof("xds: failed to connect to the control plane to do load reporting for authority %q: %v", server, err) 36 return nil, func() {} 37 } 38 // Hold the ref before starting load reporting. 39 a.refLocked() 40 c.authorityMu.Unlock() 41 42 store, cancelF := a.reportLoad() 43 return store, func() { 44 cancelF() 45 c.unrefAuthority(a) 46 } 47 } 48