...

Source file src/google.golang.org/grpc/xds/internal/xdsclient/clientimpl_loadreport.go

Documentation: google.golang.org/grpc/xds/internal/xdsclient

     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  

View as plain text