66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
// Copyright 2015 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"
|
|
"strings"
|
|
|
|
"github.com/coreos/etcd/pkg/types"
|
|
)
|
|
|
|
// URLsValue wraps "types.URLs".
|
|
type URLsValue types.URLs
|
|
|
|
// 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 *URLsValue) Set(s string) error {
|
|
ss, err := types.NewURLs(strings.Split(s, ","))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*us = URLsValue(ss)
|
|
return nil
|
|
}
|
|
|
|
// String implements "flag.Value" interface.
|
|
func (us *URLsValue) String() string {
|
|
all := make([]string, len(*us))
|
|
for i, u := range *us {
|
|
all[i] = u.String()
|
|
}
|
|
return strings.Join(all, ",")
|
|
}
|
|
|
|
// NewURLsValue implements "url.URL" slice as flag.Value interface.
|
|
// Given value is to be separated by comma.
|
|
func NewURLsValue(s string) *URLsValue {
|
|
if s == "" {
|
|
return &URLsValue{}
|
|
}
|
|
v := &URLsValue{}
|
|
if err := v.Set(s); err != nil {
|
|
plog.Panicf("new URLsValue should never fail: %v", err)
|
|
}
|
|
return v
|
|
}
|
|
|
|
// URLsFromFlag returns a slices from url got from the flag.
|
|
func URLsFromFlag(fs *flag.FlagSet, urlsFlagName string) []url.URL {
|
|
return []url.URL(*fs.Lookup(urlsFlagName).Value.(*URLsValue))
|
|
}
|