knowledge-graph-agent/backend/handlers/graph_handler.go

package handlers

import (
	"net/http"

	"github.com/gin-gonic/gin"
	"knowledge-graph-backend/services"
)

type GraphHandler struct {
	service services.GraphService
}

func NewGraphHandler(service services.GraphService) *GraphHandler {
	return &GraphHandler{
		service: service,
	}
}

func (h *GraphHandler) GetGraphData(c *gin.Context) {
	data := h.service.GetGraphData()
	c.JSON(http.StatusOK, data)
}

func (h *GraphHandler) GetStats(c *gin.Context) {
	stats := h.service.GetStats()
	c.JSON(http.StatusOK, gin.H{
		"stats": stats,
		"timestamp": gin.H{
			"totalNodes": stats["totalNodes"],
			"totalEdges": stats["totalEdges"],
		},
	})
}