Method Put

twopc/twopc.go:170–213

Put initiates a 2PC process to apply given WriteBatch on all workers.

(workers []Worker, wb WriteBatch)

// Put initiates a 2PC process to apply given WriteBatch on all workers.
func (c *Coordinator) Put(workers []Worker, wb WriteBatch) (result interface{}, err error) {
	// Initiate phase one: ask nodes to prepare for progress
	ctx, cancel := context.WithTimeout(context.Background(), c.option.timeout)
	defer cancel()

	if c.option.beforePrepare != nil {
		if err = c.option.beforePrepare(ctx); err != nil {
			log.WithError(err).Debug("before prepared failed")
			return
		}
	}

	// Check prepare results and initiate phase two
	if err = c.prepare(ctx, workers, wb); err != nil {
		goto ROLLBACK
	}

	if c.option.beforeCommit != nil {
		if err = c.option.beforeCommit(ctx); err != nil {
			log.WithError(err).Debug("before commit failed")
			goto ROLLBACK
		}
	}

	result, err = c.commit(ctx, workers, wb)

	if c.option.afterCommit != nil {
		if err = c.option.afterCommit(ctx); err != nil {
			log.WithError(err).Debug("after commit failed")
		}
	}

	return

ROLLBACK:
	if c.option.beforeRollback != nil {
		// ignore rollback fail options
		c.option.beforeRollback(ctx)
	}

	c.rollback(ctx, workers, wb)

	return
}