c4.go

package network

type NetworkElement interface {
    // methods to interact with the element
    GetName() string
    GetType() string
}

type Router struct {
    name string
    model string
}

func (r *Router) GetName() string {
    return r.name
}

func (r *Router) GetType() string {
    return "Router"
}

type Switch struct {
    name string
    model string
}

func (s *Switch) GetName() string {
    return s.name
}

func (s *Switch) GetType() string {
    return "Switch"
}

type Network struct {
    elements []NetworkElement
}

func (n *Network) AddElement(elem NetworkElement) {
    n.elements = append(n.elements, elem)
}

func (n *Network) GetElements() []NetworkElement {
    return n.elements
}