Go如何更优雅的构造结构数据

场景：你需要根据一些参数来构造一个结构体，但是有不想让用户直接构造这个结构体，于是不得不提供一个函数来构造结构体。由于Go不支持重载，你不得不编写各种参数的构造函数，对于命名困难户来说，无异于噩梦。

我们通过一个例子来看如何优雅的解决这个问题。假设我们有一个drawer包：

package drawer

type Drawer interface {
	DrawLine(p1, p2 Point)
	DrawText(p Point, text string)
	DrawCircle(p Point, r int)
	DrawRect(p1, p2 Point)
}

type Point struct {
	X, Y int
}

type Color struct {
	R, G, B, A uint8
}

type drawer struct {
	lineWidth  int
	foreground Color
	background Color
}

func (d *drawer) DrawLine(p1, p2 Point) {
	println("line")
}

func (d *drawer) DrawText(p Point, text string) {
	println("text")
}

func (d *drawer) DrawCircle(p Point, r int) {
	println("circle")
}

func (d *drawer) DrawRect(p1, p2 Point) {
	println("rect")
}

func New() Drawer {
	return &drawer{
		lineWidth: 1,
		foreground: Color{0,0,0,0},
		background: Color{255,255,255,255},
	}
}

Go语言习惯上将New作为构造函数。现在如果想自定义构造各种drawer，有几种方法。

第一种是提供各种NewXxx函数。

第二种是将参数放到一个配置结构体，New函数接受这个配置结构体作为参数。

以上两种方式均在实践中被用到。第一种方式不够优雅，也不够灵活。第二种方式可以只提供一个New函数，但是我们不得不暴露另一个结构体了。

第三种方式是利用函数是一等公民来实现灵活、优雅的配置。

type option func(*drawer)

func LineWidth(w int) option {
	return func(d *drawer) {
		d.lineWidth = w
	}
}

func ForgeColor(color Color) option {
	return func(d *drawer) {
		d.foreColor = color
	}
}

func BackgroundColor(color Color) option {
	return func(d *drawer) {
		d.background = color
	}
}

func New(ops ...option) *Drawer {
	d := drawer{
		lineWidth: 1,
		foreground: Color{0,0,0,0},
		background: Color{255,255,255,255},
	}
	for _, op := range ops {
		op(&d)
	}
	return &d
}

现在我们可以这样利用drawer包的New函数了。

package main

import (
	"drawer"
)

func main() {
	d1 := drawer.New(drawer.LineWidth(5))
	d2 := drawer.New(drawer.Background(drawer.Color{100,200,40,0}))
	d1.DrawLint(Point{1,1}, Point{10,10})
	d2.DrawText(Point{5,5}, "hello")
}

当然这种方式需要编写额外的option函数，如果包命很长的话，用起来也确实难受，不过好在go支持别名导入，我们可以换一个短一点的名字。

除了初始化，在option函数中我们还可以做更多的事。甚至对于可能出错的初始化我们可以让option函数返回一个错误。

type option func(*drawer) error

func LineWidth(w int) option {
	return func(d *drawer) error {
		if (w < 1) {
			return errors.New("线宽必须大于0")
		}
		d.lineWidth = w
		return nil
	}
}

优雅永不过时！

---