update Context

.gitignore

@@ -12,4 +12,6 @@
 *.out
 
 # goland
-.idea
+.idea
+
+internal

aggregate.go

@@ -9,6 +9,7 @@ func NewAggregateId() (id int64) {
 
 func ApplyAggregateChange(ctx context.Context, aggregate Aggregate, change aggregateChange) {
 	aggregate.Apply(aggregate, change)
+	return
 }
 
 type Aggregate interface {

context.go

@@ -2,15 +2,26 @@ package gddd
 
 import (
 	"context"
-	"database/sql"
 )
 
 type Context struct {
-	ctx context.Context
-	tx  *sql.Tx
+	ctx        context.Context
+	repository *Repository
 }
 
-func (c *Context) Apply(aggregate Aggregate, change aggregateChange) error {
+func NewContext(ctx context.Context, repository *Repository) Context {
+	return Context{ctx: ctx, repository: repository}
+}
+
+func (c *Context) Load(aggregate Aggregate) (has bool, err error) {
+	return c.repository.Load(c.ctx, aggregate)
+}
+
+func (c *Context) Save(aggregates ...Aggregate) (ok bool, err error) {
+	return c.repository.Save(c.ctx, aggregates...)
+}
+
+func (c *Context) Apply(aggregate Aggregate, change aggregateChange) {
 	ApplyAggregateChange(c.ctx, aggregate, change)
-	return nil
+	return
 }

repository.go

@@ -68,6 +68,11 @@ type Repository struct {
 	saveListener RepositorySaveListener
 }
 
+func (r *Repository) SetSaveListener(ctx context.Context, saveListener RepositorySaveListener) (err error) {
+	r.saveListener = saveListener
+	return
+}
+
 func (r *Repository) RegisterAggregates(ctx context.Context, aggregates ...Aggregate) (err error) {
 	for _, aggregate := range aggregates {
 		aggregateName := getAggregateName(aggregate)