85effbbc3f
Specifying an API type as IntOrString will allow JSON and YAML to accept either ints or strings with the same name. For example, port names or numbers.
130 lines
2.9 KiB
Go
130 lines
2.9 KiB
Go
/*
|
|
Copyright 2014 Google Inc. All rights reserved.
|
|
|
|
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 util
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"runtime"
|
|
"time"
|
|
|
|
"github.com/golang/glog"
|
|
)
|
|
|
|
// For testing, bypass HandleCrash.
|
|
var ReallyCrash bool
|
|
|
|
// Simply catches a crash and logs an error. Meant to be called via defer.
|
|
func HandleCrash() {
|
|
if ReallyCrash {
|
|
return
|
|
}
|
|
|
|
r := recover()
|
|
if r != nil {
|
|
callers := ""
|
|
for i := 0; true; i++ {
|
|
_, file, line, ok := runtime.Caller(i)
|
|
if !ok {
|
|
break
|
|
}
|
|
callers = callers + fmt.Sprintf("%v:%v\n", file, line)
|
|
}
|
|
glog.Infof("Recovered from panic: %#v (%v)\n%v", r, r, callers)
|
|
}
|
|
}
|
|
|
|
// Loops forever running f every d. Catches any panics, and keeps going.
|
|
func Forever(f func(), period time.Duration) {
|
|
for {
|
|
func() {
|
|
defer HandleCrash()
|
|
f()
|
|
}()
|
|
time.Sleep(period)
|
|
}
|
|
}
|
|
|
|
// Returns o marshalled as a JSON string, ignoring any errors.
|
|
func MakeJSONString(o interface{}) string {
|
|
data, _ := json.Marshal(o)
|
|
return string(data)
|
|
}
|
|
|
|
// IntOrString is a type that can hold an int or a string. When used in
|
|
// JSON or YAML marshalling and unmarshalling, it produces or consumes the
|
|
// inner type. This allows you to have, for example, a JSON field that can
|
|
// accept a name or number.
|
|
type IntOrString struct {
|
|
Kind IntstrKind
|
|
IntVal int
|
|
StrVal string
|
|
}
|
|
|
|
type IntstrKind int
|
|
|
|
const (
|
|
IntstrInt IntstrKind = iota
|
|
IntstrString
|
|
)
|
|
|
|
func (intstr *IntOrString) SetYAML(tag string, value interface{}) bool {
|
|
if intVal, ok := value.(int); ok {
|
|
intstr.Kind = IntstrInt
|
|
intstr.IntVal = intVal
|
|
return true
|
|
}
|
|
if strVal, ok := value.(string); ok {
|
|
intstr.Kind = IntstrString
|
|
intstr.StrVal = strVal
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (intstr IntOrString) GetYAML() (tag string, value interface{}) {
|
|
switch intstr.Kind {
|
|
case IntstrInt:
|
|
value = intstr.IntVal
|
|
case IntstrString:
|
|
value = intstr.StrVal
|
|
default:
|
|
panic("impossible IntOrString.Kind")
|
|
}
|
|
return
|
|
}
|
|
|
|
func (intstr *IntOrString) UnmarshalJSON(value []byte) error {
|
|
if value[0] == '"' {
|
|
intstr.Kind = IntstrString
|
|
return json.Unmarshal(value, &intstr.StrVal)
|
|
}
|
|
intstr.Kind = IntstrInt
|
|
return json.Unmarshal(value, &intstr.IntVal)
|
|
}
|
|
|
|
func (intstr IntOrString) MarshalJSON() ([]byte, error) {
|
|
switch intstr.Kind {
|
|
case IntstrInt:
|
|
return json.Marshal(intstr.IntVal)
|
|
case IntstrString:
|
|
return json.Marshal(intstr.StrVal)
|
|
default:
|
|
panic("impossible IntOrString.Kind")
|
|
}
|
|
}
|