123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155 |
- // Copyright 2015 CoreOS, Inc.
- //
- // 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 subnet
- import (
- "fmt"
- "path"
- "sync"
- "time"
- "github.com/coreos/flannel/Godeps/_workspace/src/github.com/coreos/go-etcd/etcd"
- log "github.com/coreos/flannel/Godeps/_workspace/src/github.com/golang/glog"
- )
- type subnetRegistry interface {
- getConfig() (*etcd.Response, error)
- getSubnets() (*etcd.Response, error)
- createSubnet(sn, data string, ttl uint64) (*etcd.Response, error)
- updateSubnet(sn, data string, ttl uint64) (*etcd.Response, error)
- watchSubnets(since uint64, stop chan bool) (*etcd.Response, error)
- }
- type EtcdConfig struct {
- Endpoints []string
- Keyfile string
- Certfile string
- CAFile string
- Prefix string
- }
- type etcdSubnetRegistry struct {
- mux sync.Mutex
- cli *etcd.Client
- etcdCfg *EtcdConfig
- }
- func newEtcdClient(c *EtcdConfig) (*etcd.Client, error) {
- if c.Keyfile != "" || c.Certfile != "" || c.CAFile != "" {
- return etcd.NewTLSClient(c.Endpoints, c.Certfile, c.Keyfile, c.CAFile)
- } else {
- return etcd.NewClient(c.Endpoints), nil
- }
- }
- func newEtcdSubnetRegistry(config *EtcdConfig) (subnetRegistry, error) {
- r := &etcdSubnetRegistry{
- etcdCfg: config,
- }
- var err error
- r.cli, err = newEtcdClient(config)
- if err != nil {
- return nil, err
- }
- return r, nil
- }
- func (esr *etcdSubnetRegistry) getConfig() (*etcd.Response, error) {
- key := path.Join(esr.etcdCfg.Prefix, "config")
- resp, err := esr.client().Get(key, false, false)
- if err != nil {
- return nil, err
- }
- return resp, nil
- }
- func (esr *etcdSubnetRegistry) getSubnets() (*etcd.Response, error) {
- key := path.Join(esr.etcdCfg.Prefix, "subnets")
- return esr.client().Get(key, false, true)
- }
- func (esr *etcdSubnetRegistry) createSubnet(sn, data string, ttl uint64) (*etcd.Response, error) {
- key := path.Join(esr.etcdCfg.Prefix, "subnets", sn)
- resp, err := esr.client().Create(key, data, ttl)
- if err != nil {
- return nil, err
- }
- ensureExpiration(resp, ttl)
- return resp, nil
- }
- func (esr *etcdSubnetRegistry) updateSubnet(sn, data string, ttl uint64) (*etcd.Response, error) {
- key := path.Join(esr.etcdCfg.Prefix, "subnets", sn)
- resp, err := esr.client().Set(key, data, ttl)
- if err != nil {
- return nil, err
- }
- ensureExpiration(resp, ttl)
- return resp, nil
- }
- func (esr *etcdSubnetRegistry) watchSubnets(since uint64, stop chan bool) (*etcd.Response, error) {
- for {
- key := path.Join(esr.etcdCfg.Prefix, "subnets")
- resp, err := esr.client().RawWatch(key, since, true, nil, stop)
- if err != nil {
- if err == etcd.ErrWatchStoppedByUser {
- return nil, nil
- } else {
- return nil, err
- }
- }
- if len(resp.Body) == 0 {
- // etcd timed out, go back but recreate the client as the underlying
- // http transport gets hosed (http://code.google.com/p/go/issues/detail?id=8648)
- esr.resetClient()
- continue
- }
- return resp.Unmarshal()
- }
- }
- func (esr *etcdSubnetRegistry) client() *etcd.Client {
- esr.mux.Lock()
- defer esr.mux.Unlock()
- return esr.cli
- }
- func (esr *etcdSubnetRegistry) resetClient() {
- esr.mux.Lock()
- defer esr.mux.Unlock()
- var err error
- esr.cli, err = newEtcdClient(esr.etcdCfg)
- if err != nil {
- panic(fmt.Errorf("resetClient: error recreating etcd client: %v", err))
- }
- }
- func ensureExpiration(resp *etcd.Response, ttl uint64) {
- if resp.Node.Expiration == nil {
- // should not be but calc it ourselves in this case
- log.Info("Expiration field missing on etcd response, calculating locally")
- exp := time.Now().Add(time.Duration(ttl) * time.Second)
- resp.Node.Expiration = &exp
- }
- }
|