|
| 1 | +package main |
| 2 | + |
| 3 | +import ( |
| 4 | + "bytes" |
| 5 | + "encoding/json" |
| 6 | + _ "github.com/go-sql-driver/mysql" |
| 7 | + "github.com/jmoiron/sqlx" |
| 8 | + "io/ioutil" |
| 9 | + "log" |
| 10 | + "net/http" |
| 11 | + "strings" |
| 12 | +) |
| 13 | + |
| 14 | +var db *sqlx.DB |
| 15 | + |
| 16 | +func main() { |
| 17 | + http.HandleFunc("/get", GetHandler) |
| 18 | + http.ListenAndServe("127.0.0.1:8000", nil) |
| 19 | +} |
| 20 | + |
| 21 | +func init() { |
| 22 | + database, err := sqlx.Open("mysql", "apijson:1234qqqq@tcp(y.tadev.cn:53306)/sys") |
| 23 | + if err != nil { |
| 24 | + log.Fatal("db connect error", err) |
| 25 | + } |
| 26 | + db = database |
| 27 | +} |
| 28 | + |
| 29 | +func GetHandler(w http.ResponseWriter, r *http.Request) { |
| 30 | + if data, err := ioutil.ReadAll(r.Body); err != nil { |
| 31 | + log.Println("read request body error", err) |
| 32 | + w.WriteHeader(http.StatusBadRequest) |
| 33 | + } else { |
| 34 | + var bodyMap map[string]interface{} |
| 35 | + if err := json.Unmarshal(data, &bodyMap); err != nil { |
| 36 | + log.Println("parse request body json error", err) |
| 37 | + w.WriteHeader(http.StatusBadRequest) |
| 38 | + return |
| 39 | + } |
| 40 | + respMap := make(map[string]interface{}) |
| 41 | + for table, fields := range bodyMap { |
| 42 | + if fields != nil { |
| 43 | + respMap[table] = QueryTable(table, fields) |
| 44 | + } |
| 45 | + log.Println("get:query table: ", table, ", fields: ", fields) |
| 46 | + } |
| 47 | + if respBody, err := json.Marshal(respMap); err != nil { |
| 48 | + w.WriteHeader(http.StatusInternalServerError) |
| 49 | + } else { |
| 50 | + w.WriteHeader(http.StatusOK) |
| 51 | + w.Write(respBody) |
| 52 | + } |
| 53 | + } |
| 54 | +} |
| 55 | + |
| 56 | +func QueryTable(table string, fields interface{}) interface{} { |
| 57 | + var buffer bytes.Buffer |
| 58 | + buffer.WriteString("select * from ") |
| 59 | + buffer.WriteString(table) |
| 60 | + buffer.WriteString(" where ") |
| 61 | + if fieldMap, ok := fields.(map[string]interface{}); !ok { |
| 62 | + return "fields error, only support object." |
| 63 | + } else { |
| 64 | + size := len(fieldMap) |
| 65 | + cols := make([]string, size) |
| 66 | + values := make([]interface{}, size) |
| 67 | + i := 0 |
| 68 | + for col, value := range fieldMap { |
| 69 | + if value == nil { |
| 70 | + return "field value error, " + col + " is nil" |
| 71 | + } |
| 72 | + cols[i] = col + "=?" |
| 73 | + values[i] = value |
| 74 | + } |
| 75 | + buffer.WriteString(strings.Join(cols, " and ")) |
| 76 | + sql := buffer.String() |
| 77 | + if rows, err := db.Query(sql, values...); err != nil { |
| 78 | + return err.Error() |
| 79 | + } else { |
| 80 | + if rows.Next() { |
| 81 | + if columns, err := rows.Columns(); err != nil { |
| 82 | + return "get rows error: " + err.Error() |
| 83 | + } else { |
| 84 | + values := make([]interface{}, len(columns)) |
| 85 | + for k := range columns { |
| 86 | + str := "" |
| 87 | + values[k] = &str |
| 88 | + } |
| 89 | + err = rows.Scan(values...) |
| 90 | + if err != nil { |
| 91 | + return "rows.Scan error: " + err.Error() |
| 92 | + } |
| 93 | + resultMap := make(map[string]interface{}) |
| 94 | + for k, colName := range columns { |
| 95 | + resultMap[colName] = values[k] |
| 96 | + } |
| 97 | + return resultMap |
| 98 | + } |
| 99 | + } else { |
| 100 | + return "" |
| 101 | + } |
| 102 | + } |
| 103 | + } |
| 104 | +} |
0 commit comments