Add codelab contents taskmanager

taskmanager/function.go

@@ -33,12 +33,61 @@ func TaskManager(w http.ResponseWriter, r *http.Request) {
 
 	// 一覧取得
 	case http.MethodGet:
+		t, err := getAllTask()
+		if err != nil {
+			e := errors.Errorf("get error: %v", err)
+			responseWrite(w, http.StatusInternalServerError, e.Error(), e)
+			return
+		}
+
+		if len(t) < 1 {
+			responseWrite(w, http.StatusOK, "0 tickets", nil)
+			return
+		}
+
+		b, err := json.Marshal(t)
+		if err != nil {
+			e := errors.Errorf("json marshal error: %v", err)
+			responseWrite(w, http.StatusInternalServerError, e.Error(), e)
+			return
+		}
+
+		w.WriteHeader(http.StatusOK)
+		w.Write(b)
 
 	// ステータス変更
 	case http.MethodPatch:
+		param, code, err := getJSON(r.Header.Get("Content-Type"), r.Body)
+		if err != nil {
+			responseWrite(w, code, err.Error(), err)
+			return
+		}
+
+		t := setTask(param.ID, param.Title, param.Status)
+		if err := t.update(); err != nil {
+			e := errors.Errorf("patch error: %v", err)
+			responseWrite(w, http.StatusInternalServerError, e.Error(), e)
+			return
+		}
+		msg := fmt.Sprintf("%v updated", t)
+		responseWrite(w, http.StatusOK, msg, nil)
 
 	// 削除
 	case http.MethodDelete:
+		param, code, err := getJSON(r.Header.Get("Content-Type"), r.Body)
+		if err != nil {
+			responseWrite(w, code, err.Error(), err)
+			return
+		}
+
+		t := setTask(param.ID, "", 0)
+		if err := t.delete(); err != nil {
+			e := errors.Errorf("delete error: %v", err)
+			responseWrite(w, http.StatusInternalServerError, e.Error(), e)
+			return
+		}
+		msg := fmt.Sprintf("%v deleted", t)
+		responseWrite(w, http.StatusOK, msg, nil)
 
 	default:
 		e := errors.New("method not allowed")

taskmanager/task.go

@@ -10,10 +10,10 @@ import (
 
 // Task is the structure of the task.
 type Task struct {
-	ID        int64     `datastore:"-"`
-	Title     string    `datastore:"title"`
-	Status    Status    `datastore:"status"`
-	CreatedAt time.Time `datastore:"createdAt"`
+	ID        int64     `datastore:"-" json:"ID"`
+	Title     string    `datastore:"title" json:"title"`
+	Status    Status    `datastore:"status" json:"status"`
+	CreatedAt time.Time `datastore:"createdAt" json:"createdAt"`
 }
 
 func newTask(title string) *Task {
@@ -36,3 +36,78 @@ func (t Task) add() error {
 	_, err = client.Put(ctx, newKey, &t)
 	return err
 }
+
+func getAllTask() ([]Task, error) {
+	ctx := context.Background()
+
+	client, err := datastore.NewClient(ctx, "wwgt-codelabs")
+	if err != nil {
+		return nil, err
+	}
+
+	var t []Task
+
+	q := datastore.NewQuery(os.Getenv("MY_CODE"))
+	keys, err := client.GetAll(ctx, q, &t)
+	if err != nil {
+		return nil, err
+	}
+
+	for i, key := range keys {
+		t[i].ID = key.ID
+	}
+	return t, nil
+}
+
+func setTask(id int64, title string, status Status) *Task {
+	return &Task{
+		ID:        id,
+		Title:     title,
+		Status:    status,
+		CreatedAt: time.Now(),
+	}
+}
+
+func (t Task) update() error {
+	ctx := context.Background()
+
+	client, err := datastore.NewClient(ctx, "wwgt-codelabs")
+	if err != nil {
+		return err
+	}
+
+	key := datastore.IDKey(os.Getenv("MY_CODE"), t.ID, nil)
+
+	_, err = client.RunInTransaction(ctx, func(tx *datastore.Transaction) error {
+		var gt Task
+
+		if err := tx.Get(key, &gt); err != nil {
+			return err
+		}
+
+		if t.Title == "" {
+			t.Title = gt.Title
+		}
+
+		if err := t.Status.validate(); err != nil {
+			t.Status = gt.Status
+		}
+
+		t.CreatedAt = gt.CreatedAt
+
+		_, err := tx.Put(key, &t)
+		return err
+	})
+	return err
+}
+
+func (t Task) delete() error {
+	ctx := context.Background()
+
+	client, err := datastore.NewClient(ctx, "wwgt-codelabs")
+	if err != nil {
+		return err
+	}
+
+	return client.Delete(ctx, datastore.IDKey(os.Getenv("MY_CODE"), t.ID, nil))
+}