feat: code cleanup
This commit is contained in:
20
consumer.go
20
consumer.go
@@ -15,10 +15,6 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
// InitConsumer 初始化消费者
|
// InitConsumer 初始化消费者
|
||||||
//
|
|
||||||
// Author : go_developer@163.com<白茶清欢>
|
|
||||||
//
|
|
||||||
// Date : 8:45 下午 2021/9/22
|
|
||||||
func InitConsumer(topic string, consumerGroup string, hostList []string, cfg *sarama.Config) (*Consumer, error) {
|
func InitConsumer(topic string, consumerGroup string, hostList []string, cfg *sarama.Config) (*Consumer, error) {
|
||||||
if nil == cfg {
|
if nil == cfg {
|
||||||
cfg = sarama.NewConfig()
|
cfg = sarama.NewConfig()
|
||||||
@@ -36,10 +32,6 @@ func InitConsumer(topic string, consumerGroup string, hostList []string, cfg *sa
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Consumer ...
|
// Consumer ...
|
||||||
//
|
|
||||||
// Author : go_developer@163.com<白茶清欢>
|
|
||||||
//
|
|
||||||
// Date : 8:52 下午 2021/9/22
|
|
||||||
type Consumer struct {
|
type Consumer struct {
|
||||||
topic string
|
topic string
|
||||||
host []string
|
host []string
|
||||||
@@ -50,10 +42,6 @@ type Consumer struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// init ...
|
// init ...
|
||||||
//
|
|
||||||
// Author : go_developer@163.com<白茶清欢>
|
|
||||||
//
|
|
||||||
// Date : 8:55 下午 2021/9/22
|
|
||||||
func (c *Consumer) init() error {
|
func (c *Consumer) init() error {
|
||||||
var (
|
var (
|
||||||
err error
|
err error
|
||||||
@@ -68,10 +56,6 @@ func (c *Consumer) init() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Consume 消费消息
|
// Consume 消费消息
|
||||||
//
|
|
||||||
// Author : go_developer@163.com<白茶清欢>
|
|
||||||
//
|
|
||||||
// Date : 9:04 下午 2021/9/22
|
|
||||||
func (c *Consumer) Consume(dataHandler IConsumeDataHandler) {
|
func (c *Consumer) Consume(dataHandler IConsumeDataHandler) {
|
||||||
ctx := context.Background()
|
ctx := context.Background()
|
||||||
for {
|
for {
|
||||||
@@ -106,10 +90,6 @@ func (h consumerGroupHandler) ConsumeClaim(sess sarama.ConsumerGroupSession, cla
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Destroy ...
|
// Destroy ...
|
||||||
//
|
|
||||||
// Author : go_developer@163.com<白茶清欢>
|
|
||||||
//
|
|
||||||
// Date : 8:59 下午 2021/9/22
|
|
||||||
func (c *Consumer) Destroy() {
|
func (c *Consumer) Destroy() {
|
||||||
_ = c.consumerGroupInstance.Close()
|
_ = c.consumerGroupInstance.Close()
|
||||||
}
|
}
|
||||||
|
|||||||
28
producer.go
28
producer.go
@@ -16,10 +16,6 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
// InitProducer 初始化生产者
|
// InitProducer 初始化生产者
|
||||||
//
|
|
||||||
// Author : go_developer@163.com<白茶清欢>
|
|
||||||
//
|
|
||||||
// Date : 3:52 下午 2021/9/21
|
|
||||||
func InitProducer(hostList []string, topic string, cfg *sarama.Config) (*Producer, error) {
|
func InitProducer(hostList []string, topic string, cfg *sarama.Config) (*Producer, error) {
|
||||||
if cfg == nil {
|
if cfg == nil {
|
||||||
cfg = sarama.NewConfig()
|
cfg = sarama.NewConfig()
|
||||||
@@ -43,10 +39,6 @@ func InitProducer(hostList []string, topic string, cfg *sarama.Config) (*Produce
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Producer 生产者
|
// Producer 生产者
|
||||||
//
|
|
||||||
// Author : go_developer@163.com<白茶清欢>
|
|
||||||
//
|
|
||||||
// Date : 3:57 下午 2021/9/21
|
|
||||||
type Producer struct {
|
type Producer struct {
|
||||||
cfg *sarama.Config
|
cfg *sarama.Config
|
||||||
hostList []string
|
hostList []string
|
||||||
@@ -57,10 +49,6 @@ type Producer struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// init kafka初始化
|
// init kafka初始化
|
||||||
//
|
|
||||||
// Author : go_developer@163.com<白茶清欢>
|
|
||||||
//
|
|
||||||
// Date : 3:58 下午 2021/9/21
|
|
||||||
func (p *Producer) init() error {
|
func (p *Producer) init() error {
|
||||||
var (
|
var (
|
||||||
err error
|
err error
|
||||||
@@ -85,20 +73,12 @@ func (p *Producer) init() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Destroy 断开生产者连接时,触发的逻辑
|
// Destroy 断开生产者连接时,触发的逻辑
|
||||||
//
|
|
||||||
// Author : go_developer@163.com<白茶清欢>
|
|
||||||
//
|
|
||||||
// Date : 3:58 下午 2021/9/21
|
|
||||||
func (p *Producer) Destroy() {
|
func (p *Producer) Destroy() {
|
||||||
// 断开客户端链接
|
// 断开客户端链接
|
||||||
_ = p.client.Close()
|
_ = p.client.Close()
|
||||||
}
|
}
|
||||||
|
|
||||||
// Sync 同步生产数据
|
// Sync 同步生产数据
|
||||||
//
|
|
||||||
// Author : go_developer@163.com<白茶清欢>
|
|
||||||
//
|
|
||||||
// Date : 4:08 下午 2021/9/21
|
|
||||||
func (p *Producer) Sync(data ProducerData) ProducerResult {
|
func (p *Producer) Sync(data ProducerData) ProducerResult {
|
||||||
mes := p.buildMessage(data)
|
mes := p.buildMessage(data)
|
||||||
result := ProducerResult{}
|
result := ProducerResult{}
|
||||||
@@ -107,10 +87,6 @@ func (p *Producer) Sync(data ProducerData) ProducerResult {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Async 异步生产数据
|
// Async 异步生产数据
|
||||||
//
|
|
||||||
// Author : go_developer@163.com<白茶清欢>
|
|
||||||
//
|
|
||||||
// Date : 4:09 下午 2021/9/21
|
|
||||||
func (p *Producer) Async(data ProducerData) ProducerResult {
|
func (p *Producer) Async(data ProducerData) ProducerResult {
|
||||||
result := ProducerResult{}
|
result := ProducerResult{}
|
||||||
p.asyncProducer.Input() <- p.buildMessage(data)
|
p.asyncProducer.Input() <- p.buildMessage(data)
|
||||||
@@ -129,10 +105,6 @@ func (p *Producer) Async(data ProducerData) ProducerResult {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// buildMessage 构建消息
|
// buildMessage 构建消息
|
||||||
//
|
|
||||||
// Author : go_developer@163.com<白茶清欢>
|
|
||||||
//
|
|
||||||
// Date : 7:32 下午 2021/9/21
|
|
||||||
func (p *Producer) buildMessage(data ProducerData) *sarama.ProducerMessage {
|
func (p *Producer) buildMessage(data ProducerData) *sarama.ProducerMessage {
|
||||||
mes := &sarama.ProducerMessage{
|
mes := &sarama.ProducerMessage{
|
||||||
Topic: p.topic,
|
Topic: p.topic,
|
||||||
|
|||||||
Reference in New Issue
Block a user