overflow_cloud/consumer
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ing

Browse Source
This commit is contained in:
crusader 2018-05-11 19:02:16 +09:00
parent 3072c9dff3
commit a7922fda42
1 changed files with 9 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
consumer.go
View File

@ -41,10 +41,15 @@ func (c *Consumer) Start() error {
return fmt.Errorf("%s Init error %v", c.logHeader(), err) return fmt.Errorf("%s Init error %v", c.logHeader(), err)
} }
reader := oek.Reader(c.ConsumerHandler.GetConsumerName())
if nil == reader {
return fmt.Errorf("%s Reader[%s] is not valid", c.logHeader(), c.ConsumerHandler.GetConsumerName())
}
c.stopChan = make(chan struct{}) c.stopChan = make(chan struct{})
c.stopWg.Add(1) c.stopWg.Add(1)
go c.handleConsumer() go c.handleConsumer(reader)
return nil return nil
} }
@ -67,7 +72,7 @@ func (c *Consumer) logHeader() string {
return fmt.Sprintf("Consumer[%s]:", c.ConsumerHandler.GetName()) return fmt.Sprintf("Consumer[%s]:", c.ConsumerHandler.GetName())
} }
func (c *Consumer) handleConsumer() { func (c *Consumer) handleConsumer(reader *kafka.Reader) {
defer func() { defer func() {
c.ConsumerHandler.OnStop(c.ctx) c.ConsumerHandler.OnStop(c.ctx)
logging.Logger().Infof("%s Stopped", c.logHeader()) logging.Logger().Infof("%s Stopped", c.logHeader())
@ -82,7 +87,7 @@ func (c *Consumer) handleConsumer() {
stopChan := make(chan struct{}) stopChan := make(chan struct{})
receiveDoneChan := make(chan error) receiveDoneChan := make(chan error)
go c.handleReceive(stopChan, receiveDoneChan) go c.handleReceive(stopChan, receiveDoneChan, reader)
logging.Logger().Infof("%s Started", c.logHeader()) logging.Logger().Infof("%s Started", c.logHeader())
@ -95,7 +100,7 @@ func (c *Consumer) handleConsumer() {
} }
} }
func (c *Consumer) handleReceive(stopChan <-chan struct{}, doneChan chan<- error) { func (c *Consumer) handleReceive(stopChan <-chan struct{}, doneChan chan<- error, reader *kafka.Reader) {
var ( var (
m kafka.Message m kafka.Message
stopping atomic.Value stopping atomic.Value
@ -106,12 +111,6 @@ func (c *Consumer) handleReceive(stopChan <-chan struct{}, doneChan chan<- error
doneChan <- err doneChan <- err
}() }()
reader := oek.Reader(c.ConsumerHandler.GetConsumerName())
if nil == reader {
err = fmt.Errorf("%s Reader[%s] is not valid", c.logHeader(), c.ConsumerHandler.GetConsumerName())
return
}
for { for {
readChan := make(chan struct{}) readChan := make(chan struct{})
go func() { go func() {