Method Prepare

twopc/twopc_test.go:222–274

(ctx context.Context, wb twopc.WriteBatch)

}

func (r *RaftNode) Prepare(ctx context.Context, wb twopc.WriteBatch) (err error) {
	log.WithFields(log.Fields{
		"addr":  r.addr,
		"phase": "prepare",
	}).Debug("executing 2pc")
	defer log.WithFields(log.Fields{
		"addr":  r.addr,
		"phase": "prepare",
	}).WithError(err).Debug("2pc result")

	rwb, ok := wb.(*RaftWriteBatchReq)

	if !ok {
		err = errors.New("unexpected WriteBatch type")
		return err
	}

	cipher := etls.NewCipher([]byte(pass))
	conn, err := etls.Dial("tcp", r.addr, cipher)

	if err != nil {
		return err
	}

	muxconn, err := mux.NewOneOffMuxConn(conn)
	if err != nil {
		return
	}

	client := rpc.NewClient(muxconn)
	d, ok := ctx.Deadline()

	if ok {
		err = muxconn.SetDeadline(d)

		if err != nil {
			return err
		}
	}

	resp := new(RaftWriteBatchResp)
	err = client.Call("Raft.RPCPrepare", &rwb, resp)

	if err != nil {
		return err
	}

	if resp.ErrCode > 0 {
		err = fmt.Errorf(resp.ErrString)
	}

	return err
}

func (r *RaftNode) Commit(ctx context.Context, wb twopc.WriteBatch) (result interface{}, err error) {
	log.Debugf("executing 2pc: addr = %s, phase = commit", r.addr)