Initial commit: 网络拓扑发现系统

- 支持Cisco、华为、H3C、ASA、Linux、Windows设备
- SSH远程采集设备信息
- 自动发现网络拓扑(LLDP/CDP)
- Web可视化界面
- 支持旧版SSH加密算法兼容

internal/topology/builder.go

@@ -0,0 +1,122 @@
+package topology
+
+import (
+	"fmt"
+	"network-topology-discovery/pkg/models"
+)
+
+// Builder 拓扑构建器
+type Builder struct {
+	devices []models.Device
+}
+
+// NewBuilder 创建拓扑构建器
+func NewBuilder() *Builder {
+	return &Builder{}
+}
+
+// AddDevice 添加设备
+func (b *Builder) AddDevice(device models.Device) {
+	b.devices = append(b.devices, device)
+}
+
+// AddDevices 批量添加设备
+func (b *Builder) AddDevices(devices []models.Device) {
+	b.devices = append(b.devices, devices...)
+}
+
+// Build 构建拓扑图
+func (b *Builder) Build() models.TopologyGraph {
+	graph := models.TopologyGraph{
+		Nodes: make([]models.TopologyNode, 0),
+		Edges: make([]models.TopologyEdge, 0),
+	}
+
+	// 构建节点
+	nodeMap := make(map[string]models.TopologyNode)
+	for _, device := range b.devices {
+		node := models.TopologyNode{
+			ID:       device.IP,
+			IP:       device.IP,
+			Hostname: device.Hostname,
+			Type:     string(device.Type),
+			Icon:     getDeviceIcon(device.Type),
+		}
+		nodeMap[device.IP] = node
+		graph.Nodes = append(graph.Nodes, node)
+	}
+
+	// 构建边(基于邻居信息)
+	edgeMap := make(map[string]bool) // 用于去重
+	for _, device := range b.devices {
+		for _, neighbor := range device.Neighbors {
+			// 检查邻居是否在设备列表中
+			if _, exists := nodeMap[neighbor.RemoteIP]; !exists && neighbor.RemoteDevice != "" {
+				// 尝试通过设备名匹配
+				for _, d := range b.devices {
+					if d.Hostname == neighbor.RemoteDevice {
+						neighbor.RemoteIP = d.IP
+						break
+					}
+				}
+			}
+
+			if neighbor.RemoteIP == "" {
+				continue
+			}
+
+			// 创建唯一的边ID
+			edgeID := fmt.Sprintf("%s-%s-%s", device.IP, neighbor.LocalInterface, neighbor.RemoteIP)
+			reverseEdgeID := fmt.Sprintf("%s-%s-%s", neighbor.RemoteIP, neighbor.RemoteInterface, device.IP)
+
+			// 避免重复边
+			if edgeMap[edgeID] || edgeMap[reverseEdgeID] {
+				continue
+			}
+
+			edge := models.TopologyEdge{
+				ID:              edgeID,
+				Source:          device.IP,
+				Target:          neighbor.RemoteIP,
+				SourceInterface: neighbor.LocalInterface,
+				TargetInterface: neighbor.RemoteInterface,
+				Protocol:        neighbor.Protocol,
+			}
+
+			graph.Edges = append(graph.Edges, edge)
+			edgeMap[edgeID] = true
+		}
+	}
+
+	return graph
+}
+
+// getDeviceIcon 获取设备图标
+func getDeviceIcon(deviceType models.DeviceType) string {
+	switch deviceType {
+	case models.DeviceTypeCisco:
+		return "router"
+	case models.DeviceTypeHuawei:
+		return "router"
+	case models.DeviceTypeH3C:
+		return "switch"
+	case models.DeviceTypeASA:
+		return "firewall"
+	case models.DeviceTypeLinux:
+		return "server"
+	case models.DeviceTypeWindows:
+		return "server"
+	default:
+		return "device"
+	}
+}
+
+// GetDevices 获取所有设备
+func (b *Builder) GetDevices() []models.Device {
+	return b.devices
+}
+
+// Clear 清空拓扑
+func (b *Builder) Clear() {
+	b.devices = make([]models.Device, 0)
+}