-
Notifications
You must be signed in to change notification settings - Fork 0
/
rotate.go
99 lines (82 loc) · 1.72 KB
/
rotate.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
// Copyright 2017 by caixw, All rights reserved.
// Use of this source code is governed by a MIT
// license that can be found in the LICENSE file.
package initfunc
import (
"errors"
"fmt"
"io"
"strconv"
"strings"
"modules/logs/logger/writers"
)
const (
b int64 = 1 << (10 * iota)
kb
mb
gb
)
// Rotate 的初始化函数。
func Rotate(args map[string]string) (io.Writer, error) {
prefix, found := args["prefix"]
if !found {
prefix = ""
}
dir, found := args["dir"]
if !found {
return nil, argNotFoundErr("rotate", "dir")
}
sizeStr, found := args["size"]
if !found {
return nil, argNotFoundErr("rotate", "size")
}
size, err := toByte(sizeStr)
if err != nil {
return nil, err
}
return writers.NewRotate(prefix, dir, int(size))
}
// 将字符串转换成以字节为单位的数值。
// 粗略计算,并不 100% 正确,小数只取整数部分。
// 支持以下格式:
// 1024
// 1k
// 1M
// 1G
// 后缀单位只支持 k,g,m,不区分大小写。
func toByte(str string) (int64, error) {
if len(str) == 0 {
return -1, errors.New("不能传递空值")
}
str = strings.ToLower(str)
scale := b
unit := str[len(str)-1]
switch {
case unit >= '0' && unit <= '9':
scale = b
case unit == 'b':
scale = b
case unit == 'k':
scale = kb
case unit == 'm':
scale = mb
case unit == 'g':
scale = gb
default:
return -1, fmt.Errorf("无法识别的单位:[%v]", unit)
}
if scale > 1 {
str = str[:len(str)-1]
}
if len(str) == 0 {
return -1, errors.New("传递了一个空值")
}
size, err := strconv.ParseFloat(str, 32)
if err != nil {
return -1, err
}
if size <= 0 {
return -1, fmt.Errorf("大小不能小于 0,当前值为:[%d]", size)
}
return int64(size) * scale, nil
}