1
0
mirror of https://github.com/tmrts/go-patterns.git synced 2024-11-25 06:26:06 +03:00

Add functional options idiom

This commit is contained in:
Tamer Tas 2016-04-15 15:21:44 +03:00
parent abcdf32672
commit c0937dd592
3 changed files with 91 additions and 66 deletions

View File

@ -104,7 +104,7 @@ __Idioms__:
| Pattern | Description | | Pattern | Description |
|:-------:| ----------- | |:-------:| ----------- |
| [Functional Options](functional_options.go) | Allows creating clean APIs with sane defaults and idiomatic overrides | | [Functional Options](idiom/functional_options.md) | Allows creating clean APIs with sane defaults and idiomatic overrides |
__Anti-Patterns__: __Anti-Patterns__:

View File

@ -1,65 +0,0 @@
package main
import "os"
type Arguments struct {
UID int
GID int
Flags int
Contents string
Permissions os.FileMode
}
type Argument func(*Arguments)
func UID(userID int) Argument {
return func(args *Arguments) {
args.UID = userID
}
}
func GID(groupID int) Argument {
return func(args *Arguments) {
args.GID = groupID
}
}
func Contents(c string) Argument {
return func(args *Arguments) {
args.Contents = c
}
}
func Permissions(perms os.FileMode) Argument {
return func(args *Arguments) {
args.Permissions = perms
}
}
func New(filepath string, setters ...Argument) error {
// Default Arguments
args := &Arguments{
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)
}

View File

@ -0,0 +1,90 @@
# Functional Options
Functional options are a method of implementing clean/eloquent APIs in Go.
Options implemented as a function set the state of that option.
## Implementation
### Options
```go
package file
type Options struct {
UID int
GID int
Flags int
Contents string
Permissions os.FileMode
}
type Option func(*Options)
func UID(userID int) Option {
return func(args *Options) {
args.UID = userID
}
}
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
}
}
```
### Constructor
```go
package file
func New(filepath string, setters ...Option) error {
// Default Options
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)
}
```
## Usage
```go
emptyFile, err := file.New("/tmp/empty.txt")
if err != nil {
panic(err)
}
fillerFile, err := file.New("/tmp/file.txt", file.UID(1000), file.Contents("Lorem Ipsum Dolor Amet"))
if err != nil {
panic(err)
}
```