|
1 | 1 | package pkg |
2 | 2 |
|
3 | 3 | import ( |
| 4 | + "fmt" |
4 | 5 | "log" |
5 | 6 | "sync" |
6 | 7 | "time" |
@@ -150,6 +151,16 @@ type NetworkTopology struct { |
150 | 151 | mutex sync.RWMutex |
151 | 152 | } |
152 | 153 |
|
| 154 | +// NewNetworkTopology creates a new NetworkTopology instance |
| 155 | +func NewNetworkTopology() *NetworkTopology { |
| 156 | + return &NetworkTopology{ |
| 157 | + Nodes: make(map[string]*TopologyNode), |
| 158 | + Links: make(map[string]*TopologyLink), |
| 159 | + LastUpdated: time.Now(), |
| 160 | + Version: "1.0.0", |
| 161 | + } |
| 162 | +} |
| 163 | + |
153 | 164 | // TopologyNode represents a node in the network |
154 | 165 | type TopologyNode struct { |
155 | 166 | Name string `json:"name"` |
@@ -450,3 +461,49 @@ func NewNetworkState() *NetworkState { |
450 | 461 | activeSlices: make(map[string]*SliceState), |
451 | 462 | } |
452 | 463 | } |
| 464 | + |
| 465 | +// UpdateVXLANConfig updates VXLAN configuration for a slice |
| 466 | +func (ns *NetworkState) UpdateVXLANConfig(sliceID string, config *DynamicVXLANConfig) error { |
| 467 | + ns.mutex.Lock() |
| 468 | + defer ns.mutex.Unlock() |
| 469 | + ns.sliceConfigs[sliceID] = config |
| 470 | + return nil |
| 471 | +} |
| 472 | + |
| 473 | +// GetVXLANConfig retrieves VXLAN configuration for a slice |
| 474 | +func (ns *NetworkState) GetVXLANConfig(sliceID string) (*DynamicVXLANConfig, error) { |
| 475 | + ns.mutex.RLock() |
| 476 | + defer ns.mutex.RUnlock() |
| 477 | + config, exists := ns.sliceConfigs[sliceID] |
| 478 | + if !exists { |
| 479 | + return nil, fmt.Errorf("VXLAN config not found for slice %s", sliceID) |
| 480 | + } |
| 481 | + return config, nil |
| 482 | +} |
| 483 | + |
| 484 | +// UpdateQoSStrategy updates QoS strategy for a slice |
| 485 | +func (ns *NetworkState) UpdateQoSStrategy(sliceID string, strategy *QoSStrategy) error { |
| 486 | + ns.mutex.Lock() |
| 487 | + defer ns.mutex.Unlock() |
| 488 | + ns.qosStrategies[sliceID] = strategy |
| 489 | + return nil |
| 490 | +} |
| 491 | + |
| 492 | +// GetQoSStrategy retrieves QoS strategy for a slice |
| 493 | +func (ns *NetworkState) GetQoSStrategy(sliceID string) (*QoSStrategy, error) { |
| 494 | + ns.mutex.RLock() |
| 495 | + defer ns.mutex.RUnlock() |
| 496 | + strategy, exists := ns.qosStrategies[sliceID] |
| 497 | + if !exists { |
| 498 | + return nil, fmt.Errorf("QoS strategy not found for slice %s", sliceID) |
| 499 | + } |
| 500 | + return strategy, nil |
| 501 | +} |
| 502 | + |
| 503 | +// UpdateTopology updates the network topology |
| 504 | +func (ns *NetworkState) UpdateTopology(topology *NetworkTopology) error { |
| 505 | + ns.mutex.Lock() |
| 506 | + defer ns.mutex.Unlock() |
| 507 | + ns.topology = topology |
| 508 | + return nil |
| 509 | +} |
0 commit comments