Browse Source

Merge branch 'master' of http://git.shengws.com/csx/gdyb

csx 2 years ago
parent
commit
c34dd5e2a8
1 changed files with 120 additions and 112 deletions
  1. 120 112
      main.go

+ 120 - 112
main.go View File

1
 package main
1
 package main
2
 
2
 
3
 import (
3
 import (
4
+	"context"
4
 	//"context"
5
 	//"context"
5
 	"fmt"
6
 	"fmt"
7
+	"gdyb/models"
8
+	"gdyb/service"
9
+	"github.com/qiniu/api.v7/auth/qbox"
10
+	"github.com/qiniu/api.v7/storage"
11
+	"strconv"
12
+	"time"
13
+
6
 	//"gdyb/models"
14
 	//"gdyb/models"
7
 	_ "gdyb/routers"
15
 	_ "gdyb/routers"
8
 	//"github.com/qiniu/api.v7/auth/qbox"
16
 	//"github.com/qiniu/api.v7/auth/qbox"
21
 )
29
 )
22
 
30
 
23
 func init() {
31
 func init() {
24
-	//service.ConnectDB()
25
-	//org_id, _ := beego.AppConfig.Int64("org_id")
26
-	//miConfig, _ := service.FindMedicalInsuranceInfo(org_id)
27
-	//CreateLog(miConfig)
28
-	//UploadLog(miConfig)
29
-	//if org_id == 10191 {
30
-	//	service.GetFjAuthorizationInfo(miConfig.Code)
31
-	//}
32
+	service.ConnectDB()
33
+	org_id, _ := beego.AppConfig.Int64("org_id")
34
+	miConfig, _ := service.FindMedicalInsuranceInfo(org_id)
35
+	CreateLog(miConfig)
36
+	UploadLog(miConfig)
37
+	if org_id == 10191 {
38
+		service.GetFjAuthorizationInfo(miConfig.Code)
39
+	}
32
 }
40
 }
33
 func main() {
41
 func main() {
34
 	beego.Run()
42
 	beego.Run()
790
 //	}
798
 //	}
791
 //}
799
 //}
792
 
800
 
793
-//func CreateLog(miConfig models.MedicalInsuranceOrgConfig) {
794
-//	Mkdir(miConfig.OrgName + "日志")
795
-//	//org_id,_ := beego.AppConfig.Int("org_id")
796
-//	month := time.Unix(1557042972, 0).Format("1")
797
-//	year := time.Now().Format("2006")
798
-//	month = time.Now().Format("01")
799
-//	day := time.Now().Format("02")
800
-//	file := strconv.FormatInt(miConfig.UserOrgId, 10) + "_" + year + month + day + "_log"
801
-//	file_name := file + ".txt"
802
-//	file_path := miConfig.OrgName + "日志" + "/" + file_name
803
-//	exist, _ := PathExists(file_path)
804
-//	if exist { //存在
805
-//		fmt.Println("存在")
806
-//		f, err := os.OpenFile(file_path, os.O_WRONLY, 0644)
807
-//		if err != nil {
808
-//			fmt.Println("read fail")
809
-//		}
810
-//		content := "\r\r\r"
811
-//		n, _ := f.Seek(0, 2)
812
-//		_, err = f.WriteAt([]byte(content), n)
813
-//
814
-//	} else { //不存在
815
-//		fmt.Println("文件不存在,创建文件")
816
-//		f, err := os.Create(miConfig.OrgName + "日志" + "/" + file_name)
817
-//		defer f.Close()
818
-//		if err != nil {
819
-//		} else {
820
-//			_, err = f.Write([]byte("记录日志"))
821
-//		}
822
-//	}
823
-//
824
-//}
801
+func CreateLog(miConfig models.MedicalInsuranceOrgConfig) {
802
+	Mkdir(miConfig.OrgName + "日志")
803
+	//org_id,_ := beego.AppConfig.Int("org_id")
804
+	month := time.Unix(1557042972, 0).Format("1")
805
+	year := time.Now().Format("2006")
806
+	month = time.Now().Format("01")
807
+	day := time.Now().Format("02")
808
+	file := strconv.FormatInt(miConfig.UserOrgId, 10) + "_" + year + month + day + "_log"
809
+	file_name := file + ".txt"
810
+	file_path := miConfig.OrgName + "日志" + "/" + file_name
811
+	exist, _ := PathExists(file_path)
812
+	if exist { //存在
813
+		fmt.Println("存在")
814
+		f, err := os.OpenFile(file_path, os.O_WRONLY, 0644)
815
+		if err != nil {
816
+			fmt.Println("read fail")
817
+		}
818
+		content := "\r\r\r"
819
+		n, _ := f.Seek(0, 2)
820
+		_, err = f.WriteAt([]byte(content), n)
825
 
821
 
826
-//func UploadLog(miConfig models.MedicalInsuranceOrgConfig) {
827
-//	//获取七牛云上的文件信息,不存在则表示没上传到七牛云,上传昨天日志信息
828
-//	nTime := time.Now()
829
-//	yesTime := nTime.AddDate(0, 0, -1)
830
-//	file_key := strconv.FormatInt(miConfig.UserOrgId, 10) + "_" + yesTime.Format("20060102") + "_log.txt"
831
-//	mac := qbox.NewMac(beego.AppConfig.String("qiniu_accesskey"), beego.AppConfig.String("qiniu_secretkey"))
832
-//	cfg := storage.Config{
833
-//		// 是否使用https域名进行资源管理
834
-//		UseHTTPS: true,
835
-//	}
836
-//	// 指定空间所在的区域,如果不指定将自动探测
837
-//	// 如果没有特殊需求,默认不需要指定
838
-//	//cfg.Zone=&storage.ZoneHuabei
839
-//	bucketManager := storage.NewBucketManager(mac, &cfg)
840
-//	bucket := beego.AppConfig.String("qiniu_bucket")
841
-//	key := file_key
842
-//	file, sErr := bucketManager.Stat(bucket, key)
843
-//	fmt.Println(file)
844
-//	fmt.Println(sErr)
845
-//	fmt.Println(miConfig.OrgName + "日志" + "/" + file_key)
846
-//
847
-//	if sErr != nil {
848
-//		fmt.Println("七牛云上文件不存在,上传到七牛云")
849
-//		//判断本地文件是否存在,存在则执行上传操作
850
-//		exist, _ := PathExists(miConfig.OrgName + "日志" + "/" + file_key)
851
-//		if exist {
852
-//			upload_qiniu(miConfig.OrgName+"日志"+"/"+file_key, file_key)
853
-//		} else {
854
-//			fmt.Println("本地文件不存在")
855
-//
856
-//		}
857
-//	} else {
858
-//		fmt.Println("七牛云文件存在")
859
-//	}
860
-//
861
-//}
822
+	} else { //不存在
823
+		fmt.Println("文件不存在,创建文件")
824
+		f, err := os.Create(miConfig.OrgName + "日志" + "/" + file_name)
825
+		defer f.Close()
826
+		if err != nil {
827
+		} else {
828
+			_, err = f.Write([]byte("记录日志"))
829
+		}
830
+	}
831
+
832
+}
833
+
834
+func UploadLog(miConfig models.MedicalInsuranceOrgConfig) {
835
+	//获取七牛云上的文件信息,不存在则表示没上传到七牛云,上传昨天日志信息
836
+	nTime := time.Now()
837
+	yesTime := nTime.AddDate(0, 0, -1)
838
+	file_key := strconv.FormatInt(miConfig.UserOrgId, 10) + "_" + yesTime.Format("20060102") + "_log.txt"
839
+	mac := qbox.NewMac(beego.AppConfig.String("qiniu_accesskey"), beego.AppConfig.String("qiniu_secretkey"))
840
+	cfg := storage.Config{
841
+		// 是否使用https域名进行资源管理
842
+		UseHTTPS: true,
843
+	}
844
+	// 指定空间所在的区域,如果不指定将自动探测
845
+	// 如果没有特殊需求,默认不需要指定
846
+	//cfg.Zone=&storage.ZoneHuabei
847
+	bucketManager := storage.NewBucketManager(mac, &cfg)
848
+	bucket := beego.AppConfig.String("qiniu_bucket")
849
+	key := file_key
850
+	file, sErr := bucketManager.Stat(bucket, key)
851
+	fmt.Println(file)
852
+	fmt.Println(sErr)
853
+	fmt.Println(miConfig.OrgName + "日志" + "/" + file_key)
854
+
855
+	if sErr != nil {
856
+		fmt.Println("七牛云上文件不存在,上传到七牛云")
857
+		//判断本地文件是否存在,存在则执行上传操作
858
+		exist, _ := PathExists(miConfig.OrgName + "日志" + "/" + file_key)
859
+		if exist {
860
+			upload_qiniu(miConfig.OrgName+"日志"+"/"+file_key, file_key)
861
+		} else {
862
+			fmt.Println("本地文件不存在")
863
+
864
+		}
865
+	} else {
866
+		fmt.Println("七牛云文件存在")
867
+	}
868
+
869
+}
862
 func Mkdir(dir string) {
870
 func Mkdir(dir string) {
863
 	// 创建文件夹
871
 	// 创建文件夹
864
 	exist, err := PathExists(dir)
872
 	exist, err := PathExists(dir)
889
 	return false, err
897
 	return false, err
890
 }
898
 }
891
 
899
 
892
-//func upload_qiniu(filePath string, keys string) {
893
-//	key := keys
894
-//	//上传凭证,关于凭证这块大家可以去看看官方文档
895
-//	putPolicy := storage.PutPolicy{
896
-//		Scope: beego.AppConfig.String("qiniu_bucket"),
897
-//	}
898
-//	mac := qbox.NewMac(beego.AppConfig.String("qiniu_accesskey"), beego.AppConfig.String("qiniu_secretkey"))
899
-//	upToken := putPolicy.UploadToken(mac)
900
-//	cfg := storage.Config{}
901
-//	//空间对应机房
902
-//	//其中关于Zone对象和机房的关系如下:
903
-//	//    机房    Zone对象
904
-//	//    华东    storage.ZoneHuadong
905
-//	//    华北    storage.ZoneHuabei
906
-//	//    华南    storage.ZoneHuanan
907
-//	//    北美    storage.ZoneBeimei
908
-//	//七牛云存储空间设置首页有存储区域
909
-//	cfg.Zone = &storage.ZoneHuadong
910
-//	//不启用HTTPS域名
911
-//	cfg.UseHTTPS = false
912
-//	//不使用CND加速
913
-//	cfg.UseCdnDomains = false
914
-//	//构建上传表单对象
915
-//	formUploader := storage.NewFormUploader(&cfg)
916
-//	ret := storage.PutRet{}
917
-//	// 可选
918
-//	putExtra := storage.PutExtra{
919
-//		Params: map[string]string{},
920
-//	}
921
-//	err := formUploader.PutFile(context.Background(), &ret, upToken, key, filePath, &putExtra)
922
-//	if err != nil {
923
-//		fmt.Println(err)
924
-//		return
925
-//	}
926
-//	fmt.Println(ret.Key, ret.Hash)
927
-//}
900
+func upload_qiniu(filePath string, keys string) {
901
+	key := keys
902
+	//上传凭证,关于凭证这块大家可以去看看官方文档
903
+	putPolicy := storage.PutPolicy{
904
+		Scope: beego.AppConfig.String("qiniu_bucket"),
905
+	}
906
+	mac := qbox.NewMac(beego.AppConfig.String("qiniu_accesskey"), beego.AppConfig.String("qiniu_secretkey"))
907
+	upToken := putPolicy.UploadToken(mac)
908
+	cfg := storage.Config{}
909
+	//空间对应机房
910
+	//其中关于Zone对象和机房的关系如下:
911
+	//    机房    Zone对象
912
+	//    华东    storage.ZoneHuadong
913
+	//    华北    storage.ZoneHuabei
914
+	//    华南    storage.ZoneHuanan
915
+	//    北美    storage.ZoneBeimei
916
+	//七牛云存储空间设置首页有存储区域
917
+	cfg.Zone = &storage.ZoneHuadong
918
+	//不启用HTTPS域名
919
+	cfg.UseHTTPS = false
920
+	//不使用CND加速
921
+	cfg.UseCdnDomains = false
922
+	//构建上传表单对象
923
+	formUploader := storage.NewFormUploader(&cfg)
924
+	ret := storage.PutRet{}
925
+	// 可选
926
+	putExtra := storage.PutExtra{
927
+		Params: map[string]string{},
928
+	}
929
+	err := formUploader.PutFile(context.Background(), &ret, upToken, key, filePath, &putExtra)
930
+	if err != nil {
931
+		fmt.Println(err)
932
+		return
933
+	}
934
+	fmt.Println(ret.Key, ret.Hash)
935
+}