-
Notifications
You must be signed in to change notification settings - Fork 18
/
visitor.go
60 lines (52 loc) · 1.52 KB
/
visitor.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
package visitor
import (
"fmt"
"path"
)
// Visitor 访问者
type Visitor interface {
Visit(IResourceFile) error
}
// IResourceFile IResourceFile
type IResourceFile interface {
Accept(Visitor) error
}
// NewResourceFile NewResourceFile
func NewResourceFile(filepath string) (IResourceFile, error) {
switch path.Ext(filepath) {
case ".ppt":
return &PPTFile{path: filepath}, nil
case ".pdf":
return &PdfFile{path: filepath}, nil
default:
return nil, fmt.Errorf("not found file type: %s", filepath)
}
}
// Compressor 实现压缩功能
type Compressor struct{}
// Visit 实现访问者模式方法
// 我们可以发现由于没有函数重载,我们只能通过断言来根据不同的类型调用不同函数
// 但是我们即使不采用访问者模式,我们其实也是可以这么操作的
// 并且由于采用了类型断言,所以如果需要操作的对象比较多的话,这个函数其实也会膨胀的比较厉害
// 后续可以考虑按照命名约定使用 generate 自动生成代码
// 或者是使用反射简化
func (c *Compressor) Visit(r IResourceFile) error {
switch f := r.(type) {
case *PPTFile:
return c.VisitPPTFile(f)
case *PdfFile:
return c.VisitPDFFile(f)
default:
return fmt.Errorf("not found resource typr: %#v", r)
}
}
// VisitPPTFile VisitPPTFile
func (c *Compressor) VisitPPTFile(f *PPTFile) error {
fmt.Println("this is ppt file")
return nil
}
// VisitPDFFile VisitPDFFile
func (c *Compressor) VisitPDFFile(f *PdfFile) error {
fmt.Println("this is pdf file")
return nil
}