eko54r
b5e3a778ce
All checks were successful
Build, Push and Deploy / build-and-push (push) Successful in 3m3s
Gitea Actions Demo / Explore-Gitea-Actions (push) Successful in 16s
Build, Push and Deploy / deploy-staging (push) Successful in 32s
Build, Push and Deploy / deploy-production (push) Has been skipped
Build, Push and Deploy / cleanup (push) Successful in 1s
87 lines
2.9 KiB
PHP
87 lines
2.9 KiB
PHP
<?php
|
|
/*
|
|
*
|
|
* Copyright 2018 gRPC authors.
|
|
*
|
|
* 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.
|
|
*
|
|
*/
|
|
namespace Grpc\Gcp;
|
|
|
|
/**
|
|
* GCPCallInvoker updates the channel pool(GcpExtensionChannel) for
|
|
* each RPC. The idea is:
|
|
* Before the RPC starts, pick a channel from the channel pool:
|
|
* - if the RPC is bound to a channel, use that channel.
|
|
* - if the RPC doesn't bound to a channel, use the one with minimum active streams.
|
|
* After the RPC finishes, update the active stream ref count.
|
|
* - if the RPC is defined as bind, bind the channel with corresponding key like
|
|
* spanner session name.
|
|
* - if the RPC is defined as unbind, unbind the channel with the key.
|
|
*/
|
|
class GCPCallInvoker implements \Grpc\CallInvoker
|
|
{
|
|
private $channel;
|
|
private $affinity_conf;
|
|
|
|
/**
|
|
* @param array $affinity_conf Store the affinity config for process each RPC.
|
|
*/
|
|
public function __construct($affinity_conf)
|
|
{
|
|
$this->affinity_conf = $affinity_conf;
|
|
}
|
|
|
|
/**
|
|
* @param string $hostname
|
|
* @param array $opts
|
|
* @return GcpExtensionChannel
|
|
*/
|
|
public function createChannelFactory($hostname, $opts)
|
|
{
|
|
if ($this->channel) {
|
|
// $call_invoker object has already created from previews PHP-FPM scripts.
|
|
// Only need to update the $opts including the credentials.
|
|
$this->channel->updateOpts($opts);
|
|
} else {
|
|
$opts['affinity_conf'] = $this->affinity_conf;
|
|
$channel = new GcpExtensionChannel($hostname, $opts);
|
|
$this->channel = $channel;
|
|
}
|
|
return $this->channel;
|
|
}
|
|
|
|
// _getChannel is used for testing only.
|
|
public function GetChannel()
|
|
{
|
|
return $this->channel;
|
|
}
|
|
|
|
public function UnaryCall($channel, $method, $deserialize, $options)
|
|
{
|
|
return new GCPUnaryCall($channel, $method, $deserialize, $options);
|
|
}
|
|
public function ClientStreamingCall($channel, $method, $deserialize, $options)
|
|
{
|
|
return new GCPClientStreamCall($channel, $method, $deserialize, $options);
|
|
}
|
|
public function ServerStreamingCall($channel, $method, $deserialize, $options)
|
|
{
|
|
return new GCPServerStreamCall($channel, $method, $deserialize, $options);
|
|
}
|
|
public function BidiStreamingCall($channel, $method, $deserialize, $options)
|
|
{
|
|
return new GCPBidiStreamingCall($channel, $method, $deserialize, $options);
|
|
}
|
|
}
|