google.golang.org/grpc@v1.72.2/xds/internal/xdsclient/clientimpl_loadreport.go (about)

     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  	xc, releaseChannelRef, err := c.getChannelForLRS(server)
    32  	if err != nil {
    33  		c.logger.Warningf("Failed to create a channel to the management server to report load: %v", server, err)
    34  		return nil, func() {}
    35  	}
    36  	load, stopLoadReporting := xc.reportLoad()
    37  	return load, func() {
    38  		stopLoadReporting()
    39  		releaseChannelRef()
    40  	}
    41  }