Skip to content

Commit

Permalink
client: honor last transaction IDs
Browse files Browse the repository at this point in the history
Signed-off-by: Dan Williams <[email protected]>
  • Loading branch information
dcbw committed Jan 19, 2022
1 parent 22bd5be commit fa20454
Show file tree
Hide file tree
Showing 3 changed files with 123 additions and 17 deletions.
37 changes: 37 additions & 0 deletions cache/cache.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@ import (
"github.com/ovn-org/libovsdb/mapper"
"github.com/ovn-org/libovsdb/model"
"github.com/ovn-org/libovsdb/ovsdb"
"github.com/ovn-org/libovsdb/util"
)

const (
Expand Down Expand Up @@ -703,6 +704,42 @@ func (t *TableCache) Purge(dbModel model.DatabaseModel) {
}
}

// PurgeTable drops all data in the given table's cache and reinitializes it using the
// provided database model
func (t *TableCache) PurgeTable(dbModel model.DatabaseModel, name string) error {
return t.PurgeTableRows(dbModel, name, nil)
}

// PurgeTableRows drops all rows in the given table's cache that match the given conditions
func (t *TableCache) PurgeTableRows(dbModel model.DatabaseModel, name string, conditions []ovsdb.Condition) error {
t.mutex.Lock()
defer t.mutex.Unlock()
t.dbModel = dbModel
tableTypes := t.dbModel.Types()
dataType, ok := tableTypes[name]
if !ok {
return fmt.Errorf("table %s not found", name)
}
if len(conditions) == 0 {
t.cache[name] = newRowCache(name, t.dbModel, dataType)
return nil
}

r := t.cache[name]
rows, err := r.RowsByCondition(conditions)
if err != nil {
return err
}
delErrors := []error{}
for uuid := range rows {
if err := r.Delete(uuid); err != nil {
delErrors = append(delErrors, fmt.Errorf("failed to delete %s: %w", uuid, err))
}
}

return util.CombineErrors(delErrors, "failed to delete rows")
}

// AddEventHandler registers the supplied EventHandler to receive cache events
func (t *TableCache) AddEventHandler(handler EventHandler) {
t.eventProcessor.AddEventHandler(handler)
Expand Down
83 changes: 66 additions & 17 deletions client/client.go
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ import (
"github.com/ovn-org/libovsdb/model"
"github.com/ovn-org/libovsdb/ovsdb"
"github.com/ovn-org/libovsdb/ovsdb/serverdb"
"github.com/ovn-org/libovsdb/util"
)

// Constants defined for libovsdb
Expand Down Expand Up @@ -199,6 +200,65 @@ func (o *ovsdbClient) Connect(ctx context.Context) error {
return nil
}

func (db *database) purge(name string) error {
// If a table has any !Since monitors or has no conditions, purge it
// If a table only has Since monitors with conditions, purge only rows that match the conditions
type purge struct {
conditions []ovsdb.Condition
purgeAll bool
}

purges := make(map[string]*purge)
for _, monitor := range db.monitors {
for _, tm := range monitor.Tables {
p, ok := purges[tm.Table]
if !ok {
p = &purge{}
purges[tm.Table] = p
}
if monitor.Method == ovsdb.ConditionalMonitorSinceRPC {
model, err := db.model.NewModel(tm.Table)
if err != nil {
p.purgeAll = true
continue
}
info, err := db.model.NewModelInfo(model)
if err != nil {
p.purgeAll = true
continue
}
ovsdbCond, err := db.model.Mapper.NewCondition(info, tm.Condition.Field, tm.Condition.Function, tm.Condition.Value)
if err != nil {
p.purgeAll = true
continue
}
p.conditions = append(p.conditions, *ovsdbCond)
} else {
p.purgeAll = true
}
}
}
if len(purges) == 0 {
db.cache.Purge(db.model)
return nil
}

var purgeErrors []error
for name, p := range purges {
if p.purgeAll {
if err := db.cache.PurgeTable(db.model, name); err != nil {
purgeErrors = append(purgeErrors, err)
}
} else {
if err := db.cache.PurgeTableRows(db.model, name, p.conditions); err != nil {
purgeErrors = append(purgeErrors, err)
}
}
}

return util.CombineErrors(purgeErrors, fmt.Sprintf("failed to purge database %s", name))
}

func (o *ovsdbClient) connect(ctx context.Context, reconnect bool) error {
o.rpcMutex.Lock()
defer o.rpcMutex.Unlock()
Expand Down Expand Up @@ -226,15 +286,7 @@ func (o *ovsdbClient) connect(ctx context.Context, reconnect bool) error {
}

if !connected {
if len(connectErrors) == 1 {
return connectErrors[0]
}
var combined []string
for _, e := range connectErrors {
combined = append(combined, e.Error())
}

return fmt.Errorf("unable to connect to any endpoints: %s", strings.Join(combined, ". "))
return util.CombineErrors(connectErrors, "unable to connect to any endpoints")
}

// if we're reconnecting, re-start all the monitors
Expand All @@ -243,6 +295,10 @@ func (o *ovsdbClient) connect(ctx context.Context, reconnect bool) error {
for dbName, db := range o.databases {
db.monitorsMutex.Lock()
defer db.monitorsMutex.Unlock()

if err := db.purge(dbName); err != nil {
o.logger.V(3).Error(err, "failed to purge")
}
for id, request := range db.monitors {
err := o.monitor(ctx, MonitorCookie{DatabaseName: dbName, ID: id}, true, request)
if err != nil {
Expand Down Expand Up @@ -349,8 +405,6 @@ func (o *ovsdbClient) tryEndpoint(ctx context.Context, u *url.URL) error {
return err
}
db.api = newAPI(db.cache, o.logger)
} else {
db.cache.Purge(db.model)
}
db.cacheMutex.Unlock()
}
Expand Down Expand Up @@ -832,12 +886,7 @@ func (o *ovsdbClient) monitor(ctx context.Context, cookie MonitorCookie, reconne

var args []interface{}
if monitor.Method == ovsdb.ConditionalMonitorSinceRPC {
// FIXME: We should pass the monitor.LastTransactionID here
// But that would require delaying clearing the cache until
// after the monitors have been re-established - the logic
// would also need to be different for monitor and monitor_cond
// as we must always clear the cache in that instance
args = ovsdb.NewMonitorCondSinceArgs(dbName, cookie, requests, emptyUUID)
args = ovsdb.NewMonitorCondSinceArgs(dbName, cookie, requests, monitor.LastTransactionID)
} else {
args = ovsdb.NewMonitorArgs(dbName, cookie, requests)
}
Expand Down
20 changes: 20 additions & 0 deletions util/errors.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
package util

import (
"fmt"
"strings"
)

func CombineErrors(errors []error, msg string) error {
if len(errors) == 0 {
return nil
} else if len(errors) == 1 {
return errors[0]
}

var combined []string
for _, e := range errors {
combined = append(combined, e.Error())
}
return fmt.Errorf("%s: %s", msg, strings.Join(combined, ". "))
}

0 comments on commit fa20454

Please sign in to comment.