码迷,mamicode.com
首页 > 其他好文 > 详细

config.go

时间:2017-08-30 14:17:18      阅读:177      评论:0      收藏:0      [点我收藏+]

标签:lin   tls   with   color   root   cat   fun   lis   nbsp   


package clientv3

import (
    "crypto/tls"
    "crypto/x509"
    "io/ioutil"
    "time"

    "github.com/coreos/etcd/pkg/tlsutil"
    "github.com/ghodss/yaml"
)

type Config struct {
    // Endpoints is a list of URLs
    Endpoints []string

    // AutoSyncInterval is the interval to update endpoints with its latest members.
    // 0 disables auto-sync. By default auto-sync is disabled.
    AutoSyncInterval time.Duration

    // DialTimeout is the timeout for failing to establish a connection.
    DialTimeout time.Duration

    // TLS holds the client secure credentials, if any.
    TLS *tls.Config

    // Username is a username for authentication
    Username string

    // Password is a password for authentication
    Password string
}

type yamlConfig struct {
    Endpoints             []string      `json:"endpoints"`
    AutoSyncInterval      time.Duration `json:"auto-sync-interval"`
    DialTimeout           time.Duration `json:"dial-timeout"`
    InsecureTransport     bool          `json:"insecure-transport"`
    InsecureSkipTLSVerify bool          `json:"insecure-skip-tls-verify"`
    Certfile              string        `json:"cert-file"`
    Keyfile               string        `json:"key-file"`
    CAfile                string        `json:"ca-file"`
}

func configFromFile(fpath string) (*Config, error) {
    b, err := ioutil.ReadFile(fpath)
    if err != nil {
        return nil, err
    }

    yc := &yamlConfig{}

    err = yaml.Unmarshal(b, yc)
    if err != nil {
        return nil, err
    }

    cfg := &Config{
        Endpoints:        yc.Endpoints,
        AutoSyncInterval: yc.AutoSyncInterval,
        DialTimeout:      yc.DialTimeout,
    }

    if yc.InsecureTransport {
        cfg.TLS = nil
        return cfg, nil
    }

    var (
        cert *tls.Certificate
        cp   *x509.CertPool
    )

    if yc.Certfile != "" && yc.Keyfile != "" {
        cert, err = tlsutil.NewCert(yc.Certfile, yc.Keyfile, nil)
        if err != nil {
            return nil, err
        }
    }

    if yc.CAfile != "" {
        cp, err = tlsutil.NewCertPool([]string{yc.CAfile})
        if err != nil {
            return nil, err
        }
    }

    tlscfg := &tls.Config{
        MinVersion:         tls.VersionTLS10,
        InsecureSkipVerify: yc.InsecureSkipTLSVerify,
        RootCAs:            cp,
    }
    if cert != nil {
        tlscfg.Certificates = []tls.Certificate{*cert}
    }
    cfg.TLS = tlscfg

    return cfg, nil
}

config.go

标签:lin   tls   with   color   root   cat   fun   lis   nbsp   

原文地址:http://www.cnblogs.com/zhangboyu/p/7452674.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!