getAnswerInformatioという関数を生やした

Author: proelbtn

vmdb-api/controller.go

@@ -7,6 +7,7 @@ import (
 	"strings"
 	"time"
 
+	"github.com/go-chi/chi/v5"
 	"github.com/google/uuid"
 )
 
@@ -140,7 +141,7 @@ func (c *Controller) listLatestUnconfirmedAnswersForLocalProblem(w http.Response
 	for _, code := range strings.Split(value, ",") {
 		code = strings.TrimSpace(code)
 
-		problem, err := c.repo.findProblemBy(ctx, code)
+		problem, err := c.repo.findProblemByCode(ctx, code)
 		if err != nil {
 			slog.WarnContext(ctx, "failed to find Problem", "error", err, "code", code)
 			continue
@@ -164,20 +165,7 @@ func (c *Controller) listLatestUnconfirmedAnswersForLocalProblem(w http.Response
 		}
 
 		for _, answer := range latestAnswers {
-			bodies := []string{}
-			for _, b := range answer.Bodies {
-				bodies = append(bodies, b...)
-			}
-
-			response = append(response, listLatestUnconfirmedAnswersForLocalProblemResponseItem{
-				ID:          answer.ID,
-				ProblemID:   answer.ProblemID,
-				ProblemCode: problem.Code,
-				TeamID:      answer.TeamID,
-				CreatedAt:   answer.CreatedAt,
-				UpdatedAt:   answer.UpdatedAt,
-				Body:        strings.Join(bodies, "\n"),
-			})
+			response = append(response, newAnswerResponseFrom(answer, *problem))
 		}
 	}
 
@@ -186,3 +174,38 @@ func (c *Controller) listLatestUnconfirmedAnswersForLocalProblem(w http.Response
 		w.WriteHeader(http.StatusInternalServerError)
 	}
 }
+
+type getAnswerInformationResponse answerResponse
+
+func (c *Controller) getAnswerInformation(w http.ResponseWriter, r *http.Request) {
+	ctx := r.Context()
+
+	answerIDStr := chi.URLParam(r, "answerID")
+	answerID, err := uuid.Parse(answerIDStr)
+	if err != nil {
+		slog.WarnContext(ctx, "invalid query parameters", "answer_id", answerIDStr)
+		w.WriteHeader(http.StatusBadRequest)
+		return
+	}
+
+	answer, err := c.repo.findAnswerBy(ctx, answerID)
+	if err != nil {
+		slog.WarnContext(ctx, "failed to find Answer", "error", err)
+		w.WriteHeader(http.StatusNotFound)
+		return
+	}
+
+	problem, err := c.repo.findProblemBy(ctx, answer.ProblemID)
+	if err != nil {
+		slog.ErrorContext(ctx, "failed to find Problem", "error", err)
+		w.WriteHeader(http.StatusInternalServerError)
+		return
+	}
+
+	response := getAnswerInformationResponse(newAnswerResponseFrom(*answer, *problem))
+
+	if err := renderJSON(w, http.StatusOK, response); err != nil {
+		slog.ErrorContext(ctx, "failed to render JSON", "error", err)
+		w.WriteHeader(http.StatusInternalServerError)
+	}
+}

vmdb-api/main.go

@@ -75,10 +75,13 @@ func (c *Command) RunE(cmd *cobra.Command, _ []string) error {
 	r.Use(middleware.Heartbeat("/healthz"))
 	r.Use(middleware.Recoverer)
 
+	// 後方互換性のために、そのままのI/Fで提供する
 	r.Get("/", controller.hello)
 	r.Get("/problem-environments", controller.listProblemEnvironments)
 	r.Get("/answer-id", controller.getAnswerID)
+
 	r.Get("/local-problem-answers", controller.listLatestUnconfirmedAnswersForLocalProblem)
+	r.Get("/answers/{answerID}", controller.getAnswerInformation)
 
 	server := Server{
 		ListenAddr: c.listenAddr,

vmdb-api/repository.go

@@ -50,7 +50,18 @@ func (r *Repository) findProblemEnvironmentBy(ctx context.Context, name string)
 	return &result, nil
 }
 
-func (r *Repository) findProblemBy(ctx context.Context, code string) (*Problem, error) {
+func (r *Repository) findProblemBy(ctx context.Context, problemID uuid.UUID) (*Problem, error) {
+	var result Problem
+	err := r.db.NewSelect().Model(&result).
+		Where("id = ?", problemID).
+		Scan(ctx)
+	if err != nil {
+		return nil, err
+	}
+	return &result, nil
+}
+
+func (r *Repository) findProblemByCode(ctx context.Context, code string) (*Problem, error) {
 	var result Problem
 	err := r.db.NewSelect().Model(&result).
 		Where("code = ?", code).
@@ -61,6 +72,17 @@ func (r *Repository) findProblemBy(ctx context.Context, code string) (*Problem,
 	return &result, nil
 }
 
+func (r *Repository) findAnswerBy(ctx context.Context, answerID uuid.UUID) (*Answer, error) {
+	var result Answer
+	err := r.db.NewSelect().Model(&result).
+		Where("id = ?", answerID).
+		Scan(ctx)
+	if err != nil {
+		return nil, err
+	}
+	return &result, nil
+}
+
 func (r *Repository) listLatestUnscoredAnswersFor(ctx context.Context, problemID uuid.UUID) ([]Answer, error) {
 	var result []Answer
 	err := r.db.NewSelect().ColumnExpr("answers.*").