-
Notifications
You must be signed in to change notification settings - Fork 5
/
example_test.go
80 lines (72 loc) · 1.69 KB
/
example_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
package di_test
import (
"context"
"fmt"
"io/ioutil"
"os"
"path/filepath"
"github.com/go-tk/di"
)
func Example() {
var program di.Program
writeTempFile(&program)
provideTempDirName(&program)
provideTempFile(&program)
// NOTE: Program will rearrange Functions properly basing on dependency analysis.
defer program.Clean()
program.MustRun(context.Background())
// Output:
// 1. create temp dir
// 2. create and open temp file
// 3. write temp file
// 4. close and delete temp file
// 5. delete temp dir
}
func provideTempDirName(program *di.Program) {
var tempDirName string
program.MustNewFunction(
di.Result("TEMP_DIR_NAME", &tempDirName),
di.Body(func(context.Context) error {
fmt.Println("1. create temp dir")
var err error
tempDirName, err = ioutil.TempDir("", "")
return err
}),
di.Cleanup(func() {
fmt.Println("5. delete temp dir")
os.Remove(tempDirName)
}),
)
}
func provideTempFile(program *di.Program) {
var (
tempDirName string
tempFile *os.File
)
program.MustNewFunction(
di.Argument("TEMP_DIR_NAME", &tempDirName),
di.Result("TEMP_FILE", &tempFile),
di.Body(func(context.Context) error {
fmt.Println("2. create and open temp file")
var err error
tempFile, err = os.Create(filepath.Join(tempDirName, "temp"))
return err
}),
di.Cleanup(func() {
fmt.Println("4. close and delete temp file")
tempFile.Close()
os.Remove(tempFile.Name())
}),
)
}
func writeTempFile(program *di.Program) {
var tempFile *os.File
program.MustNewFunction(
di.Argument("TEMP_FILE", &tempFile),
di.Body(func(context.Context) error {
fmt.Println("3. write temp file")
_, err := tempFile.WriteString("hello world")
return err
}),
)
}