diff --git a/precode.go b/precode.go index d082d83..2ffd8c7 100644 --- a/precode.go +++ b/precode.go @@ -1,6 +1,8 @@ package main import ( + "bytes" + "encoding/json" "fmt" "net/http" @@ -42,11 +44,89 @@ var tasks = map[string]Task{ // Ниже напишите обработчики для каждого эндпоинта // ... +// Обработчик для получения всех задач +func getTasks(w http.ResponseWriter, r *http.Request) { + resp, err := json.MarshalIndent(tasks, "", " ") + //resp, err := json.Marshal(tasks) + if err != nil { + http.Error(w, err.Error(), http.StatusInternalServerError) + return + } + w.Header().Set("Counter-Type", "application/json") + w.WriteHeader(http.StatusOK) + w.Write(resp) +} + +// Обработчик для добавления нового элемента +func postTasks(w http.ResponseWriter, r *http.Request) { + var task Task + var buf bytes.Buffer + + _, err := buf.ReadFrom(r.Body) + if err != nil { + http.Error(w, err.Error(), http.StatusInternalServerError) + return + } + + if err = json.Unmarshal(buf.Bytes(), &task); err != nil { + http.Error(w, err.Error(), http.StatusBadRequest) + return + } + + tasks[task.ID] = task + + w.Header().Set("Counter-Type", "application/json") + w.WriteHeader(http.StatusCreated) +} + +// Обработчик для получения задачи по ID +func getTask(w http.ResponseWriter, r *http.Request) { + id := chi.URLParam(r, "id") + + task, ok := tasks[id] + if !ok { + http.Error(w, "Bad ID", http.StatusBadRequest) + return + } + + resp, err := json.MarshalIndent(task, "", " ") + if err != nil { + http.Error(w, err.Error(), http.StatusBadRequest) + return + } + w.Header().Set("Counter-Type", "application/json") + w.WriteHeader(http.StatusOK) + w.Write(resp) +} + +// Обработчик удаления задачи по ID +func deleteTask(w http.ResponseWriter, r *http.Request) { + id := chi.URLParam(r, "id") + + _, ok := tasks[id] + if !ok { + http.Error(w, "Bad ID", http.StatusBadRequest) + return + } + + delete(tasks, id) + + w.Header().Set("Counter-Type", "application/json") + w.WriteHeader(http.StatusOK) +} + func main() { r := chi.NewRouter() // здесь регистрируйте ваши обработчики - // ... + // метод GET, для получения всех задач, используя обработчик `getTasks` + r.Get("/tasks", getTasks) + // метод POST, для добавления нового элемента, используя обработчик `postTasks` + r.Post("/tasks", postTasks) + // метод GET, для получения задачи по ID, используя обработчик `getTasks` + r.Get("/tasks/{id}", getTask) + // метод DELETE, для удаления задачи по ID, используя обработчик `deleteTask` + r.Delete("/tasks/{id}", deleteTask) if err := http.ListenAndServe(":8080", r); err != nil { fmt.Printf("Ошибка при запуске сервера: %s", err.Error())