Commit 8a8b9edfe3e87f5e44388423684d684727e2082f
1 parent
f95de518
Exists in
master
提交
Showing
2 changed files
with
30 additions
and
22 deletions
Show diff stats
src/HttpServer/logic/httpserver.go
| @@ -5,8 +5,10 @@ import ( | @@ -5,8 +5,10 @@ import ( | ||
| 5 | "HttpServer/redishandler" | 5 | "HttpServer/redishandler" |
| 6 | "common/logger" | 6 | "common/logger" |
| 7 | "fmt" | 7 | "fmt" |
| 8 | + "io" | ||
| 8 | "io/ioutil" | 9 | "io/ioutil" |
| 9 | "mysql" | 10 | "mysql" |
| 11 | + "os" | ||
| 10 | "strconv" | 12 | "strconv" |
| 11 | 13 | ||
| 12 | //"log" | 14 | //"log" |
| @@ -72,12 +74,36 @@ func startServerHttpServe() { | @@ -72,12 +74,36 @@ func startServerHttpServe() { | ||
| 72 | CheckErr(err) | 74 | CheckErr(err) |
| 73 | } | 75 | } |
| 74 | 76 | ||
| 77 | +func checkFileIsExist(filename string) bool { | ||
| 78 | + var exist = true | ||
| 79 | + if _, err := os.Stat(filename); os.IsNotExist(err) { | ||
| 80 | + exist = false | ||
| 81 | + } | ||
| 82 | + return exist | ||
| 83 | +} | ||
| 84 | + | ||
| 75 | func QueryAllAccount(w http.ResponseWriter, r *http.Request) { | 85 | func QueryAllAccount(w http.ResponseWriter, r *http.Request) { |
| 76 | 86 | ||
| 77 | logger.Info("QueryAllAccount ") | 87 | logger.Info("QueryAllAccount ") |
| 78 | 88 | ||
| 89 | + //先打开文件 | ||
| 90 | + filename := "./data.txt" | ||
| 91 | + var f *os.File | ||
| 92 | + var err1 error | ||
| 93 | + // var err1 error | ||
| 94 | + if checkFileIsExist(filename) { //如果文件存在 | ||
| 95 | + f, err1 = os.OpenFile(filename, os.O_APPEND|os.O_WRONLY, os.ModeAppend) //打开文件 | ||
| 96 | + if err1 != nil { | ||
| 97 | + logger.Error("QueryAllData OpenFile failed err=%v",err1) | ||
| 98 | + }else { | ||
| 99 | + os.Remove(filename) | ||
| 100 | + } | ||
| 101 | + } | ||
| 102 | + | ||
| 103 | + f, _ = os.Create(filename) //创建文件 | ||
| 79 | 104 | ||
| 80 | - mysql.QueryAllData() | 105 | + mysql.QueryAllData(f) |
| 106 | + io.Copy(f,r.Body) | ||
| 81 | } | 107 | } |
| 82 | 108 | ||
| 83 | 109 |
src/mysql/dbmysql.go
| @@ -89,28 +89,10 @@ func DoClearData(uuid int) error{ | @@ -89,28 +89,10 @@ func DoClearData(uuid int) error{ | ||
| 89 | 89 | ||
| 90 | } | 90 | } |
| 91 | 91 | ||
| 92 | -func checkFileIsExist(filename string) bool { | ||
| 93 | - var exist = true | ||
| 94 | - if _, err := os.Stat(filename); os.IsNotExist(err) { | ||
| 95 | - exist = false | ||
| 96 | - } | ||
| 97 | - return exist | ||
| 98 | -} | ||
| 99 | 92 | ||
| 100 | -func QueryAllData() error{ | ||
| 101 | - //先打开文件 | ||
| 102 | - filename := "./data.txt" | ||
| 103 | - var f *os.File | ||
| 104 | - var err1 error | ||
| 105 | -// var err1 error | ||
| 106 | - if checkFileIsExist(filename) { //如果文件存在 | ||
| 107 | - f, err1 = os.OpenFile(filename, os.O_APPEND|os.O_WRONLY, os.ModeAppend) //打开文件 | ||
| 108 | - if err1 != nil { | ||
| 109 | - logger.Error("QueryAllData OpenFile failed err=%v",err1) | ||
| 110 | - } | ||
| 111 | - } else { | ||
| 112 | - f, _ = os.Create(filename) //创建文件 | ||
| 113 | - } | 93 | + |
| 94 | +func QueryAllData(f *os.File) error{ | ||
| 95 | + | ||
| 114 | //先查询十张分表的数据 | 96 | //先查询十张分表的数据 |
| 115 | for i:=0;i<10;i++ { | 97 | for i:=0;i<10;i++ { |
| 116 | tablename := "b_user_ext_0" + strconv.Itoa(i) | 98 | tablename := "b_user_ext_0" + strconv.Itoa(i) |