Perl:通过 mojolicious 的异步 http 代理
Perl: An asynchronous http proxy via mojolicious
我做了一个简单的 http 代理,它工作正常,但速度不快,因为在函数 handle_request 中,我使用
my $tx = $ua->start( Mojo::Transaction::HTTP->new(req=>$request) );
做请求,阻塞。
我尝试使用这样的回调:
$ua->start( Mojo::Transaction::HTTP->new(req=>$request) )=>sub{ ... }
让它成为非阻塞的,然后,犯了一个错误:
'error' => { 'message' => 'Premature connection close'}
我想那是因为函数 handle_request
return 立即执行,它不会等待回调完成。如果我使用信号量等待回调,那就意味着它再次阻塞。
#!/usr/bin/perl
use strict;
use warnings;
use utf8;
use Mojo::IOLoop::Server;
use Mojo::UserAgent;
use Mojo::Message::Response;
use Mojo::Message::Request;
use Mojo::Transaction::HTTP;
use Data::Dumper;
binmode STDOUT, ":encoding(UTF-8)";
my %buffer;
Mojo::IOLoop->server( {port => 3128} => sub {
my ($loop, $stream, $client) = @_;
$stream->on(
read => sub {
my ($stream, $chunk) = @_;
my $buffer = $buffer{$client}{read_buffer} .= $chunk;
if ($buffer =~ /^GET\s+|POST\s+|HEAD\s+(.*)\r\n\r\n$/i) {
$buffer{$client}{read_buffer} = '';
&handle_request($client,$stream,$buffer);
}
elsif ($buffer =~ /^CONNECT\s+(.*)\r\n\r\n$/i) {
$buffer{$client}{read_buffer} = '';
&handle_connect($stream,$buffer);
}
elsif($buffer{$client}{connection})
{
$buffer{$client}{read_buffer} = '';
Mojo::IOLoop->stream($buffer{$client}{connection})->write($chunk);
}
if(length($buffer)>= 20 *1024 * 1024) {
delete $buffer{$client};
Mojo::IOLoop->remove($client);
return;
}
});
});
sub handle_request{
my($client,$stream,$chunk) = @_;
my $request = Mojo::Message::Request->new;
$request = $request->parse($chunk);
my $ua = Mojo::UserAgent->new;
my $tx = $ua->start( Mojo::Transaction::HTTP->new(req=>$request) );
$stream->write( $tx->res->to_string );
}
sub handle_connect{
my ($stream, $chunk) = @_;
my $request = Mojo::Message::Request->new;
my $ua = Mojo::UserAgent->new;
$request = $request->parse($chunk);
print Dumper($request);
}
Mojo::IOLoop->start;
希望得到一些建议。
你有两个问题:
- 当您的代码具有阻塞样式时,您尝试调用
$ua->start
的非阻塞变体。函数 handle_request
必须将回调作为参数。
如果你有回调链那么实现它的最好方法是使用 Mojo::IOLoop::Delay
.
- 当你在 sub
handle_request
中以非阻塞方式创建变量 $ua 时,你的变量将被垃圾收集器销毁,因为首先执行 sub handle_request
的退出并且 $ua 被销毁,因为它是局部变量,然后从 $ua 得到答案。所以你得到 Premature connection close
。您需要将 $ua 的实例保存在别处以防止此类错误。
更新.
我 write http/https 代理的错误变体,它只能通过 CONNECT 方法工作,并且有错误,第一条 http 消息不完整。
更新.
我添加了另一个 http/https 代理示例,它正确读取了第一条 http 消息并且不仅通过 CONNECT 方法工作。
更新.
哦,Mojo的作者写了example of https proxy
我做了一个简单的 http 代理,它工作正常,但速度不快,因为在函数 handle_request 中,我使用
my $tx = $ua->start( Mojo::Transaction::HTTP->new(req=>$request) );
做请求,阻塞。
我尝试使用这样的回调:
$ua->start( Mojo::Transaction::HTTP->new(req=>$request) )=>sub{ ... }
让它成为非阻塞的,然后,犯了一个错误:
'error' => { 'message' => 'Premature connection close'}
我想那是因为函数 handle_request
return 立即执行,它不会等待回调完成。如果我使用信号量等待回调,那就意味着它再次阻塞。
#!/usr/bin/perl
use strict;
use warnings;
use utf8;
use Mojo::IOLoop::Server;
use Mojo::UserAgent;
use Mojo::Message::Response;
use Mojo::Message::Request;
use Mojo::Transaction::HTTP;
use Data::Dumper;
binmode STDOUT, ":encoding(UTF-8)";
my %buffer;
Mojo::IOLoop->server( {port => 3128} => sub {
my ($loop, $stream, $client) = @_;
$stream->on(
read => sub {
my ($stream, $chunk) = @_;
my $buffer = $buffer{$client}{read_buffer} .= $chunk;
if ($buffer =~ /^GET\s+|POST\s+|HEAD\s+(.*)\r\n\r\n$/i) {
$buffer{$client}{read_buffer} = '';
&handle_request($client,$stream,$buffer);
}
elsif ($buffer =~ /^CONNECT\s+(.*)\r\n\r\n$/i) {
$buffer{$client}{read_buffer} = '';
&handle_connect($stream,$buffer);
}
elsif($buffer{$client}{connection})
{
$buffer{$client}{read_buffer} = '';
Mojo::IOLoop->stream($buffer{$client}{connection})->write($chunk);
}
if(length($buffer)>= 20 *1024 * 1024) {
delete $buffer{$client};
Mojo::IOLoop->remove($client);
return;
}
});
});
sub handle_request{
my($client,$stream,$chunk) = @_;
my $request = Mojo::Message::Request->new;
$request = $request->parse($chunk);
my $ua = Mojo::UserAgent->new;
my $tx = $ua->start( Mojo::Transaction::HTTP->new(req=>$request) );
$stream->write( $tx->res->to_string );
}
sub handle_connect{
my ($stream, $chunk) = @_;
my $request = Mojo::Message::Request->new;
my $ua = Mojo::UserAgent->new;
$request = $request->parse($chunk);
print Dumper($request);
}
Mojo::IOLoop->start;
希望得到一些建议。
你有两个问题:
- 当您的代码具有阻塞样式时,您尝试调用
$ua->start
的非阻塞变体。函数handle_request
必须将回调作为参数。 如果你有回调链那么实现它的最好方法是使用Mojo::IOLoop::Delay
. - 当你在 sub
handle_request
中以非阻塞方式创建变量 $ua 时,你的变量将被垃圾收集器销毁,因为首先执行 subhandle_request
的退出并且 $ua 被销毁,因为它是局部变量,然后从 $ua 得到答案。所以你得到Premature connection close
。您需要将 $ua 的实例保存在别处以防止此类错误。
更新.
我 write http/https 代理的错误变体,它只能通过 CONNECT 方法工作,并且有错误,第一条 http 消息不完整。
更新.
我添加了另一个 http/https 代理示例,它正确读取了第一条 http 消息并且不仅通过 CONNECT 方法工作。
更新.
哦,Mojo的作者写了example of https proxy