|
@@ -1,159 +1,151 @@
|
1
|
1
|
package main
|
2
|
2
|
|
3
|
3
|
import (
|
4
|
|
- "context"
|
5
|
|
- "fmt"
|
6
|
|
- "gdyb/models"
|
7
|
4
|
_ "gdyb/routers"
|
8
|
|
- "gdyb/service"
|
9
|
5
|
"github.com/astaxie/beego"
|
10
|
|
- "github.com/qiniu/api.v7/auth/qbox"
|
11
|
|
- "github.com/qiniu/api.v7/storage"
|
12
|
|
- "os"
|
13
|
|
- "strconv"
|
14
|
|
- "time"
|
15
|
6
|
)
|
16
|
7
|
|
17
|
8
|
func init() {
|
18
|
|
- service.ConnectDB()
|
19
|
|
- org_id, _ := beego.AppConfig.Int64("org_id")
|
20
|
|
- miConfig, _ := service.FindMedicalInsuranceInfo(org_id)
|
21
|
|
- CreateLog(miConfig)
|
22
|
|
- UploadLog(miConfig)
|
|
9
|
+ //service.ConnectDB()
|
|
10
|
+ //org_id, _ := beego.AppConfig.Int64("org_id")
|
|
11
|
+ //miConfig, _ := service.FindMedicalInsuranceInfo(org_id)
|
|
12
|
+ //CreateLog(miConfig)
|
|
13
|
+ //UploadLog(miConfig)
|
23
|
14
|
}
|
24
|
15
|
func main() {
|
25
|
16
|
beego.Run()
|
26
|
17
|
}
|
27
|
|
-func CreateLog(miConfig models.MedicalInsuranceOrgConfig) {
|
28
|
|
- Mkdir(miConfig.OrgName + "日志")
|
29
|
|
- //org_id,_ := beego.AppConfig.Int("org_id")
|
30
|
|
- month := time.Unix(1557042972, 0).Format("1")
|
31
|
|
- year := time.Now().Format("2006")
|
32
|
|
- month = time.Now().Format("01")
|
33
|
|
- day := time.Now().Format("02")
|
34
|
|
- file := strconv.FormatInt(miConfig.UserOrgId, 10) + "_" + year + month + day + "_log"
|
35
|
|
- file_name := file + ".txt"
|
36
|
|
- file_path := miConfig.OrgName + "日志" + "/" + file_name
|
37
|
|
- exist, _ := PathExists(file_path)
|
38
|
|
- if exist { //存在
|
39
|
|
- fmt.Println("存在")
|
40
|
|
- f, err := os.OpenFile(file_path, os.O_WRONLY, 0644)
|
41
|
|
- if err != nil {
|
42
|
|
- fmt.Println("read fail")
|
43
|
|
- }
|
44
|
|
- content := "\r\r\r"
|
45
|
|
- n, _ := f.Seek(0, 2)
|
46
|
|
- _, err = f.WriteAt([]byte(content), n)
|
47
|
18
|
|
48
|
|
- } else { //不存在
|
49
|
|
- fmt.Println("文件不存在,创建文件")
|
50
|
|
- f, err := os.Create(miConfig.OrgName + "日志" + "/" + file_name)
|
51
|
|
- defer f.Close()
|
52
|
|
- if err != nil {
|
53
|
|
- } else {
|
54
|
|
- _, err = f.Write([]byte("记录日志"))
|
55
|
|
- }
|
56
|
|
- }
|
57
|
|
-
|
58
|
|
-}
|
59
|
|
-func UploadLog(miConfig models.MedicalInsuranceOrgConfig) {
|
60
|
|
- //获取七牛云上的文件信息,不存在则表示没上传到七牛云,上传昨天日志信息
|
61
|
|
- nTime := time.Now()
|
62
|
|
- yesTime := nTime.AddDate(0, 0, -1)
|
63
|
|
- file_key := strconv.FormatInt(miConfig.UserOrgId, 10) + "_" + yesTime.Format("20060102") + "_log.txt"
|
64
|
|
- mac := qbox.NewMac(beego.AppConfig.String("qiniu_accesskey"), beego.AppConfig.String("qiniu_secretkey"))
|
65
|
|
- cfg := storage.Config{
|
66
|
|
- // 是否使用https域名进行资源管理
|
67
|
|
- UseHTTPS: true,
|
68
|
|
- }
|
69
|
|
- // 指定空间所在的区域,如果不指定将自动探测
|
70
|
|
- // 如果没有特殊需求,默认不需要指定
|
71
|
|
- //cfg.Zone=&storage.ZoneHuabei
|
72
|
|
- bucketManager := storage.NewBucketManager(mac, &cfg)
|
73
|
|
- bucket := beego.AppConfig.String("qiniu_bucket")
|
74
|
|
- key := file_key
|
75
|
|
- file, sErr := bucketManager.Stat(bucket, key)
|
76
|
|
- fmt.Println(file)
|
77
|
|
- fmt.Println(sErr)
|
78
|
|
- fmt.Println(miConfig.OrgName + "日志" + "/" + file_key)
|
79
|
|
-
|
80
|
|
- if sErr != nil {
|
81
|
|
- fmt.Println("七牛云上文件不存在,上传到七牛云")
|
82
|
|
- //判断本地文件是否存在,存在则执行上传操作
|
83
|
|
- exist, _ := PathExists(miConfig.OrgName + "日志" + "/" + file_key)
|
84
|
|
- if exist {
|
85
|
|
- upload_qiniu(miConfig.OrgName+"日志"+"/"+file_key, file_key)
|
86
|
|
- } else {
|
87
|
|
- fmt.Println("本地文件不存在")
|
88
|
|
-
|
89
|
|
- }
|
90
|
|
- } else {
|
91
|
|
- fmt.Println("七牛云文件存在")
|
92
|
|
- }
|
93
|
|
-
|
94
|
|
-}
|
95
|
|
-func Mkdir(dir string) {
|
96
|
|
- // 创建文件夹
|
97
|
|
- exist, err := PathExists(dir)
|
98
|
|
- if err != nil {
|
99
|
|
- fmt.Println(err.Error())
|
100
|
|
- } else {
|
101
|
|
- if exist {
|
102
|
|
- fmt.Println(dir + "文件夹已存在!")
|
103
|
|
- } else {
|
104
|
|
- // 文件夹名称,权限
|
105
|
|
- err := os.Mkdir(dir, os.ModePerm)
|
106
|
|
- if err != nil {
|
107
|
|
- fmt.Println(dir+"文件夹创建失败:", err.Error())
|
108
|
|
- } else {
|
109
|
|
- fmt.Println(dir + "文件夹创建成功!")
|
110
|
|
- }
|
111
|
|
- }
|
112
|
|
- }
|
113
|
|
-}
|
114
|
|
-func PathExists(path string) (bool, error) {
|
115
|
|
- _, err := os.Stat(path)
|
116
|
|
- if err == nil {
|
117
|
|
- return true, nil
|
118
|
|
- }
|
119
|
|
- if os.IsNotExist(err) {
|
120
|
|
- return false, nil
|
121
|
|
- }
|
122
|
|
- return false, err
|
123
|
|
-}
|
124
|
|
-func upload_qiniu(filePath string, keys string) {
|
125
|
|
- key := keys
|
126
|
|
- //上传凭证,关于凭证这块大家可以去看看官方文档
|
127
|
|
- putPolicy := storage.PutPolicy{
|
128
|
|
- Scope: beego.AppConfig.String("qiniu_bucket"),
|
129
|
|
- }
|
130
|
|
- mac := qbox.NewMac(beego.AppConfig.String("qiniu_accesskey"), beego.AppConfig.String("qiniu_secretkey"))
|
131
|
|
- upToken := putPolicy.UploadToken(mac)
|
132
|
|
- cfg := storage.Config{}
|
133
|
|
- //空间对应机房
|
134
|
|
- //其中关于Zone对象和机房的关系如下:
|
135
|
|
- // 机房 Zone对象
|
136
|
|
- // 华东 storage.ZoneHuadong
|
137
|
|
- // 华北 storage.ZoneHuabei
|
138
|
|
- // 华南 storage.ZoneHuanan
|
139
|
|
- // 北美 storage.ZoneBeimei
|
140
|
|
- //七牛云存储空间设置首页有存储区域
|
141
|
|
- cfg.Zone = &storage.ZoneHuadong
|
142
|
|
- //不启用HTTPS域名
|
143
|
|
- cfg.UseHTTPS = false
|
144
|
|
- //不使用CND加速
|
145
|
|
- cfg.UseCdnDomains = false
|
146
|
|
- //构建上传表单对象
|
147
|
|
- formUploader := storage.NewFormUploader(&cfg)
|
148
|
|
- ret := storage.PutRet{}
|
149
|
|
- // 可选
|
150
|
|
- putExtra := storage.PutExtra{
|
151
|
|
- Params: map[string]string{},
|
152
|
|
- }
|
153
|
|
- err := formUploader.PutFile(context.Background(), &ret, upToken, key, filePath, &putExtra)
|
154
|
|
- if err != nil {
|
155
|
|
- fmt.Println(err)
|
156
|
|
- return
|
157
|
|
- }
|
158
|
|
- fmt.Println(ret.Key, ret.Hash)
|
159
|
|
-}
|
|
19
|
+//func CreateLog(miConfig models.MedicalInsuranceOrgConfig) {
|
|
20
|
+// Mkdir(miConfig.OrgName + "日志")
|
|
21
|
+// //org_id,_ := beego.AppConfig.Int("org_id")
|
|
22
|
+// month := time.Unix(1557042972, 0).Format("1")
|
|
23
|
+// year := time.Now().Format("2006")
|
|
24
|
+// month = time.Now().Format("01")
|
|
25
|
+// day := time.Now().Format("02")
|
|
26
|
+// file := strconv.FormatInt(miConfig.UserOrgId, 10) + "_" + year + month + day + "_log"
|
|
27
|
+// file_name := file + ".txt"
|
|
28
|
+// file_path := miConfig.OrgName + "日志" + "/" + file_name
|
|
29
|
+// exist, _ := PathExists(file_path)
|
|
30
|
+// if exist { //存在
|
|
31
|
+// fmt.Println("存在")
|
|
32
|
+// f, err := os.OpenFile(file_path, os.O_WRONLY, 0644)
|
|
33
|
+// if err != nil {
|
|
34
|
+// fmt.Println("read fail")
|
|
35
|
+// }
|
|
36
|
+// content := "\r\r\r"
|
|
37
|
+// n, _ := f.Seek(0, 2)
|
|
38
|
+// _, err = f.WriteAt([]byte(content), n)
|
|
39
|
+//
|
|
40
|
+// } else { //不存在
|
|
41
|
+// fmt.Println("文件不存在,创建文件")
|
|
42
|
+// f, err := os.Create(miConfig.OrgName + "日志" + "/" + file_name)
|
|
43
|
+// defer f.Close()
|
|
44
|
+// if err != nil {
|
|
45
|
+// } else {
|
|
46
|
+// _, err = f.Write([]byte("记录日志"))
|
|
47
|
+// }
|
|
48
|
+// }
|
|
49
|
+//
|
|
50
|
+//}
|
|
51
|
+//func UploadLog(miConfig models.MedicalInsuranceOrgConfig) {
|
|
52
|
+// //获取七牛云上的文件信息,不存在则表示没上传到七牛云,上传昨天日志信息
|
|
53
|
+// nTime := time.Now()
|
|
54
|
+// yesTime := nTime.AddDate(0, 0, -1)
|
|
55
|
+// file_key := strconv.FormatInt(miConfig.UserOrgId, 10) + "_" + yesTime.Format("20060102") + "_log.txt"
|
|
56
|
+// mac := qbox.NewMac(beego.AppConfig.String("qiniu_accesskey"), beego.AppConfig.String("qiniu_secretkey"))
|
|
57
|
+// cfg := storage.Config{
|
|
58
|
+// // 是否使用https域名进行资源管理
|
|
59
|
+// UseHTTPS: true,
|
|
60
|
+// }
|
|
61
|
+// // 指定空间所在的区域,如果不指定将自动探测
|
|
62
|
+// // 如果没有特殊需求,默认不需要指定
|
|
63
|
+// //cfg.Zone=&storage.ZoneHuabei
|
|
64
|
+// bucketManager := storage.NewBucketManager(mac, &cfg)
|
|
65
|
+// bucket := beego.AppConfig.String("qiniu_bucket")
|
|
66
|
+// key := file_key
|
|
67
|
+// file, sErr := bucketManager.Stat(bucket, key)
|
|
68
|
+// fmt.Println(file)
|
|
69
|
+// fmt.Println(sErr)
|
|
70
|
+// fmt.Println(miConfig.OrgName + "日志" + "/" + file_key)
|
|
71
|
+//
|
|
72
|
+// if sErr != nil {
|
|
73
|
+// fmt.Println("七牛云上文件不存在,上传到七牛云")
|
|
74
|
+// //判断本地文件是否存在,存在则执行上传操作
|
|
75
|
+// exist, _ := PathExists(miConfig.OrgName + "日志" + "/" + file_key)
|
|
76
|
+// if exist {
|
|
77
|
+// upload_qiniu(miConfig.OrgName+"日志"+"/"+file_key, file_key)
|
|
78
|
+// } else {
|
|
79
|
+// fmt.Println("本地文件不存在")
|
|
80
|
+//
|
|
81
|
+// }
|
|
82
|
+// } else {
|
|
83
|
+// fmt.Println("七牛云文件存在")
|
|
84
|
+// }
|
|
85
|
+//
|
|
86
|
+//}
|
|
87
|
+//func Mkdir(dir string) {
|
|
88
|
+// // 创建文件夹
|
|
89
|
+// exist, err := PathExists(dir)
|
|
90
|
+// if err != nil {
|
|
91
|
+// fmt.Println(err.Error())
|
|
92
|
+// } else {
|
|
93
|
+// if exist {
|
|
94
|
+// fmt.Println(dir + "文件夹已存在!")
|
|
95
|
+// } else {
|
|
96
|
+// // 文件夹名称,权限
|
|
97
|
+// err := os.Mkdir(dir, os.ModePerm)
|
|
98
|
+// if err != nil {
|
|
99
|
+// fmt.Println(dir+"文件夹创建失败:", err.Error())
|
|
100
|
+// } else {
|
|
101
|
+// fmt.Println(dir + "文件夹创建成功!")
|
|
102
|
+// }
|
|
103
|
+// }
|
|
104
|
+// }
|
|
105
|
+//}
|
|
106
|
+//func PathExists(path string) (bool, error) {
|
|
107
|
+// _, err := os.Stat(path)
|
|
108
|
+// if err == nil {
|
|
109
|
+// return true, nil
|
|
110
|
+// }
|
|
111
|
+// if os.IsNotExist(err) {
|
|
112
|
+// return false, nil
|
|
113
|
+// }
|
|
114
|
+// return false, err
|
|
115
|
+//}
|
|
116
|
+//func upload_qiniu(filePath string, keys string) {
|
|
117
|
+// key := keys
|
|
118
|
+// //上传凭证,关于凭证这块大家可以去看看官方文档
|
|
119
|
+// putPolicy := storage.PutPolicy{
|
|
120
|
+// Scope: beego.AppConfig.String("qiniu_bucket"),
|
|
121
|
+// }
|
|
122
|
+// mac := qbox.NewMac(beego.AppConfig.String("qiniu_accesskey"), beego.AppConfig.String("qiniu_secretkey"))
|
|
123
|
+// upToken := putPolicy.UploadToken(mac)
|
|
124
|
+// cfg := storage.Config{}
|
|
125
|
+// //空间对应机房
|
|
126
|
+// //其中关于Zone对象和机房的关系如下:
|
|
127
|
+// // 机房 Zone对象
|
|
128
|
+// // 华东 storage.ZoneHuadong
|
|
129
|
+// // 华北 storage.ZoneHuabei
|
|
130
|
+// // 华南 storage.ZoneHuanan
|
|
131
|
+// // 北美 storage.ZoneBeimei
|
|
132
|
+// //七牛云存储空间设置首页有存储区域
|
|
133
|
+// cfg.Zone = &storage.ZoneHuadong
|
|
134
|
+// //不启用HTTPS域名
|
|
135
|
+// cfg.UseHTTPS = false
|
|
136
|
+// //不使用CND加速
|
|
137
|
+// cfg.UseCdnDomains = false
|
|
138
|
+// //构建上传表单对象
|
|
139
|
+// formUploader := storage.NewFormUploader(&cfg)
|
|
140
|
+// ret := storage.PutRet{}
|
|
141
|
+// // 可选
|
|
142
|
+// putExtra := storage.PutExtra{
|
|
143
|
+// Params: map[string]string{},
|
|
144
|
+// }
|
|
145
|
+// err := formUploader.PutFile(context.Background(), &ret, upToken, key, filePath, &putExtra)
|
|
146
|
+// if err != nil {
|
|
147
|
+// fmt.Println(err)
|
|
148
|
+// return
|
|
149
|
+// }
|
|
150
|
+// fmt.Println(ret.Key, ret.Hash)
|
|
151
|
+//}
|