golang设计模式-成员变量赋值

gauze · · 1732 次点击 · · 开始浏览

这是一个创建于的文章，其中的信息可能已经有所发展或是发生改变。

常见golang的struct赋值有两种：

　　1）定义变量同时初始化

　　val := &Options{
	UID:int(1),
　　}

　　2）先定义变量，再赋值

val := new(Options)
val.UID = int(3)

赋值设计模式：

　　1）每个属性一个方法，方法名称与属性有关

　　2）赋值方法的返回值是统一的，可由同一个方法无差异执行

package main

import (
	"os"
)

func main() {

	//err := New("/tmp/empty.txt")
	//if err != nil {
	//	panic(err)
	//}

	err := New("/tmp/file.txt", Length(200), UID(1000), Contents("Lorem Ipsum Dolor Amet"))
	if err != nil {
		panic(err)
	}

}

// 结构体定义
type Options struct {
	UID         int
	GID         int
	Flags       int
	Contents    string
	Permissions os.FileMode
	Length 		int
}

// 赋值方法约束，接口型函数(接口只有一个函数，接口型函数解决很完美，没必要单独定义接口并实现)
type Option func(*Options)

// 其中一个属性的赋值函数
func UID(userID int) Option {
	return func(args *Options) {
		args.UID = userID
	}
}

func Length(Length int) Option {
	return func(args *Options) {
		args.Length = Length
	}
}

func GID(groupID int) Option {
	return func(args *Options) {
		args.GID = groupID
	}
}

func Contents(c string) Option {
	return func(args *Options) {
		args.Contents = c
	}
}

func Permissions(perms os.FileMode) Option {
	return func(args *Options) {
		args.Permissions = perms
	}
}

// 创建对象唯一入口
func New(filepath string, setters ...Option) error {
	// 创建默认值对象
	args := &Options{
		UID:         os.Getuid(),
		GID:         os.Getgid(),
		Contents:    "",
		Permissions: 0666,
		Flags:       os.O_CREATE | os.O_EXCL | os.O_WRONLY,
	}

	// 赋值操作
	for _, setter := range setters {
		setter(args)
	}

	// 对象其他初始化操作
	f, err := os.OpenFile(filepath, args.Flags, args.Permissions)
	if err != nil {
		return err
	} else {
		defer f.Close()
	}

	if _, err := f.WriteString(args.Contents); err != nil {
		return err
	}

	return f.Chown(args.UID, args.GID)
}

有疑问加站长微信联系（非本文作者）