From b23d27f5078f91dec0b4e30302b893fec57ccc5b Mon Sep 17 00:00:00 2001
From: Matt Jaffee <matthew.jaffee@gmail.com>
Date: Mon, 20 Apr 2020 22:22:40 -0500
Subject: [PATCH] add cluster state validation to API methods for transactions

---
 api.go | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/api.go b/api.go
index 761a14dc5..22479c61b 100644
--- a/api.go
+++ b/api.go
@@ -1586,18 +1586,30 @@ func (api *API) PrimaryReplicaNodeURL() url.URL {
 }
 
 func (api *API) StartTransaction(ctx context.Context, id string, timeout time.Duration, exclusive bool, remote bool) (Transaction, error) {
+	if err := api.validate(apiStartTransaction); err != nil {
+		return Transaction{}, errors.Wrap(err, "validating api method")
+	}
 	return api.server.StartTransaction(ctx, id, timeout, exclusive, remote)
 }
 
 func (api *API) FinishTransaction(ctx context.Context, id string, remote bool) (Transaction, error) {
+	if err := api.validate(apiFinishTransaction); err != nil {
+		return Transaction{}, errors.Wrap(err, "validating api method")
+	}
 	return api.server.FinishTransaction(ctx, id, remote)
 }
 
 func (api *API) Transactions(ctx context.Context) (map[string]Transaction, error) {
+	if err := api.validate(apiTransactions); err != nil {
+		return nil, errors.Wrap(err, "validating api method")
+	}
 	return api.server.Transactions(ctx)
 }
 
 func (api *API) GetTransaction(ctx context.Context, id string, remote bool) (Transaction, error) {
+	if err := api.validate(apiGetTransaction); err != nil {
+		return Transaction{}, errors.Wrap(err, "validating api method")
+	}
 	return api.server.GetTransaction(ctx, id, remote)
 }
 
@@ -1648,6 +1660,10 @@ const (
 	//apiVersion // not implemented
 	apiViews
 	apiApplySchema
+	apiStartTransaction
+	apiFinishTransaction
+	apiTransactions
+	apiGetTransaction
 )
 
 var methodsCommon = map[apiMethod]struct{}{
@@ -1683,4 +1699,8 @@ var methodsNormal = map[apiMethod]struct{}{
 	apiShardNodes:           {},
 	apiViews:                {},
 	apiApplySchema:          {},
+	apiStartTransaction:     {},
+	apiFinishTransaction:    {},
+	apiTransactions:         {},
+	apiGetTransaction:       {},
 }