1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192 |
- // Copyright 2018 The etcd 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 flags
- import (
- "flag"
- "net/url"
- "sort"
- "strings"
- "go.etcd.io/etcd/pkg/types"
- )
- // UniqueURLs contains unique URLs
- // with non-URL exceptions.
- type UniqueURLs struct {
- Values map[string]struct{}
- uss []url.URL
- Allowed map[string]struct{}
- }
- // Set parses a command line set of URLs formatted like:
- // http://127.0.0.1:2380,http://10.1.1.2:80
- // Implements "flag.Value" interface.
- func (us *UniqueURLs) Set(s string) error {
- if _, ok := us.Values[s]; ok {
- return nil
- }
- if _, ok := us.Allowed[s]; ok {
- us.Values[s] = struct{}{}
- return nil
- }
- ss, err := types.NewURLs(strings.Split(s, ","))
- if err != nil {
- return err
- }
- us.Values = make(map[string]struct{})
- us.uss = make([]url.URL, 0)
- for _, v := range ss {
- us.Values[v.String()] = struct{}{}
- us.uss = append(us.uss, v)
- }
- return nil
- }
- // String implements "flag.Value" interface.
- func (us *UniqueURLs) String() string {
- all := make([]string, 0, len(us.Values))
- for u := range us.Values {
- all = append(all, u)
- }
- sort.Strings(all)
- return strings.Join(all, ",")
- }
- // NewUniqueURLsWithExceptions implements "url.URL" slice as flag.Value interface.
- // Given value is to be separated by comma.
- func NewUniqueURLsWithExceptions(s string, exceptions ...string) *UniqueURLs {
- us := &UniqueURLs{Values: make(map[string]struct{}), Allowed: make(map[string]struct{})}
- for _, v := range exceptions {
- us.Allowed[v] = struct{}{}
- }
- if s == "" {
- return us
- }
- if err := us.Set(s); err != nil {
- plog.Panicf("new UniqueURLs should never fail: %v", err)
- }
- return us
- }
- // UniqueURLsFromFlag returns a slice from urls got from the flag.
- func UniqueURLsFromFlag(fs *flag.FlagSet, urlsFlagName string) []url.URL {
- return (*fs.Lookup(urlsFlagName).Value.(*UniqueURLs)).uss
- }
- // UniqueURLsMapFromFlag returns a map from url strings got from the flag.
- func UniqueURLsMapFromFlag(fs *flag.FlagSet, urlsFlagName string) map[string]struct{} {
- return (*fs.Lookup(urlsFlagName).Value.(*UniqueURLs)).Values
- }
|