goland实现函数式链式编程

• 先来看一段代码

package main

import (
    "fmt"

    elastic "gopkg.in/olivere/elastic.v2"
)

type Tweet struct {
    User    string
    Message string
}

func main() {
    client, err := elastic.NewClient(elastic.SetSniff(false), elastic.SetURL("http://127.0.0.1:9200/"))
    if err != nil {
        fmt.Println("connect es error", err)
        return
    }

    fmt.Println("conn es succ")

    tweet := Tweet{User: "olivere", Message: "Take Five"}
    //给Elasticsearch插入数据用到了函数链式编程
    _, err = client.Index().
        Index("twitter").
        Type("tweet").
        Id("1").
        BodyJson(tweet).
        Do()
    if err != nil {
        // Handle error
        panic(err)
        return
    }

    fmt.Println("insert succ")
}

• 可以看到，上面操作Elasticsearch中用到函数链式编程，在其他语言中也是支持的，那么它是如何实现的呢？

_, err = client.Index().
        Index("twitter").
        Type("tweet").
        Id("1").
        BodyJson(tweet).
        Do()

• 其实实现起来还是蛮简单的，原理就是：方法调用返回值是方法调用对象本身。
• 下面是实现代码：

package main

import "fmt"

type Stu struct {
    Name string
    Age  int
}

func (p *Stu) SetName(name string) *Stu {
    p.Name = name
    return p
}

func (p *Stu) SetAge(age int) *Stu {
    p.Age = age
    return p
}

func (p *Stu) Print() {
    fmt.Printf("age:%d name:%s\n", p.Age, p.Name)
}

func main() {
    stu := &Stu{}
    stu.SetAge(12).SetName("stu01").Print()
}

学会了吗？是不是很容易呢？