Go语言实现GoF设计模式：适配器模式

// demo/db/sql.go
package db

// Adaptee SQL语句执行返回的结果，并未实现Target接口
type SqlResult struct {
    fields []string
    vals   []interface{}
}

func (s *SqlResult) Add(field string, record interface{}) {
    s.fields = append(s.fields, field)
    s.vals = append(s.vals, record)
}

func (s *SqlResult) ToMap() map[string]interface{} {
    results := make(map[string]interface{})
    for i, f := range s.fields {
        results[f] = s.vals[i]
    }
    return results
}

// demo/db/console.go
package db

// Client 终端控制台
type Console struct {
    db Db
}

// Output 调用ConsoleRender完成对查询结果的渲染输出
func (c *Console) Output(render ConsoleRender) {
    fmt.Println(render.Render())
}

// Target接口，控制台db查询结果渲染接口
type ConsoleRender interface {
    Render() string
}

// TableRender表格形式的查询结果渲染Adapter
// 关键点1: 定义Adapter结构体/类
type TableRender struct {
    // 关键点2: 在Adapter中聚合Adaptee，这里是把SqlResult作为TableRender的成员变量
    result *SqlResult
}

// 关键点3: 实现Target接口，这里是实现了ConsoleRender接口
func (t *TableRender) Render() string {
    // 关键点4: 在Target接口实现中，调用Adaptee的原有方法实现具体的业务逻辑
    vals := t.result.ToMap()
    var header []string
    var data []string
    for key, val := range vals {
        header = append(header, key)
        data = append(data, fmt.Sprintf("%v", val))
    }
    builder := &strings.Builder{}
    table := tablewriter.NewWriter(builder)
    table.SetHeader(header)
    table.Append(data)
    table.Render()
    return builder.String()
}

// 这里是另一个Adapter，实现了将error渲染的功能
type ErrorRender struct {
    err error
}

func (e *ErrorRender) Render() string {
    return e.err.Error()
}

func (c *Console) Start() {
    fmt.Println("welcome to Demo DB, enter exit to end!")
    fmt.Println("> please enter a sql expression:")
    fmt.Print("> ")
    scanner := bufio.NewScanner(os.Stdin)
    for scanner.Scan() {
        sql := scanner.Text()
        if sql == "exit" {
            break
        }
        result, err := c.db.ExecSql(sql)
        if err == nil {
            // 关键点5：在需要Target接口的地方，传入适配器Adapter实例，其中创建Adapter实例时需要传入Adaptee实例
            c.Output(NewTableRender(result))
        } else {
            c.Output(NewErrorRender(err))
        }
        fmt.Println("> please enter a sql expression:")
        fmt.Print("> ")
    }
}

// 用户自定义的请求处理函数，类型为gin.HandlerFunc
func myGinHandler(c *gin.Context) {
    ... // 具体处理请求的逻辑
}

func main() {
    // 创建默认的route引擎,类型为gin.Engine
    r := gin.Default()
    // route定义
    r.GET("/my-route", myGinHandler)
    // route引擎启动
    r.Run()
}

// 用户自定义的请求处理函数，类型为http.Handler
func myHttpHandler(w http.ResponseWriter, r *http.Request) {
    ... // 具体处理请求的逻辑
}

func main() {
    // route定义
    http.HandleFunc("/my-route", myHttpHandler)
    // route启动
    http.ListenAndServe(":8080", nil)
}

// WrapH is a helper function for wrapping http.Handler and returns a Gin middleware.
func WrapH(h http.Handler) HandlerFunc {
      return func(c *Context) {
          h.ServeHTTP(c.Writer, c.Request)
      }
}

// 用户自定义的请求处理函数，类型为http.Handler
func myHttpHandler(w http.ResponseWriter, r *http.Request) {
    ... // 具体处理请求的逻辑
}

func main() {
    // 创建默认的route引擎
    r := gin.Default()
    // route定义
    r.GET("/my-route", gin.WrapH(myHttpHandler))
    // route引擎启动
    r.Run()
}