陆恒 / new_catcafe

Download as
Browse Code »

Commit 8a8b9edfe3e87f5e44388423684d684727e2082f

Authored by 陆恒
1 parent f95de518
Exists in master

提交

Showing 2 changed files with 30 additions and 22 deletions   Show diff stats
src/HttpServer/logic/httpserver.go
  Diff comments   View file @8a8b9ed
... ... @@ -5,8 +5,10 @@ import (
5 5 "HttpServer/redishandler"
6 6 "common/logger"
7 7 "fmt"
  8 + "io"
8 9 "io/ioutil"
9 10 "mysql"
  11 + "os"
10 12 "strconv"
11 13  
12 14 //"log"
... ... @@ -72,12 +74,36 @@ func startServerHttpServe() {
72 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 85 func QueryAllAccount(w http.ResponseWriter, r *http.Request) {
76 86  
77 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
  Diff comments   View file @8a8b9ed
... ... @@ -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 97 for i:=0;i<10;i++ {
116 98 tablename := "b_user_ext_0" + strconv.Itoa(i)
... ...