| 
									
										
										
										
											2019-06-09 00:10:57 -07:00
										 |  |  | package kubernetes
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | import (
 | 
					
						
							|  |  |  | 	"context"
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-01-31 14:35:22 -05:00
										 |  |  | 	"github.com/coredns/coredns/plugin"
 | 
					
						
							| 
									
										
										
										
											2019-06-09 00:10:57 -07:00
										 |  |  | 	"github.com/coredns/coredns/plugin/metadata"
 | 
					
						
							|  |  |  | 	"github.com/coredns/coredns/request"
 | 
					
						
							|  |  |  | )
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | // Metadata implements the metadata.Provider interface.
 | 
					
						
							|  |  |  | func (k *Kubernetes) Metadata(ctx context.Context, state request.Request) context.Context {
 | 
					
						
							| 
									
										
										
										
											2020-01-31 14:35:22 -05:00
										 |  |  | 	zone := plugin.Zones(k.Zones).Matches(state.Name())
 | 
					
						
							| 
									
										
										
										
											2019-06-09 00:10:57 -07:00
										 |  |  | 	// possible optimization: cache r so it doesn't need to be calculated again in ServeDNS
 | 
					
						
							| 
									
										
										
										
											2020-01-31 14:35:22 -05:00
										 |  |  | 	r, err := parseRequest(state.Name(), zone)
 | 
					
						
							| 
									
										
										
										
											2019-06-09 00:10:57 -07:00
										 |  |  | 	if err != nil {
 | 
					
						
							|  |  |  | 		metadata.SetValueFunc(ctx, "kubernetes/parse-error", func() string {
 | 
					
						
							|  |  |  | 			return err.Error()
 | 
					
						
							|  |  |  | 		})
 | 
					
						
							|  |  |  | 		return ctx
 | 
					
						
							|  |  |  | 	}
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	metadata.SetValueFunc(ctx, "kubernetes/port-name", func() string {
 | 
					
						
							|  |  |  | 		return r.port
 | 
					
						
							|  |  |  | 	})
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	metadata.SetValueFunc(ctx, "kubernetes/protocol", func() string {
 | 
					
						
							|  |  |  | 		return r.protocol
 | 
					
						
							|  |  |  | 	})
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	metadata.SetValueFunc(ctx, "kubernetes/endpoint", func() string {
 | 
					
						
							|  |  |  | 		return r.endpoint
 | 
					
						
							|  |  |  | 	})
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	metadata.SetValueFunc(ctx, "kubernetes/service", func() string {
 | 
					
						
							|  |  |  | 		return r.service
 | 
					
						
							|  |  |  | 	})
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	metadata.SetValueFunc(ctx, "kubernetes/namespace", func() string {
 | 
					
						
							|  |  |  | 		return r.namespace
 | 
					
						
							|  |  |  | 	})
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	metadata.SetValueFunc(ctx, "kubernetes/kind", func() string {
 | 
					
						
							|  |  |  | 		return r.podOrSvc
 | 
					
						
							|  |  |  | 	})
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	pod := k.podWithIP(state.IP())
 | 
					
						
							|  |  |  | 	if pod == nil {
 | 
					
						
							|  |  |  | 		return ctx
 | 
					
						
							|  |  |  | 	}
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	metadata.SetValueFunc(ctx, "kubernetes/client-namespace", func() string {
 | 
					
						
							|  |  |  | 		return pod.Namespace
 | 
					
						
							|  |  |  | 	})
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	metadata.SetValueFunc(ctx, "kubernetes/client-pod-name", func() string {
 | 
					
						
							|  |  |  | 		return pod.Name
 | 
					
						
							|  |  |  | 	})
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	return ctx
 | 
					
						
							|  |  |  | }
 |