e620833c9e
With this changeset, we provide an implementation for a transaction event API. The goal is to support the emission of events with transactions and sub-transactions, allowing a process post events that may be rolled back later due to an error in the entire process. When journaled, a consumer will be able to commit and rollback with the same behavior as the producer. Events are left to full definition by each component. We may require more structure in the future to ensure consistency but we need more use case before making decisions in that direction. Events may be organized by a topic. A topic defines a single stream of messages, that could be associated with a specific component. The topic defines a grouped stream that may be compacted as one. Only the contextual API is implemented here. After using, we will make a determination to see how useful this across the board. Signed-off-by: Stephen J Day <stephen.day@docker.com>
94 lines
2.2 KiB
Go
94 lines
2.2 KiB
Go
package events
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync"
|
|
"sync/atomic"
|
|
"time"
|
|
)
|
|
|
|
var txCounter int64 // replace this with something that won't break
|
|
|
|
// nextTXID provides the next transaction identifier.
|
|
func nexttxID() int64 {
|
|
// TODO(stevvooe): Need to coordinate this with existing transaction logs.
|
|
// For now, this is a toy, but not a racy one.
|
|
return atomic.AddInt64(&txCounter, 1)
|
|
}
|
|
|
|
type transaction struct {
|
|
id int64
|
|
parent *transaction // if nil, no parent transaction
|
|
finish sync.Once
|
|
start, end time.Time // informational
|
|
}
|
|
|
|
// begin creates a sub-transaction.
|
|
func (tx *transaction) begin(poster Poster) *transaction {
|
|
id := nexttxID()
|
|
|
|
child := &transaction{
|
|
id: id,
|
|
parent: tx,
|
|
start: time.Now(),
|
|
}
|
|
|
|
// post the transaction started event
|
|
poster.Post(child.makeTransactionEvent("begin")) // tranactions are really just events
|
|
|
|
return child
|
|
}
|
|
|
|
// commit writes out the transaction.
|
|
func (tx *transaction) commit(poster Poster) {
|
|
tx.finish.Do(func() {
|
|
tx.end = time.Now()
|
|
poster.Post(tx.makeTransactionEvent("commit"))
|
|
})
|
|
}
|
|
|
|
func (tx *transaction) rollback(poster Poster, cause error) {
|
|
tx.finish.Do(func() {
|
|
tx.end = time.Now()
|
|
event := tx.makeTransactionEvent("rollback")
|
|
event = fmt.Sprintf("%s error=%q", event, cause.Error())
|
|
poster.Post(event)
|
|
})
|
|
}
|
|
|
|
func (tx *transaction) makeTransactionEvent(status string) Event {
|
|
// TODO(stevvooe): obviously, we need more structure than this.
|
|
event := fmt.Sprintf("%v %v", status, tx.id)
|
|
if tx.parent != nil {
|
|
event += " parent=" + fmt.Sprint(tx.parent.id)
|
|
}
|
|
|
|
return event
|
|
}
|
|
|
|
type txKey struct{}
|
|
|
|
func getTx(ctx context.Context) (*transaction, bool) {
|
|
tx := ctx.Value(txKey{})
|
|
if tx == nil {
|
|
return nil, false
|
|
}
|
|
|
|
return tx.(*transaction), true
|
|
}
|
|
|
|
// WithTx returns a new context with an event transaction, such that events
|
|
// posted to the underlying context will be committed to the event log as a
|
|
// group, organized by a transaction id, when commit is called.
|
|
func WithTx(pctx context.Context) (ctx context.Context, commit func(), rollback func(err error)) {
|
|
poster := G(pctx)
|
|
parent, _ := getTx(pctx)
|
|
tx := parent.begin(poster)
|
|
|
|
return context.WithValue(pctx, txKey{}, tx), func() {
|
|
tx.commit(poster)
|
|
}, func(err error) {
|
|
tx.rollback(poster, err)
|
|
}
|
|
}
|