mirror of
https://github.com/Oxalide/vsphere-influxdb-go.git
synced 2023-10-10 11:36:51 +00:00
add vendoring with go dep
This commit is contained in:
195
vendor/github.com/vmware/govmomi/event/processor.go
generated
vendored
Normal file
195
vendor/github.com/vmware/govmomi/event/processor.go
generated
vendored
Normal file
@@ -0,0 +1,195 @@
|
||||
/*
|
||||
Copyright (c) 2016 VMware, 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 event
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
|
||||
"github.com/vmware/govmomi/property"
|
||||
"github.com/vmware/govmomi/view"
|
||||
"github.com/vmware/govmomi/vim25/types"
|
||||
)
|
||||
|
||||
const latestPageProp = "latestPage"
|
||||
|
||||
type tailInfo struct {
|
||||
t *eventTailer
|
||||
obj types.ManagedObjectReference
|
||||
collector types.ManagedObjectReference
|
||||
}
|
||||
|
||||
type eventProcessor struct {
|
||||
mgr Manager
|
||||
pageSize int32
|
||||
tailers map[types.ManagedObjectReference]*tailInfo // tailers by collector ref
|
||||
callback func(types.ManagedObjectReference, []types.BaseEvent) error
|
||||
}
|
||||
|
||||
func newEventProcessor(mgr Manager, pageSize int32, callback func(types.ManagedObjectReference, []types.BaseEvent) error) *eventProcessor {
|
||||
return &eventProcessor{
|
||||
mgr: mgr,
|
||||
tailers: make(map[types.ManagedObjectReference]*tailInfo),
|
||||
callback: callback,
|
||||
pageSize: pageSize,
|
||||
}
|
||||
}
|
||||
|
||||
func (p *eventProcessor) addObject(ctx context.Context, obj types.ManagedObjectReference) error {
|
||||
filter := types.EventFilterSpec{
|
||||
Entity: &types.EventFilterSpecByEntity{
|
||||
Entity: obj,
|
||||
Recursion: types.EventFilterSpecRecursionOptionAll,
|
||||
},
|
||||
}
|
||||
|
||||
collector, err := p.mgr.CreateCollectorForEvents(ctx, filter)
|
||||
if err != nil {
|
||||
return fmt.Errorf("[%#v] %s", obj, err)
|
||||
}
|
||||
|
||||
err = collector.SetPageSize(ctx, p.pageSize)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
p.tailers[collector.Reference()] = &tailInfo{
|
||||
t: newEventTailer(),
|
||||
obj: obj,
|
||||
collector: collector.Reference(),
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *eventProcessor) run(ctx context.Context, tail bool) error {
|
||||
if len(p.tailers) == 0 {
|
||||
return nil
|
||||
}
|
||||
|
||||
var err error
|
||||
var collectors []types.ManagedObjectReference
|
||||
for _, t := range p.tailers {
|
||||
collectors = append(collectors, t.collector)
|
||||
}
|
||||
|
||||
if len(p.tailers) > 1 {
|
||||
// create and populate a ListView
|
||||
viewMgr := view.NewManager(p.mgr.Client())
|
||||
var listView *view.ListView
|
||||
listView, err = viewMgr.CreateListView(ctx, collectors)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
count := 0
|
||||
// Retrieve the property from the objects in the ListView
|
||||
err = property.WaitForView(ctx, property.DefaultCollector(p.mgr.Client()), listView.Reference(), collectors[0], []string{latestPageProp}, func(c types.ManagedObjectReference, pc []types.PropertyChange) bool {
|
||||
if err = p.process(c, pc); err != nil {
|
||||
return false
|
||||
}
|
||||
|
||||
count++
|
||||
if count == len(collectors) && !tail {
|
||||
return true
|
||||
}
|
||||
|
||||
return false
|
||||
})
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
// only one object to follow
|
||||
err = property.Wait(ctx, property.DefaultCollector(p.mgr.Client()), collectors[0], []string{latestPageProp}, func(pc []types.PropertyChange) bool {
|
||||
if err = p.process(collectors[0], pc); err != nil {
|
||||
return false
|
||||
}
|
||||
|
||||
if !tail {
|
||||
return true
|
||||
}
|
||||
|
||||
return false
|
||||
})
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func (p *eventProcessor) process(c types.ManagedObjectReference, pc []types.PropertyChange) error {
|
||||
t := p.tailers[c]
|
||||
if t == nil {
|
||||
return fmt.Errorf("unknown collector %s", c.String())
|
||||
}
|
||||
|
||||
for _, u := range pc {
|
||||
if u.Name != latestPageProp {
|
||||
continue
|
||||
}
|
||||
|
||||
evs := t.t.newEvents(u.Val.(types.ArrayOfEvent).Event)
|
||||
if len(evs) == 0 {
|
||||
continue
|
||||
}
|
||||
|
||||
if err := p.callback(t.obj, evs); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
const invalidKey = int32(-1)
|
||||
|
||||
type eventTailer struct {
|
||||
lastKey int32
|
||||
}
|
||||
|
||||
func newEventTailer() *eventTailer {
|
||||
return &eventTailer{
|
||||
lastKey: invalidKey,
|
||||
}
|
||||
}
|
||||
|
||||
func (t *eventTailer) newEvents(evs []types.BaseEvent) []types.BaseEvent {
|
||||
var ret []types.BaseEvent
|
||||
if t.lastKey == invalidKey {
|
||||
ret = evs
|
||||
} else {
|
||||
found := false
|
||||
for i := range evs {
|
||||
if evs[i].GetEvent().Key != t.lastKey {
|
||||
continue
|
||||
}
|
||||
|
||||
found = true
|
||||
ret = evs[:i]
|
||||
break
|
||||
}
|
||||
|
||||
if !found {
|
||||
ret = evs
|
||||
}
|
||||
}
|
||||
|
||||
if len(ret) > 0 {
|
||||
t.lastKey = ret[0].GetEvent().Key
|
||||
}
|
||||
|
||||
return ret
|
||||
}
|
||||
Reference in New Issue
Block a user