nsq源码解读之nsq_to_file

nsq_to_file创建消费者,读取nsq消息并写入文件中,支持topic的模糊匹配和实时更新,以及消息落文件的定时rotate等设置。研究nsq_to_file的代码有助于理解nsq消息队列的工作流程。
创建消费者的代码为:

func newConsumerFileLogger(topic string, cfg *nsq.Config) (*ConsumerFileLogger, error) {
    f, err := NewFileLogger(*gzipEnabled, *gzipLevel, *filenameFormat, topic)
    if err != nil {
        return nil, err
    }

    consumer, err := nsq.NewConsumer(topic, *channel, cfg)
    if err != nil {
        return nil, err
    }

    consumer.AddHandler(f)

    err = consumer.ConnectToNSQDs(nsqdTCPAddrs)
    if err != nil {
        log.Fatal(err)
    }

    err = consumer.ConnectToNSQLookupds(lookupdHTTPAddrs)
    if err != nil {
        log.Fatal(err)
    }

    return &ConsumerFileLogger{
        C: consumer,
        F: f,
    }, nil
}

其中调用了go-nsq库。在go-nsq中通过consumer.ConnectToNSQLookupds,调用nsqlookupd的/lookup接口查询当前的所有生产者。然后连接每一个nsqd,并订阅相应的channel(不指定就默认为"nsq_to_file",如果nsqd
不存在该channel会自动创建)。
当与nsqd的TCP连接收到消息时,go-nsq中会回调消费者的HandleMessage函数。

func (f *FileLogger) HandleMessage(m *nsq.Message) error {
    m.DisableAutoResponse()
    f.logChan <- m
    return nil
}

func (f *FileLogger) router(r *nsq.Consumer) {
    pos := 0
    output := make([]*nsq.Message, *maxInFlight)
    sync := false
    ticker := time.NewTicker(time.Duration(30) * time.Second)
    closing := false
    closeFile := false
    exit := false

    for {
        select {
        case <-r.StopChan:
            sync = true
            closeFile = true
            exit = true
        case <-f.termChan:
            ticker.Stop()
            r.Stop()
            sync = true
            closing = true
        case <-f.hupChan:
            sync = true
            closeFile = true
        case <-ticker.C:
            if f.needsFileRotate() {
                if *skipEmptyFiles {
                    closeFile = true
                } else {
                    f.updateFile()
                }
            }
            sync = true
        case m := <-f.logChan:
            if f.needsFileRotate() {
                f.updateFile()
                sync = true
            }
            _, err := f.writer.Write(m.Body)
            if err != nil {
                log.Fatalf("ERROR: writing message to disk - %s", err)
            }
            _, err = f.writer.Write([]byte("\n"))
            if err != nil {
                log.Fatalf("ERROR: writing newline to disk - %s", err)
            }
            output[pos] = m
            pos++
            if pos == cap(output) {
                sync = true
            }
        }

        if closing || sync || r.IsStarved() {
            if pos > 0 {
                log.Printf("syncing %d records to disk", pos)
                err := f.Sync()
                if err != nil {
                    log.Fatalf("ERROR: failed syncing messages - %s", err)
                }
                for pos > 0 {
                    pos--
                    m := output[pos]
                    m.Finish()
                    output[pos] = nil
                }
            }
            sync = false
        }

        if closeFile {
            f.Close()
            closeFile = false
        }
        if exit {
            close(f.ExitChan)
            break
        }
    }
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,793评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,567评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,342评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,825评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,814评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,680评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,033评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,687评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 42,175评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,668评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,775评论 1 332
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,419评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,020评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,978评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,206评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,092评论 2 351
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,510评论 2 343

推荐阅读更多精彩内容

  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,596评论 18 139
  • 1. 介绍 最近在研究一些消息中间件,常用的MQ如RabbitMQ,ActiveMQ,Kafka等。NSQ是一个基...
    aoho阅读 8,916评论 1 16
  • 你们知道吗?有一群人,她们每天清晨6点起床后准点迎接到校的孩子。 你们知道吗?她们为了让新到校的孩子养成好习惯,数...
    米粒2020阅读 315评论 0 0
  • 不管你现在是不是千里马,或许在若干年之后,当你有了自己的公司,或是成为某公司的高管后,你的角色都会转变,你有可能会...
    夏夏夏子桑阅读 658评论 0 0
  • 被身边的朋友常问及的一个饮酒问题是“怎样鉴别变质的葡萄酒”,解释的次数多了,就抽空归纳总结了一下,写出来与读者分享...
    红酒说WineSay阅读 243评论 0 0