mirror of
https://github.com/kubeshark/kubeshark.git
synced 2025-09-29 14:26:18 +00:00
* no message * infinite scroll + new ws implementation * no message * scrolling top * fetch button * more Backend changes * fix go mod and sum * mire fixes against develop * unused code * small ui refactor Co-authored-by: Roee Gadot <roee.gadot@up9.com>
95 lines
2.1 KiB
Go
95 lines
2.1 KiB
Go
package controllers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/google/martian/har"
|
|
"mizuserver/pkg/database"
|
|
"mizuserver/pkg/models"
|
|
"mizuserver/pkg/utils"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
const (
|
|
HardLimit = 200
|
|
)
|
|
|
|
func GetEntries(c *fiber.Ctx) error {
|
|
limit, e := strconv.Atoi(c.Query("limit", "200"))
|
|
utils.CheckErr(e)
|
|
if limit > HardLimit {
|
|
limit = HardLimit
|
|
}
|
|
|
|
sortOption := c.Query("operator", "lt")
|
|
var sortingOperator string
|
|
var ordering string
|
|
if strings.ToLower(sortOption) == "gt" {
|
|
sortingOperator = ">"
|
|
ordering = "asc"
|
|
} else if strings.ToLower(sortOption) == "lt" {
|
|
sortingOperator = "<"
|
|
ordering = "desc"
|
|
} else {
|
|
fmt.Println("Unsupported")
|
|
return nil
|
|
}
|
|
|
|
timestamp, e := strconv.Atoi(c.Query("timestamp", "-1"))
|
|
utils.CheckErr(e)
|
|
|
|
var entries []models.MizuEntry
|
|
|
|
database.GetEntriesTable().
|
|
Order(fmt.Sprintf("timestamp %s", ordering)).
|
|
Where(fmt.Sprintf("timestamp %s %v",sortingOperator, timestamp)).
|
|
Omit("entry"). // remove the "big" entry field
|
|
Limit(limit).
|
|
Find(&entries)
|
|
|
|
if len(entries) > 0 && ordering == "desc"{
|
|
utils.ReverseSlice(entries)
|
|
}
|
|
|
|
// Convert to base entries
|
|
baseEntries := make([]models.BaseEntryDetails, 0, limit)
|
|
for _, entry := range entries {
|
|
baseEntries = append(baseEntries, models.BaseEntryDetails{
|
|
Id: entry.EntryId,
|
|
Url: entry.Url,
|
|
Service: entry.Service,
|
|
Path: entry.Path,
|
|
StatusCode: entry.Status,
|
|
Method: entry.Method,
|
|
Timestamp: entry.Timestamp,
|
|
})
|
|
}
|
|
|
|
return c.Status(fiber.StatusOK).JSON(baseEntries)
|
|
}
|
|
|
|
func GetEntry(c *fiber.Ctx) error {
|
|
var entryData models.EntryData
|
|
database.GetEntriesTable().
|
|
Select("entry").
|
|
Where(map[string]string{"entryId": c.Params("entryId")}).
|
|
First(&entryData)
|
|
|
|
var fullEntry har.Entry
|
|
unmarshallErr := json.Unmarshal([]byte(entryData.Entry), &fullEntry)
|
|
utils.CheckErr(unmarshallErr)
|
|
|
|
return c.Status(fiber.StatusOK).JSON(fullEntry)
|
|
}
|
|
|
|
func DeleteAllEntries(c *fiber.Ctx) error {
|
|
database.GetEntriesTable().
|
|
Where("1 = 1").
|
|
Delete(&models.MizuEntry{})
|
|
|
|
return c.Status(fiber.StatusOK).JSON(fiber.Map{
|
|
"msg": "Success",
|
|
})
|
|
} |