| // Copyright Istio Authors |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| package model |
| |
| import ( |
| "strings" |
| "sync" |
| ) |
| |
| import ( |
| "github.com/apache/dubbo-go-pixiu/pkg/config/host" |
| ) |
| |
| var ( |
| defaultClusterLocalNamespaces = []string{"kube-system"} |
| defaultClusterLocalServices = []string{"kubernetes.default.svc"} |
| ) |
| |
| // ClusterLocalHosts is a map of host names or wildcard patterns which should only |
| // be made accessible from within the same cluster. |
| type ClusterLocalHosts map[host.Name]struct{} |
| |
| // IsClusterLocal indicates whether the given host should be treated as a |
| // cluster-local destination. |
| func (c ClusterLocalHosts) IsClusterLocal(h host.Name) bool { |
| _, ok := MostSpecificHostMatch2(h, c) |
| return ok |
| } |
| |
| // ClusterLocalProvider provides the cluster-local hosts. |
| type ClusterLocalProvider interface { |
| // GetClusterLocalHosts returns the list of cluster-local hosts, sorted in |
| // ascending order. The caller must not modify the returned list. |
| GetClusterLocalHosts() ClusterLocalHosts |
| } |
| |
| // NewClusterLocalProvider returns a new ClusterLocalProvider for the Environment. |
| func NewClusterLocalProvider(e *Environment) ClusterLocalProvider { |
| c := &clusterLocalProvider{} |
| |
| // Register a handler to update the environment when the mesh config is updated. |
| e.AddMeshHandler(func() { |
| c.onMeshUpdated(e) |
| }) |
| |
| // Update the cluster-local hosts now. |
| c.onMeshUpdated(e) |
| return c |
| } |
| |
| var _ ClusterLocalProvider = &clusterLocalProvider{} |
| |
| type clusterLocalProvider struct { |
| mutex sync.Mutex |
| hosts ClusterLocalHosts |
| } |
| |
| func (c *clusterLocalProvider) GetClusterLocalHosts() ClusterLocalHosts { |
| c.mutex.Lock() |
| out := c.hosts |
| c.mutex.Unlock() |
| return out |
| } |
| |
| func (c *clusterLocalProvider) onMeshUpdated(e *Environment) { |
| // Create the default list of cluster-local hosts. |
| domainSuffix := e.DomainSuffix |
| defaultClusterLocalHosts := make([]host.Name, 0) |
| for _, n := range defaultClusterLocalNamespaces { |
| defaultClusterLocalHosts = append(defaultClusterLocalHosts, host.Name("*."+n+".svc."+domainSuffix)) |
| } |
| for _, s := range defaultClusterLocalServices { |
| defaultClusterLocalHosts = append(defaultClusterLocalHosts, host.Name(s+"."+domainSuffix)) |
| } |
| |
| if discoveryHost, _, err := e.GetDiscoveryAddress(); err != nil { |
| log.Errorf("failed to make discoveryAddress cluster-local: %v", err) |
| } else { |
| if !strings.HasSuffix(string(discoveryHost), domainSuffix) { |
| discoveryHost += host.Name("." + domainSuffix) |
| } |
| defaultClusterLocalHosts = append(defaultClusterLocalHosts, discoveryHost) |
| } |
| |
| // Collect the cluster-local hosts. |
| hosts := make(ClusterLocalHosts, 0) |
| for _, serviceSettings := range e.Mesh().ServiceSettings { |
| if serviceSettings.Settings.ClusterLocal { |
| for _, h := range serviceSettings.Hosts { |
| hosts[host.Name(h)] = struct{}{} |
| } |
| } else { |
| // Remove defaults if specified to be non-cluster-local. |
| for _, h := range serviceSettings.Hosts { |
| for i, defaultClusterLocalHost := range defaultClusterLocalHosts { |
| if len(defaultClusterLocalHost) > 0 { |
| if h == string(defaultClusterLocalHost) || |
| (defaultClusterLocalHost.IsWildCarded() && |
| strings.HasSuffix(h, string(defaultClusterLocalHost[1:]))) { |
| // This default was explicitly overridden, so remove it. |
| defaultClusterLocalHosts[i] = "" |
| } |
| } |
| } |
| } |
| } |
| } |
| |
| // Add any remaining defaults to the end of the list. |
| for _, defaultClusterLocalHost := range defaultClusterLocalHosts { |
| if len(defaultClusterLocalHost) > 0 { |
| hosts[defaultClusterLocalHost] = struct{}{} |
| } |
| } |
| |
| c.mutex.Lock() |
| c.hosts = hosts |
| c.mutex.Unlock() |
| } |