mirror of
https://github.com/kubeshark/kubeshark.git
synced 2025-06-21 05:53:29 +00:00
237002ef29
* Update main.go, main.go, and 3 more files... * WIP * Update main.go, oas_controller.go, and 3 more files... * Update main.go, oas_generator.go, and servicemap.go * Update loader.go and resolver.go * Update oas_generator.go * Update oas_generator.go, specgen_test.go, and 3 more files... * Update service_map_controller_test.go * Update oas_controller_test.go
40 lines
903 B
Go
40 lines
903 B
Go
package controllers
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/up9inc/mizu/agent/pkg/dependency"
|
|
"github.com/up9inc/mizu/agent/pkg/servicemap"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type ServiceMapController struct {
|
|
service servicemap.ServiceMap
|
|
}
|
|
|
|
func NewServiceMapController() *ServiceMapController {
|
|
serviceMapGenerator := dependency.GetInstance(dependency.ServiceMapGeneratorDependency).(servicemap.ServiceMap)
|
|
return &ServiceMapController{
|
|
service: serviceMapGenerator,
|
|
}
|
|
}
|
|
|
|
func (s *ServiceMapController) Status(c *gin.Context) {
|
|
c.JSON(http.StatusOK, s.service.GetStatus())
|
|
}
|
|
|
|
func (s *ServiceMapController) Get(c *gin.Context) {
|
|
response := &servicemap.ServiceMapResponse{
|
|
Status: s.service.GetStatus(),
|
|
Nodes: s.service.GetNodes(),
|
|
Edges: s.service.GetEdges(),
|
|
}
|
|
c.JSON(http.StatusOK, response)
|
|
}
|
|
|
|
func (s *ServiceMapController) Reset(c *gin.Context) {
|
|
s.service.Reset()
|
|
s.Status(c)
|
|
}
|