github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2025-06-26 08:14:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Mizu tap - adding logs for troubleshooting

Browse Source
This commit is contained in:
Igor Gov 2021-07-12 20:33:15 +03:00
parent e3049fb5a5
commit 6aa38f071f
3 changed files with 17 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
api/pkg/controllers/entries_controller.go
View File

@ -5,6 +5,7 @@ import (
"fmt" "fmt"
"github.com/gofiber/fiber/v2" "github.com/gofiber/fiber/v2"
"github.com/google/martian/har" "github.com/google/martian/har"
"github.com/romana/rlog"
"mizuserver/pkg/database" "mizuserver/pkg/database"
"mizuserver/pkg/models" "mizuserver/pkg/models"
"mizuserver/pkg/up9" "mizuserver/pkg/up9"
@ -140,6 +141,8 @@ func GetHARs(c *fiber.Ctx) error {
} }
func UploadEntries(c *fiber.Ctx) error { func UploadEntries(c *fiber.Ctx) error {
rlog.Debugf("Upload entries - started\n")
uploadRequestBody := &models.UploadEntriesRequestBody{} uploadRequestBody := &models.UploadEntriesRequestBody{}
if err := c.QueryParser(uploadRequestBody); err != nil { if err := c.QueryParser(uploadRequestBody); err != nil {
return c.Status(fiber.StatusBadRequest).JSON(err) return c.Status(fiber.StatusBadRequest).JSON(err)
@ -150,8 +153,12 @@ func UploadEntries(c *fiber.Ctx) error {
if up9.GetAnalyzeInfo().IsAnalyzing { if up9.GetAnalyzeInfo().IsAnalyzing {
return c.Status(fiber.StatusBadRequest).SendString("Cannot analyze, mizu is already analyzing") return c.Status(fiber.StatusBadRequest).SendString("Cannot analyze, mizu is already analyzing")
} }
rlog.Debugf("Upload entries - creating token. dest %s\n", uploadRequestBody.Dest)
token, _ := up9.CreateAnonymousToken(uploadRequestBody.Dest) token, err := up9.CreateAnonymousToken(uploadRequestBody.Dest)
if err != nil {
return c.Status(fiber.StatusServiceUnavailable).SendString("Can't get token")
}
rlog.Infof("Upload entries - uploading. token: %s model: %s\n", token.Token, token.Model)
go up9.UploadEntriesImpl(token.Token, token.Model, uploadRequestBody.Dest) go up9.UploadEntriesImpl(token.Token, token.Model, uploadRequestBody.Dest)
return c.Status(fiber.StatusOK).SendString("OK") return c.Status(fiber.StatusOK).SendString("OK")
} }

4
api/pkg/up9/main.go
View File

@ -36,7 +36,7 @@ func getGuestToken(url string, target *GuestToken) error {
return err return err
} }
defer resp.Body.Close() defer resp.Body.Close()
rlog.Debugf("Got token from the server, starting to json decode... status code: %v", resp.StatusCode)
return json.NewDecoder(resp.Body).Decode(target) return json.NewDecoder(resp.Body).Decode(target)
} }
@ -47,7 +47,7 @@ func CreateAnonymousToken(envPrefix string) (*GuestToken, error) {
} }
token := &GuestToken{} token := &GuestToken{}
if err := getGuestToken(tokenUrl, token); err != nil { if err := getGuestToken(tokenUrl, token); err != nil {
rlog.Infof("%s", err) rlog.Infof("Failed to get token, %s", err)
return nil, err return nil, err
} }
return token, nil return token, nil

9
cli/cmd/tapRunner.go
View File

@ -3,6 +3,7 @@ package cmd
import ( import (
"context" "context"
"fmt" "fmt"
"github.com/romana/rlog"
"github.com/up9inc/mizu/cli/debounce" "github.com/up9inc/mizu/cli/debounce"
"github.com/up9inc/mizu/cli/kubernetes" "github.com/up9inc/mizu/cli/kubernetes"
"github.com/up9inc/mizu/cli/mizu" "github.com/up9inc/mizu/cli/mizu"
@ -252,11 +253,12 @@ func portForwardApiPod(ctx context.Context, kubernetesProvider *kubernetes.Provi
time.Sleep(time.Second * 5) // Waiting to be sure the proxy is ready time.Sleep(time.Second * 5) // Waiting to be sure the proxy is ready
if tappingOptions.Analyze { if tappingOptions.Analyze {
url_path := fmt.Sprintf("http://%s/api/uploadEntries?dest=%s", mizuProxiedUrl, tappingOptions.AnalyzeDestination) url_path := fmt.Sprintf("http://%s/api/uploadEntries?dest=%s", mizuProxiedUrl, url.QueryEscape(tappingOptions.AnalyzeDestination))
u, err := url.ParseRequestURI(url_path) u, err := url.ParseRequestURI(url_path)
if err != nil { if err != nil {
log.Fatal(fmt.Sprintf("Failed parsing the URL %v\n", err)) log.Fatal(fmt.Sprintf("Failed parsing the URL %v\n", err))
} }
rlog.Debugf("Sending get request to %v\n", u.String())
if response, err := http.Get(u.String()); err != nil && response.StatusCode != 200 { if response, err := http.Get(u.String()); err != nil && response.StatusCode != 200 {
fmt.Printf("error sending upload entries req %v\n", err) fmt.Printf("error sending upload entries req %v\n", err)
} else { } else {
@ -324,7 +326,8 @@ func waitForFinish(ctx context.Context, cancel context.CancelFunc) {
} }
func syncApiStatus(ctx context.Context, cancel context.CancelFunc, tappingOptions *MizuTapOptions) { func syncApiStatus(ctx context.Context, cancel context.CancelFunc, tappingOptions *MizuTapOptions) {
controlSocket, err := mizu.CreateControlSocket(fmt.Sprintf("ws://%s/ws", kubernetes.GetMizuCollectorProxiedHostAndPath(tappingOptions.GuiPort, mizu.ResourcesNamespace, mizu.AggregatorPodName))) controlSocketStr := fmt.Sprintf("ws://%s/ws", kubernetes.GetMizuCollectorProxiedHostAndPath(tappingOptions.GuiPort, mizu.ResourcesNamespace, mizu.AggregatorPodName))
controlSocket, err := mizu.CreateControlSocket(controlSocketStr)
if err != nil { if err != nil {
fmt.Printf("error establishing control socket connection %s\n", err) fmt.Printf("error establishing control socket connection %s\n", err)
cancel() cancel()
@ -337,7 +340,7 @@ func syncApiStatus(ctx context.Context, cancel context.CancelFunc, tappingOption
default: default:
err = controlSocket.SendNewTappedPodsListMessage(currentlyTappedPods) err = controlSocket.SendNewTappedPodsListMessage(currentlyTappedPods)
if err != nil { if err != nil {
fmt.Printf("error Sending message via control socket %s\n", err) fmt.Printf("error Sending message via control socket %v, error: %s\n", controlSocketStr, err)
} }
time.Sleep(10 * time.Second) time.Sleep(10 * time.Second)
} }