package services

import "knowledge-graph-backend/models"

type GraphService interface {
	GetGraphData() models.GraphData
	GetSimpleGraphData() models.SimpleGraphData
	GetNodeByID(id string) (models.Node, bool)
	SearchNodes(query string) []models.Node
	GetNeighbors(nodeID string) (models.NeighborResponse, bool)
	GetStats() map[string]int

	// 新增增删改接口
	CreateNode(req models.CreateNodeRequest) (models.Node, error)
	UpdateNode(id string, req models.UpdateNodeRequest) (models.Node, error)
	DeleteNode(id string) error

	CreateEdge(req models.CreateEdgeRequest) (models.Edge, error)
	DeleteEdge(id string) error
}