6.Thrift指南 thrift go源码解析 1


前言: 一切从thrift go demo服务端开始...

⚠️thrift go源码版本 0.11.0
1.从服务端demo开始
2.源码入口

1.从服务端demo开始

1.1 服务端demo代码的工作非常简单,主要做了两步:

  • 一、初始化一个SimpleServer类实例
  • 二、运行实例方法Serve()
    这里与我们平常做法一致,通过一个结构来组织程序,类似于控制层c
func main() {
    // ....这里节选自 【3.Thrift指南 实战go服务端】
    server := thrift.NewTSimpleServer2(processor, transport)
    server.Serve()
}

2.源码入口

2.1 首先阅读TSimpleServer结构,文件头、方法注释 (thrift/simple_server.go)
通过结构定义以及相关字段对整个TSimpleServer有一个初步的认识。
通过注释对文件、方法用途有一个了解。

# 这是一个典型的TSimpleServer服务,因为它在接收套接字后不会阻塞。
# 它更像一个TThreadedServer服务,可以在不同的goroutine中处理连接。
# 如果在golang客户端实现连接池之类的东西就可以工作了。
/*
 * This is not a typical TSimpleServer as it is not blocked after accept a socket.
 * It is more like a TThreadedServer that can handle different connections in different goroutines.
 * This will work if golang user implements a conn-pool like thing in client side.
 */
type TSimpleServer struct {
    closed int32 # 服务状态标识
    wg     sync.WaitGroup  # 等待goroutine完成必备
    mu     sync.Mutex  # 锁

    processorFactory       TProcessorFactory
    serverTransport        TServerTransport  # serve具体实现
    inputTransportFactory  TTransportFactory
    outputTransportFactory TTransportFactory
    inputProtocolFactory   TProtocolFactory
    outputProtocolFactory  TProtocolFactory
}

closed、wg、mu定义简单具体,一目了然
*Factory采用接口形式组织,是常见的工厂模式设计

2.2 .Serve()这个方法主要做了两件事:

  • 一、p.serverTransport.Listen() 开始Listen
  • 二、p.AcceptLoop() 开始accept阻塞,等待客户端连接
func (p *TSimpleServer) Serve() error {
    err := p.Listen()
    if err != nil {
        return err
    }
    p.AcceptLoop()
    return nil
}
func (p *TSimpleServer) Listen() error {
    return p.serverTransport.Listen()
}
func (p *TSimpleServer) AcceptLoop() error {
    for {
        // 这里调用server_socket.go下TServerSocket类的方法
        // 实质为开始Accept阻塞,等待客户端连接
        client, err := p.serverTransport.Accept()
        if err != nil {
            return err
        }

        // ?? 黑人问号.jpg
        p.mu.Lock()
        // 严谨的if,获取当前服务端状态标识,确保服务关闭后不会处理新任务
        if atomic.LoadInt32(&p.closed) != 0 {
            return nil
        }
        // 严谨的if
        if client != nil {
            // p.wg.* 确保停止服务时,所有业务goroutine都可以正常执行完毕
            p.wg.Add(1)

            // 正如文件头描述的那般,
            // 不同的goroutine处理不同的client
            go func() {
                defer p.wg.Done()
            // 具体的业务处理,下一章节
                if err := p.processRequests(client); err != nil {
                    log.Println("error processing request:", err)
                }
            }()
        }
        p.mu.Unlock()
    }
}
func (p *TSimpleServer) Stop() error {
    // 这个加锁,atomic.LoadInt32和atomic.StoreInt32操作并发不安全
    p.mu.Lock()
    defer p.mu.Unlock()
    // 严谨的if,读取服务标识
    if atomic.LoadInt32(&p.closed) != 0 {
        return nil
    }
    // 关闭服务(由此可见通过 0,1来标识服务,那用i8岂不美哉, hahah~)
    atomic.StoreInt32(&p.closed, 1)
        
    // 连接中断,可选实现
    p.serverTransport.Interrupt()
    p.wg.Wait()
    return nil
}

上面注释了每一行代码的作用,下面整理了其中的关键点:

  • 采用wg等待业务处理完毕退出主程序。
  • 采用goroutine来实现业务处理,提高性能。
  • 采用锁确保atomic.LoadInt32和atomic.StoreInt32原子操作并发安全。
  • AcceptLoop()的锁是否有必要?if判断的必要性?
  • Stop()中的p.serverTransport.Interrupt() 的中断作用何在?(server_transport.go注释定义)

总结

TSimpleServer结构、方法的定义非常简单清晰明了,严谨的if判断,不同goroute处理不同业务,实现高性能的同时又缩减了代码。主要得益于go语法,以及工厂设计模式对复杂的业务逻辑的简化,TSimpleServer服务只需要组织起结构,这其中的减法有助于我们程序扩展性更强,也是我们需要学习的地方。
但是关键点中的疑问,我们下下章再见👋...

附录:
go 等待所以goroutine处理完毕,才退出主程序
go sync/atomic包Load和Store并发不安全

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

推荐阅读更多精彩内容