aboutsummaryrefslogtreecommitdiff
path: root/middleware/kubernetes/k8sclient/k8sclient.go
diff options
context:
space:
mode:
authorGravatar Michael Richmond <mrichmon@users.noreply.github.com> 2016-07-07 01:40:58 -0700
committerGravatar Miek Gieben <miek@miek.nl> 2016-07-07 09:40:58 +0100
commit289f53d38674a76e666aac30342eba60837466e4 (patch)
treeb68f0f4e5c956332e1b4bedd0ac57a119f0e48ec /middleware/kubernetes/k8sclient/k8sclient.go
parent558c34a23e7aec736580c5585e7eceeff9220667 (diff)
downloadcoredns-289f53d38674a76e666aac30342eba60837466e4.tar.gz
coredns-289f53d38674a76e666aac30342eba60837466e4.tar.zst
coredns-289f53d38674a76e666aac30342eba60837466e4.zip
k8s middleware cleanup, testcases, basic SRV (#181)
* Removing unnecessary gitignore pattern * Updating Makefile to run unittests for subpackages * Adding Corefile validation to ignore overlapping zones * Fixing SRV query handling * Updating README.md now that SRV works * Fixing debug message, adding code comment * Clarifying implementation of zone normalization * "Overlapping zones" is ill-defined. Reimplemented zone overlap/subzone checking to contain these functions in k8s middleware and provide better code comments explaining the normalization. * Separate build verbosity from test verbosity * Cleaning up comments to match repo code style * Merging warning messages into single message * Moving function docs to before function declaration * Adding test cases for k8sclient connector * Tests cover connector create and setting base url * Fixed bugs in connector create and setting base url functions * Updaing README to group and order development work * Priority focused on achieving functional parity with SkyDNS. * Adding work items to README and cleaning up formatting * More README format cleaning * List formating * Refactoring k8s API call to allow dependency injection * Add test cases for data parsing from k8s into dataobject structures * URL is dependency-injected to allow replacement with a mock http server during test execution * Adding more data validation for JSON parsing tests * Adding test case for GetResourceList() * Adding notes about SkyDNS embedded IP and port record names * Marked test case implemented. * Fixing formatting for example command. * Fixing formatting * Adding notes about Docker image building. * Adding SkyDNS work item * Updating TODO list * Adding name template to Corefile to specify how k8s record names are assembled * Adding template support for multi-segment zones * Updating example CoreFile for k8s with template comment * Misc whitespace cleanup * Adding SkyDNS naming notes * Adding namespace filtering to CoreFile config * Updating example k8sCoreFile to specify namespaces * Removing unused codepath * Adding check for valid namespace * More README TODO restructuring to focus effort * Adding template validation while parsing CoreFile * Record name template is considered invalid if it contains a symbol of the form ${bar} where the symbol "${bar}" is not an accepted template symbol. * Refactoring generation of answer records * Parse typeName out of query string * Refactor answer record creation as operation over list of ServiceItems * Moving k8s API caching into SkyDNS equivalency segment * Adding function to assemble record names from template * Warning: This commit may be broken. Syncing to get laptop code over to dev machine. * More todo notes * Adding comment describing sample test data. * Update k8sCorefile * Adding comment * Adding filtering support for kubernetes "type" * Required refactoring to support reuse of the StringInSlice function. * Cleaning up formatting * Adding note about SkyDNS supporting word "any". * baseUrl -> baseURL * Also removed debug statement from core/setup/kubernetes.go * Fixing test breaking from Url -> URL naming changes * Changing record name template language ${...} -> {...} * Fix formatting with go fmt * Updating all k8sclient data getters to return error value * Adding error message to k8sclient data accessors * Cleaning up setup for kubernetes * Removed verbose nils in initial k8s middleware instance * Set reasonable defaults if CoreFile has no parameters in the kubernetes block. (k8s endpoint, and name template) * Formatting cleanup -- go fmt
Diffstat (limited to 'middleware/kubernetes/k8sclient/k8sclient.go')
-rw-r--r--middleware/kubernetes/k8sclient/k8sclient.go172
1 files changed, 106 insertions, 66 deletions
diff --git a/middleware/kubernetes/k8sclient/k8sclient.go b/middleware/kubernetes/k8sclient/k8sclient.go
index a05ef8905..95300f3b9 100644
--- a/middleware/kubernetes/k8sclient/k8sclient.go
+++ b/middleware/kubernetes/k8sclient/k8sclient.go
@@ -1,117 +1,157 @@
package k8sclient
import (
-// "fmt"
- "net/url"
+ "errors"
+ "fmt"
+ "net/url"
+ "strings"
)
// API strings
const (
- apiBase = "/api/v1"
- apiNamespaces = "/namespaces"
- apiServices = "/services"
+ apiBase = "/api/v1"
+ apiNamespaces = "/namespaces"
+ apiServices = "/services"
)
// Defaults
const (
- defaultBaseUrl = "http://localhost:8080"
+ defaultBaseURL = "http://localhost:8080"
)
-
type K8sConnector struct {
- baseUrl string
+ baseURL string
}
-func (c *K8sConnector) SetBaseUrl(u string) error {
- validUrl, error := url.Parse(u)
+func (c *K8sConnector) SetBaseURL(u string) error {
+ url, error := url.Parse(u)
+
+ if error != nil {
+ return error
+ }
- if error != nil {
- return error
- }
- c.baseUrl = validUrl.String()
+ if !url.IsAbs() {
+ return errors.New("k8sclient: Kubernetes endpoint url must be an absolute URL")
+ }
- return nil
+ c.baseURL = url.String()
+ return nil
}
-func (c *K8sConnector) GetBaseUrl() string {
- return c.baseUrl
+func (c *K8sConnector) GetBaseURL() string {
+ return c.baseURL
}
+// URL constructor separated from code to support dependency injection
+// for unit tests.
+var makeURL = func(parts []string) string {
+ return strings.Join(parts, "")
+}
-func (c *K8sConnector) GetResourceList() *ResourceList {
- resources := new(ResourceList)
-
- error := getJson((c.baseUrl + apiBase), resources)
- if error != nil {
- return nil
- }
+func (c *K8sConnector) GetResourceList() (*ResourceList, error) {
+ resources := new(ResourceList)
- return resources
-}
+ url := makeURL([]string{c.baseURL, apiBase})
+ err := parseJson(url, resources)
+ // TODO: handle no response from k8s
+ if err != nil {
+ fmt.Printf("[ERROR] Response from kubernetes API for GetResourceList() is: %v\n", err)
+ return nil, err
+ }
+ return resources, nil
+}
-func (c *K8sConnector) GetNamespaceList() *NamespaceList {
- namespaces := new(NamespaceList)
+func (c *K8sConnector) GetNamespaceList() (*NamespaceList, error) {
+ namespaces := new(NamespaceList)
- error := getJson((c.baseUrl + apiBase + apiNamespaces), namespaces)
- if error != nil {
- return nil
- }
+ url := makeURL([]string{c.baseURL, apiBase, apiNamespaces})
+ err := parseJson(url, namespaces)
+ if err != nil {
+ fmt.Printf("[ERROR] Response from kubernetes API for GetNamespaceList() is: %v\n", err)
+ return nil, err
+ }
- return namespaces
+ return namespaces, nil
}
+func (c *K8sConnector) GetServiceList() (*ServiceList, error) {
+ services := new(ServiceList)
-func (c *K8sConnector) GetServiceList() *ServiceList {
- services := new(ServiceList)
+ url := makeURL([]string{c.baseURL, apiBase, apiServices})
+ err := parseJson(url, services)
+ // TODO: handle no response from k8s
+ if err != nil {
+ fmt.Printf("[ERROR] Response from kubernetes API for GetServiceList() is: %v\n", err)
+ return nil, err
+ }
- error := getJson((c.baseUrl + apiBase + apiServices), services)
- if error != nil {
- return nil
- }
-
- return services
+ return services, nil
}
+// GetServicesByNamespace returns a map of
+// namespacename :: [ kubernetesServiceItem ]
+func (c *K8sConnector) GetServicesByNamespace() (map[string][]ServiceItem, error) {
+
+ items := make(map[string][]ServiceItem)
+
+ k8sServiceList, err := c.GetServiceList()
-func (c *K8sConnector) GetServicesByNamespace() map[string][]ServiceItem {
- // GetServicesByNamespace returns a map of namespacename :: [ kubernetesServiceItem ]
+ if err != nil {
+ fmt.Printf("[ERROR] Getting service list produced error: %v", err)
+ return nil, err
+ }
- items := make(map[string][]ServiceItem)
+ // TODO: handle no response from k8s
+ if k8sServiceList == nil {
+ return nil, nil
+ }
- k8sServiceList := c.GetServiceList()
- k8sItemList := k8sServiceList.Items
+ k8sItemList := k8sServiceList.Items
- for _, i := range k8sItemList {
- namespace := i.Metadata.Namespace
- items[namespace] = append(items[namespace], i)
- }
+ for _, i := range k8sItemList {
+ namespace := i.Metadata.Namespace
+ items[namespace] = append(items[namespace], i)
+ }
- return items
+ return items, nil
}
+// GetServiceItemsInNamespace returns the ServiceItems that match
+// servicename in the namespace
+func (c *K8sConnector) GetServiceItemsInNamespace(namespace string, servicename string) ([]*ServiceItem, error) {
-func (c *K8sConnector) GetServiceItemInNamespace(namespace string, servicename string) *ServiceItem {
- // GetServiceItemInNamespace returns the ServiceItem that matches servicename in the namespace
+ itemMap, err := c.GetServicesByNamespace()
- itemMap := c.GetServicesByNamespace()
+ if err != nil {
+ fmt.Printf("[ERROR] Getting service list produced error: %v", err)
+ return nil, err
+ }
- // TODO: Handle case where namesapce == nil
+ // TODO: Handle case where namespace == nil
- for _, x := range itemMap[namespace] {
- if x.Metadata.Name == servicename {
- return &x
- }
- }
+ var serviceItems []*ServiceItem
- // No matching item found in namespace
- return nil
+ for _, x := range itemMap[namespace] {
+ if x.Metadata.Name == servicename {
+ serviceItems = append(serviceItems, &x)
+ }
+ }
+
+ return serviceItems, nil
}
+func NewK8sConnector(baseURL string) *K8sConnector {
+ k := new(K8sConnector)
+
+ if baseURL == "" {
+ baseURL = defaultBaseURL
+ }
-func NewK8sConnector(baseurl string) *K8sConnector {
- k := new(K8sConnector)
- k.SetBaseUrl(baseurl)
+ err := k.SetBaseURL(baseURL)
+ if err != nil {
+ return nil
+ }
- return k
+ return k
}