/
task.go
179 lines (148 loc) · 3.15 KB
/
task.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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
package picarcore
import (
"log"
"os"
"path/filepath"
"strings"
)
const (
JPG = ".jpg"
MOV = ".mov"
MP4 = ".mp4"
)
var (
// FINISH 表示任务已经结束
FINISH = struct{}{}
)
// Task 表示一个重命名任务
//
type Task struct {
dir string
prefix string
renameOnly bool
parseVideos bool
}
// CreateTask 创建一个新任务
//
func CreateTask(prefix string, renameOnly bool, videos bool, path string) *Task {
return &Task{
dir: path,
prefix: prefix,
parseVideos: videos,
renameOnly: renameOnly,
}
}
// Execute 执行任务
func (t *Task) Execute() error {
log.Println("正在处理文件列表")
files, err := t.getFileList()
if err != nil {
return err
}
log.Println("过滤照片文件")
finish := make(chan struct{})
count := 0
for i := 0; i < len(files); i++ {
file := files[i]
if file == "" {
continue
}
ext := strings.ToLower(filepath.Ext(file))
switch ext {
case JPG:
log.Println("\t- 照片: ", file)
count++
go t.parse(file, false, finish)
case MP4, MOV:
log.Println("\t- 影片: ", file)
go t.parse(file, true, finish)
count++
default:
log.Println("\t- 忽略: ", file)
}
}
for i := 0; i < count; i++ {
<-finish
}
log.Println("一共处理了 ", count, " 个文件")
return nil
}
// 取得文件列表
//
func (t *Task) getFileList() ([]string, error) {
log.Println("读取目录:", t.dir)
items, err := os.ReadDir(t.dir)
if err != nil {
return nil, err
}
itemsLen := len(items)
result := make([]string, itemsLen)
for i := 0; i < itemsLen; i++ {
item := items[i]
// 忽略子目录
if item.IsDir() {
continue
}
result[i] = filepath.Join(t.dir, item.Name())
}
return result, nil
}
// 重命名(和归档)照片
//
// 流程:
// 1. 如果需要归档照片,则生成新文件名时,加上要放置的目录;否则只重命名文件名
// 2. 将照片重新命名为新文件名
//
func (t *Task) parse(file string, isVideoFile bool, done chan struct{}) {
onDone := func() {
done <- FINISH
}
onError := func(err interface{}) {
log.Println(err)
// 如果出错了,忽略对该文件的处理
onDone()
}
onPanic := func() {
if err := recover(); err != nil {
onError(err)
return
}
}
defer onPanic()
log.Println("\t- 正在处理:", file)
newPath := ""
mf := NewMediaFile(file)
for counter := 0; ; counter++ {
//取得新文件名
err := mf.SetNewFilename(t.prefix, counter, isVideoFile)
if err != nil {
onError(err)
return
}
if t.renameOnly {
// 只修改文件名
newPath = filepath.Join(mf.Dir, mf.NewFilename)
} else {
// 归档照片
newPath = filepath.Join(mf.Dir, mf.ArchFolder, mf.NewFilename)
err := os.MkdirAll(filepath.Join(mf.Dir, mf.ArchFolder), 0777)
if err != nil {
panic("create directory failed, " + err.Error())
}
}
log.Println("\t- 重命名为:", newPath)
// 首先检查是否已经储存一样的新文件名的文件
found, err := isExists(newPath)
if err != nil {
onError(err)
return
}
if found {
continue
}
// 没有同名文件开始重命名文件
os.Rename(file, newPath)
onDone()
return
}
}