package api import ( "database/sql" "net/http" "strconv" "strings" ) type ResellersAPI struct { resellerDB *sql.DB } func ResellersHandler(resellerDB *sql.DB) http.Handler { api := &ResellersAPI{resellerDB: resellerDB} return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { p := strings.TrimPrefix(r.URL.Path, "/api/resellers") if r.Method == http.MethodGet && p == "" { api.list(w, r) return } w.WriteHeader(http.StatusNotFound) }) } func (a *ResellersAPI) list(w http.ResponseWriter, r *http.Request) { q := r.URL.Query().Get("q") limitStr := r.URL.Query().Get("limit") limit := 2000 if limitStr != "" { if n, err := strconv.Atoi(limitStr); err == nil && n > 0 && n <= 10000 { limit = n } } sql1 := "SELECT id, COALESCE(name,'') AS name FROM reseller WHERE 1=1" args := []interface{}{} if q != "" { sql1 += " AND (CAST(id AS CHAR) LIKE ? OR name LIKE ?)" like := "%" + q + "%" args = append(args, like, like) } sql1 += " ORDER BY id ASC LIMIT ?" args = append(args, limit) rows, err := a.resellerDB.Query(sql1, args...) if err != nil { fail(w, r, http.StatusInternalServerError, err.Error()) return } defer rows.Close() out := []map[string]interface{}{} for rows.Next() { var id sql.NullInt64 var name sql.NullString if err := rows.Scan(&id, &name); err != nil { continue } if !id.Valid { continue } m := map[string]interface{}{"id": id.Int64, "name": name.String} out = append(out, m) } ok(w, r, out) }