feat(ctl): add call to retreive a key

paralin · Dec 1, 2016 · 7ed1c9b · 7ed1c9b
1 parent fb97390
commit 7ed1c9b
Show file tree

Hide file tree

Showing 3 changed files with 120 additions and 30 deletions.
diff --git a/ctl/ctl-service.pb.go b/ctl/ctl-service.pb.go
diff --git a/ctl/ctl-service.proto b/ctl/ctl-service.proto
@@ -40,9 +40,18 @@ message GetGrantsResponse {
   repeated data.SignedData grants = 1;
 }
 
+message GetKeyRequest {
+  string key = 1;
+}
+
+message GetKeyResponse {
+  tx.Transaction transaction = 1;
+}
+
 service ControlService {
   rpc PutGrant(PutGrantRequest) returns (PutGrantResponse) {}
   rpc BuildTransaction(BuildTransactionRequest) returns (BuildTransactionResponse) {}
   rpc PutTransaction(PutTransactionRequest) returns (PutTransactionResponse) {}
   rpc GetGrants(GetGrantsRequest) returns (GetGrantsResponse) {}
+  rpc GetKey(GetKeyRequest) returns (GetKeyResponse) {}
 }
diff --git a/ctl/server.go b/ctl/server.go
@@ -7,6 +7,7 @@ import (
 	"time"
 
 	log "github.com/Sirupsen/logrus"
+	"github.com/boltdb/bolt"
 	"github.com/fuserobotics/kvgossip/data"
 	"github.com/fuserobotics/kvgossip/db"
 	"github.com/fuserobotics/kvgossip/grant"
@@ -116,6 +117,21 @@ func (ct *CtlServer) PutTransaction(ctx context.Context, req *PutTransactionRequ
 	return &PutTransactionResponse{}, nil
 }
 
+func (ct *CtlServer) GetKey(ctx context.Context, req *GetKeyRequest) (*GetKeyResponse, error) {
+	if req.Key == "" {
+		return nil, errors.New("Must specify key.")
+	}
+	res := &GetKeyResponse{}
+	err := ct.DB.DB.View(func(tx *bolt.Tx) error {
+		res.Transaction = ct.DB.GetTransaction(tx, req.Key)
+		return nil
+	})
+	if err != nil {
+		return nil, err
+	}
+	return res, nil
+}
+
 func (ct *CtlServer) Start(listen string) error {
 	if ct.server != nil {
 		return nil