Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Forever310
druid
提交
cc40a339
D
druid
项目概览
Forever310
/
druid
与 Fork 源项目一致
从无法访问的项目Fork
通知
3
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
D
druid
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
cc40a339
编写于
5月 22, 2014
作者:
F
fjy
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #563 from tucksaun/fix-rabbitmq-connection-retries
Make RabbitMQ Firehose resilient to broker deconnections
上级
8c9a27ab
b7426cd0
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
45 addition
and
2 deletion
+45
-2
rabbitmq/src/main/java/io/druid/firehose/rabbitmq/RabbitMQFirehoseFactory.java
...a/io/druid/firehose/rabbitmq/RabbitMQFirehoseFactory.java
+45
-2
未找到文件。
rabbitmq/src/main/java/io/druid/firehose/rabbitmq/RabbitMQFirehoseFactory.java
浏览文件 @
cc40a339
...
@@ -25,9 +25,13 @@ import com.metamx.common.logger.Logger;
...
@@ -25,9 +25,13 @@ import com.metamx.common.logger.Logger;
import
com.rabbitmq.client.Channel
;
import
com.rabbitmq.client.Channel
;
import
com.rabbitmq.client.Connection
;
import
com.rabbitmq.client.Connection
;
import
com.rabbitmq.client.ConnectionFactory
;
import
com.rabbitmq.client.ConnectionFactory
;
import
com.rabbitmq.client.QueueingConsumer
;
import
com.rabbitmq.client.DefaultConsumer
;
import
com.rabbitmq.client.Envelope
;
import
com.rabbitmq.client.AMQP
;
import
com.rabbitmq.client.QueueingConsumer.Delivery
;
import
com.rabbitmq.client.ShutdownListener
;
import
com.rabbitmq.client.ShutdownListener
;
import
com.rabbitmq.client.ShutdownSignalException
;
import
com.rabbitmq.client.ShutdownSignalException
;
import
com.rabbitmq.client.ConsumerCancelledException
;
import
io.druid.data.input.ByteBufferInputRowParser
;
import
io.druid.data.input.ByteBufferInputRowParser
;
import
io.druid.data.input.Firehose
;
import
io.druid.data.input.Firehose
;
import
io.druid.data.input.FirehoseFactory
;
import
io.druid.data.input.FirehoseFactory
;
...
@@ -40,6 +44,8 @@ import net.jodah.lyra.retry.RetryPolicy;
...
@@ -40,6 +44,8 @@ import net.jodah.lyra.retry.RetryPolicy;
import
net.jodah.lyra.util.Duration
;
import
net.jodah.lyra.util.Duration
;
import
java.io.IOException
;
import
java.io.IOException
;
import
java.util.concurrent.BlockingQueue
;
import
java.util.concurrent.LinkedBlockingQueue
;
/**
/**
* A FirehoseFactory for RabbitMQ.
* A FirehoseFactory for RabbitMQ.
...
@@ -179,7 +185,7 @@ public class RabbitMQFirehoseFactory implements FirehoseFactory<StringInputRowPa
...
@@ -179,7 +185,7 @@ public class RabbitMQFirehoseFactory implements FirehoseFactory<StringInputRowPa
* Storing the latest delivery as a member variable should be safe since this will only be run
* Storing the latest delivery as a member variable should be safe since this will only be run
* by a single thread.
* by a single thread.
*/
*/
private
QueueingConsumer
.
Delivery
delivery
;
private
Delivery
delivery
;
/**
/**
* Store the latest delivery tag to be able to commit (acknowledge) the message delivery up to
* Store the latest delivery tag to be able to commit (acknowledge) the message delivery up to
...
@@ -268,4 +274,41 @@ public class RabbitMQFirehoseFactory implements FirehoseFactory<StringInputRowPa
...
@@ -268,4 +274,41 @@ public class RabbitMQFirehoseFactory implements FirehoseFactory<StringInputRowPa
{
{
return
parser
;
return
parser
;
}
}
private
static
class
QueueingConsumer
extends
DefaultConsumer
{
private
final
BlockingQueue
<
Delivery
>
_queue
;
public
QueueingConsumer
(
Channel
ch
)
{
this
(
ch
,
new
LinkedBlockingQueue
<
Delivery
>());
}
public
QueueingConsumer
(
Channel
ch
,
BlockingQueue
<
Delivery
>
q
)
{
super
(
ch
);
this
.
_queue
=
q
;
}
@Override
public
void
handleShutdownSignal
(
String
consumerTag
,
ShutdownSignalException
sig
)
{
_queue
.
clear
();
}
@Override
public
void
handleCancel
(
String
consumerTag
)
throws
IOException
{
_queue
.
clear
();
}
@Override
public
void
handleDelivery
(
String
consumerTag
,
Envelope
envelope
,
AMQP
.
BasicProperties
properties
,
byte
[]
body
)
throws
IOException
{
this
.
_queue
.
add
(
new
Delivery
(
envelope
,
properties
,
body
));
}
public
Delivery
nextDelivery
()
throws
InterruptedException
,
ShutdownSignalException
,
ConsumerCancelledException
{
return
_queue
.
take
();
}
}
}
}
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录