Move deps from _workspace/ to vendor/
godep restore pushd $GOPATH/src/github.com/appc/spec git co master popd go get go4.org/errorutil rm -rf Godeps godep save ./... git add vendor git add -f $(git ls-files --other vendor/) git co -- Godeps/LICENSES Godeps/.license_file_state Godeps/OWNERS
This commit is contained in:
152
vendor/github.com/coreos/etcd/alarm/alarms.go
generated
vendored
Normal file
152
vendor/github.com/coreos/etcd/alarm/alarms.go
generated
vendored
Normal file
@@ -0,0 +1,152 @@
|
||||
// Copyright 2016 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 alarm manages health status alarms in etcd.
|
||||
package alarm
|
||||
|
||||
import (
|
||||
"sync"
|
||||
|
||||
pb "github.com/coreos/etcd/etcdserver/etcdserverpb"
|
||||
"github.com/coreos/etcd/pkg/types"
|
||||
"github.com/coreos/etcd/storage/backend"
|
||||
"github.com/coreos/pkg/capnslog"
|
||||
)
|
||||
|
||||
var (
|
||||
alarmBucketName = []byte("alarm")
|
||||
plog = capnslog.NewPackageLogger("github.com/coreos/etcd", "alarm")
|
||||
)
|
||||
|
||||
type BackendGetter interface {
|
||||
Backend() backend.Backend
|
||||
}
|
||||
|
||||
type alarmSet map[types.ID]*pb.AlarmMember
|
||||
|
||||
// AlarmStore persists alarms to the backend.
|
||||
type AlarmStore struct {
|
||||
mu sync.Mutex
|
||||
types map[pb.AlarmType]alarmSet
|
||||
|
||||
bg BackendGetter
|
||||
}
|
||||
|
||||
func NewAlarmStore(bg BackendGetter) (*AlarmStore, error) {
|
||||
ret := &AlarmStore{types: make(map[pb.AlarmType]alarmSet), bg: bg}
|
||||
err := ret.restore()
|
||||
return ret, err
|
||||
}
|
||||
|
||||
func (a *AlarmStore) Activate(id types.ID, at pb.AlarmType) *pb.AlarmMember {
|
||||
a.mu.Lock()
|
||||
defer a.mu.Unlock()
|
||||
|
||||
newAlarm := &pb.AlarmMember{MemberID: uint64(id), Alarm: at}
|
||||
if m := a.addToMap(newAlarm); m != newAlarm {
|
||||
return m
|
||||
}
|
||||
|
||||
v, err := newAlarm.Marshal()
|
||||
if err != nil {
|
||||
plog.Panicf("failed to marshal alarm member")
|
||||
}
|
||||
|
||||
b := a.bg.Backend()
|
||||
b.BatchTx().Lock()
|
||||
b.BatchTx().UnsafePut(alarmBucketName, v, nil)
|
||||
b.BatchTx().Unlock()
|
||||
|
||||
return newAlarm
|
||||
}
|
||||
|
||||
func (a *AlarmStore) Deactivate(id types.ID, at pb.AlarmType) *pb.AlarmMember {
|
||||
a.mu.Lock()
|
||||
defer a.mu.Unlock()
|
||||
|
||||
t := a.types[at]
|
||||
if t == nil {
|
||||
t = make(alarmSet)
|
||||
a.types[at] = t
|
||||
}
|
||||
m := t[id]
|
||||
if m == nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
delete(t, id)
|
||||
|
||||
v, err := m.Marshal()
|
||||
if err != nil {
|
||||
plog.Panicf("failed to marshal alarm member")
|
||||
}
|
||||
|
||||
b := a.bg.Backend()
|
||||
b.BatchTx().Lock()
|
||||
b.BatchTx().UnsafeDelete(alarmBucketName, v)
|
||||
b.BatchTx().Unlock()
|
||||
|
||||
return m
|
||||
}
|
||||
|
||||
func (a *AlarmStore) Get(at pb.AlarmType) (ret []*pb.AlarmMember) {
|
||||
a.mu.Lock()
|
||||
defer a.mu.Unlock()
|
||||
if at == pb.AlarmType_NONE {
|
||||
for _, t := range a.types {
|
||||
for _, m := range t {
|
||||
ret = append(ret, m)
|
||||
}
|
||||
}
|
||||
return ret
|
||||
}
|
||||
for _, m := range a.types[at] {
|
||||
ret = append(ret, m)
|
||||
}
|
||||
return ret
|
||||
}
|
||||
|
||||
func (a *AlarmStore) restore() error {
|
||||
b := a.bg.Backend()
|
||||
tx := b.BatchTx()
|
||||
|
||||
tx.Lock()
|
||||
tx.UnsafeCreateBucket(alarmBucketName)
|
||||
err := tx.UnsafeForEach(alarmBucketName, func(k, v []byte) error {
|
||||
var m pb.AlarmMember
|
||||
if err := m.Unmarshal(k); err != nil {
|
||||
return err
|
||||
}
|
||||
a.addToMap(&m)
|
||||
return nil
|
||||
})
|
||||
tx.Unlock()
|
||||
|
||||
b.ForceCommit()
|
||||
return err
|
||||
}
|
||||
|
||||
func (a *AlarmStore) addToMap(newAlarm *pb.AlarmMember) *pb.AlarmMember {
|
||||
t := a.types[newAlarm.Alarm]
|
||||
if t == nil {
|
||||
t = make(alarmSet)
|
||||
a.types[newAlarm.Alarm] = t
|
||||
}
|
||||
m := t[types.ID(newAlarm.MemberID)]
|
||||
if m != nil {
|
||||
return m
|
||||
}
|
||||
t[types.ID(newAlarm.MemberID)] = newAlarm
|
||||
return newAlarm
|
||||
}
|
||||
Reference in New Issue
Block a user