From 2d3895d3c0e39f024de9927dd707b451a3393155 Mon Sep 17 00:00:00 2001
From: tohutohu <tohu.soy@gmail.com>
Date: Thu, 23 Nov 2023 10:06:50 +0000
Subject: [PATCH 1/3] =?UTF-8?q?feat:=20=E5=90=84=E3=83=9A=E3=83=BC?=
 =?UTF-8?q?=E3=82=B8=E3=81=AB=E3=83=AD=E3=82=B0=E3=82=92=E3=83=80=E3=82=A6?=
 =?UTF-8?q?=E3=83=B3=E3=83=AD=E3=83=BC=E3=83=89=E3=81=A7=E3=81=8D=E3=82=8B?=
 =?UTF-8?q?=E3=83=AA=E3=83=B3=E3=82=AF=E3=82=92=E8=BF=BD=E5=8A=A0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/extproc/handler.go          | 42 ++++++++++++++++++++++++++++
 internal/pprof/handler.go            | 42 ++++++++++++++++++++++++++++
 view/src/components/HttpLogEntry.vue |  2 +-
 view/src/components/SlowLogEntry.vue |  2 +-
 view/src/components/TsvTable.vue     |  4 +++
 5 files changed, 90 insertions(+), 2 deletions(-)

diff --git a/internal/extproc/handler.go b/internal/extproc/handler.go
index 9d96389..7c01ff0 100644
--- a/internal/extproc/handler.go
+++ b/internal/extproc/handler.go
@@ -3,6 +3,7 @@ package extproc
 import (
 	"fmt"
 	"net/http"
+	"slices"
 
 	"github.com/kaz/pprotein/internal/collect"
 	"github.com/labstack/echo/v4"
@@ -34,6 +35,8 @@ func (h *handler) Register(g *echo.Group) error {
 	g.GET("", h.getIndex)
 	g.POST("", h.postIndex)
 	g.GET("/:id", h.getId)
+	g.GET("/data/:id", h.getData)
+	g.GET("/data/latest", h.getLatestData)
 
 	return nil
 }
@@ -66,3 +69,42 @@ func (h *handler) getId(c echo.Context) error {
 
 	return c.Stream(http.StatusOK, "application/json", r)
 }
+
+func (h *handler) getData(c echo.Context) error {
+	id := c.Param("id")
+	entries := h.collector.List()
+	for _, entry := range entries {
+		if entry.Snapshot.ID == id {
+			bodyPath, err := entry.Snapshot.BodyPath()
+			if err != nil {
+				return echo.NewHTTPError(http.StatusInternalServerError, fmt.Sprintf("failed to get body path: %v", err))
+			}
+
+			return c.File(bodyPath)
+		}
+	}
+	return echo.NewHTTPError(http.StatusNotFound)
+}
+
+func (h *handler) getLatestData(c echo.Context) error {
+	label := c.QueryParam("label")
+
+	entries := h.collector.List()
+	slices.SortFunc(entries, func(a, b *collect.Entry) int {
+		return b.Snapshot.SnapshotMeta.Datetime.Compare(a.Snapshot.SnapshotMeta.Datetime)
+	})
+
+	if len(entries) > 0 {
+		for _, entry := range entries {
+			if label != "" && label != entry.Snapshot.SnapshotTarget.Label {
+				continue
+			}
+			bodyPath, err := entry.Snapshot.BodyPath()
+			if err != nil {
+				return echo.NewHTTPError(http.StatusInternalServerError, fmt.Sprintf("failed to get body path: %v", err))
+			}
+			return c.File(bodyPath)
+		}
+	}
+	return echo.NewHTTPError(http.StatusNotFound)
+}
diff --git a/internal/pprof/handler.go b/internal/pprof/handler.go
index ec58aec..4257764 100644
--- a/internal/pprof/handler.go
+++ b/internal/pprof/handler.go
@@ -3,6 +3,7 @@ package pprof
 import (
 	"fmt"
 	"net/http"
+	"slices"
 	"sync"
 
 	"github.com/kaz/pprotein/internal/collect"
@@ -32,6 +33,8 @@ func (h *handler) Register(g *echo.Group) error {
 
 	g.GET("", h.getIndex)
 	g.POST("", h.postIndex)
+	g.GET("/data/:id", h.getData)
+	g.GET("/data/latest", h.getLatestData)
 
 	return nil
 }
@@ -54,3 +57,42 @@ func (h *handler) postIndex(c echo.Context) error {
 
 	return c.NoContent(http.StatusOK)
 }
+
+func (h *handler) getData(c echo.Context) error {
+	id := c.Param("id")
+	entries := h.collector.List()
+	for _, entry := range entries {
+		if entry.Snapshot.ID == id {
+			bodyPath, err := entry.Snapshot.BodyPath()
+			if err != nil {
+				return echo.NewHTTPError(http.StatusInternalServerError, fmt.Sprintf("failed to get body path: %v", err))
+			}
+
+			return c.File(bodyPath)
+		}
+	}
+	return echo.NewHTTPError(http.StatusNotFound)
+}
+
+func (h *handler) getLatestData(c echo.Context) error {
+	label := c.QueryParam("label")
+
+	entries := h.collector.List()
+	slices.SortFunc(entries, func(a, b *collect.Entry) int {
+		return b.Snapshot.SnapshotMeta.Datetime.Compare(a.Snapshot.SnapshotMeta.Datetime)
+	})
+
+	if len(entries) > 0 {
+		for _, entry := range entries {
+			if label != "" && label != entry.Snapshot.SnapshotTarget.Label {
+				continue
+			}
+			bodyPath, err := entry.Snapshot.BodyPath()
+			if err != nil {
+				return echo.NewHTTPError(http.StatusInternalServerError, fmt.Sprintf("failed to get body path: %v", err))
+			}
+			return c.File(bodyPath)
+		}
+	}
+	return echo.NewHTTPError(http.StatusNotFound)
+}
diff --git a/view/src/components/HttpLogEntry.vue b/view/src/components/HttpLogEntry.vue
index 2e0041e..ed49c89 100644
--- a/view/src/components/HttpLogEntry.vue
+++ b/view/src/components/HttpLogEntry.vue
@@ -1,5 +1,5 @@
 <template>
-  <TsvTable :tsv="tsv" />
+  <TsvTable :tsv="tsv" :link="`/api/slowlog/data/${$route.params.id}`"/>
 </template>
 
 <script lang="ts">
diff --git a/view/src/components/SlowLogEntry.vue b/view/src/components/SlowLogEntry.vue
index faf4d12..d365ca4 100644
--- a/view/src/components/SlowLogEntry.vue
+++ b/view/src/components/SlowLogEntry.vue
@@ -1,5 +1,5 @@
 <template>
-  <TsvTable :tsv="tsv" />
+  <TsvTable :tsv="tsv" :link="`/api/slowlog/data/${$route.params.id}`"/>
 </template>
 
 <script lang="ts">
diff --git a/view/src/components/TsvTable.vue b/view/src/components/TsvTable.vue
index b54d764..10a60e0 100644
--- a/view/src/components/TsvTable.vue
+++ b/view/src/components/TsvTable.vue
@@ -1,5 +1,6 @@
 <template>
   <section>
+    <a :href="link" download>Download</a>
     <table border="1">
       <thead>
         <tr>
@@ -39,6 +40,9 @@ export default defineComponent({
       type: String,
       required: true,
     },
+    link: {
+      type: String
+    }
   },
   data() {
     return {

From 5b83d1dd2232a9f62cab8c15abcc0aac44fde259 Mon Sep 17 00:00:00 2001
From: tohutohu <tohu.soy@gmail.com>
Date: Fri, 24 Nov 2023 03:10:33 +0000
Subject: [PATCH 2/3] =?UTF-8?q?fix:=20Status=20OK=E3=81=A7=E3=81=AF?=
 =?UTF-8?q?=E3=81=AA=E3=81=84=E3=82=82=E3=81=AE=E3=81=AF=E7=84=A1=E8=A6=96?=
 =?UTF-8?q?=E3=81=99=E3=82=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/extproc/handler.go | 4 ++++
 internal/pprof/handler.go   | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/internal/extproc/handler.go b/internal/extproc/handler.go
index 7c01ff0..1263588 100644
--- a/internal/extproc/handler.go
+++ b/internal/extproc/handler.go
@@ -99,6 +99,10 @@ func (h *handler) getLatestData(c echo.Context) error {
 			if label != "" && label != entry.Snapshot.SnapshotTarget.Label {
 				continue
 			}
+			if entry.Status != collect.StatusOk {
+				continue
+			}
+
 			bodyPath, err := entry.Snapshot.BodyPath()
 			if err != nil {
 				return echo.NewHTTPError(http.StatusInternalServerError, fmt.Sprintf("failed to get body path: %v", err))
diff --git a/internal/pprof/handler.go b/internal/pprof/handler.go
index 4257764..211ca64 100644
--- a/internal/pprof/handler.go
+++ b/internal/pprof/handler.go
@@ -87,6 +87,10 @@ func (h *handler) getLatestData(c echo.Context) error {
 			if label != "" && label != entry.Snapshot.SnapshotTarget.Label {
 				continue
 			}
+			if entry.Status != collect.StatusOk {
+				continue
+			}
+
 			bodyPath, err := entry.Snapshot.BodyPath()
 			if err != nil {
 				return echo.NewHTTPError(http.StatusInternalServerError, fmt.Sprintf("failed to get body path: %v", err))

From 5880a86a15a3cdcd463a726978d85c95c4f42a59 Mon Sep 17 00:00:00 2001
From: tohutohu <tohu.soy@gmail.com>
Date: Fri, 24 Nov 2023 03:11:40 +0000
Subject: [PATCH 3/3] =?UTF-8?q?fix:=20=E3=83=91=E3=82=B9=E5=90=8D=E3=81=8C?=
 =?UTF-8?q?=E8=AA=A4=E3=81=A3=E3=81=A6=E3=81=84=E3=81=9F=E3=81=AE=E3=82=92?=
 =?UTF-8?q?=E4=BF=AE=E6=AD=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 view/src/components/HttpLogEntry.vue | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/view/src/components/HttpLogEntry.vue b/view/src/components/HttpLogEntry.vue
index ed49c89..d2f610e 100644
--- a/view/src/components/HttpLogEntry.vue
+++ b/view/src/components/HttpLogEntry.vue
@@ -1,5 +1,5 @@
 <template>
-  <TsvTable :tsv="tsv" :link="`/api/slowlog/data/${$route.params.id}`"/>
+  <TsvTable :tsv="tsv" :link="`/api/httplog/data/${$route.params.id}`"/>
 </template>
 
 <script lang="ts">