From 4871a4a5f35b350f80d9ed1b0e10f517c3ef803d Mon Sep 17 00:00:00 2001
From: Gyu-Ho Lee <gyuhox@gmail.com>
Date: Fri, 30 Sep 2016 10:09:28 -0700
Subject: [PATCH] lease: remove redundant get method

---
 lease/lessor.go      | 14 +-------------
 lease/lessor_test.go | 10 +++++-----
 2 files changed, 6 insertions(+), 18 deletions(-)

diff --git a/lease/lessor.go b/lease/lessor.go
index 5ebe7290c..4778c4c48 100644
--- a/lease/lessor.go
+++ b/lease/lessor.go
@@ -274,10 +274,7 @@ func (le *lessor) Renew(id LeaseID) (int64, error) {
 func (le *lessor) Lookup(id LeaseID) *Lease {
 	le.mu.Lock()
 	defer le.mu.Unlock()
-	if l, ok := le.leaseMap[id]; ok {
-		return l
-	}
-	return nil
+	return le.leaseMap[id]
 }
 
 func (le *lessor) Promote(extend time.Duration) {
@@ -408,15 +405,6 @@ func (le *lessor) findExpiredLeases() []*Lease {
 	return leases
 }
 
-// get gets the lease with given id.
-// get is a helper function for testing, at least for now.
-func (le *lessor) get(id LeaseID) *Lease {
-	le.mu.Lock()
-	defer le.mu.Unlock()
-
-	return le.leaseMap[id]
-}
-
 func (le *lessor) initAndRecover() {
 	tx := le.b.BatchTx()
 	tx.Lock()
diff --git a/lease/lessor_test.go b/lease/lessor_test.go
index adc682c81..a3a15a958 100644
--- a/lease/lessor_test.go
+++ b/lease/lessor_test.go
@@ -43,7 +43,7 @@ func TestLessorGrant(t *testing.T) {
 	if err != nil {
 		t.Fatalf("could not grant lease 1 (%v)", err)
 	}
-	gl := le.get(l.ID)
+	gl := le.Lookup(l.ID)
 
 	if !reflect.DeepEqual(gl, l) {
 		t.Errorf("lease = %v, want %v", gl, l)
@@ -107,7 +107,7 @@ func TestLessorRevoke(t *testing.T) {
 		t.Fatal("failed to revoke lease:", err)
 	}
 
-	if le.get(l.ID) != nil {
+	if le.Lookup(l.ID) != nil {
 		t.Errorf("got revoked lease %x", l.ID)
 	}
 
@@ -149,7 +149,7 @@ func TestLessorRenew(t *testing.T) {
 		t.Errorf("ttl = %d, want %d", ttl, l.TTL)
 	}
 
-	l = le.get(l.ID)
+	l = le.Lookup(l.ID)
 	if l.expiry.Sub(time.Now()) < 9*time.Second {
 		t.Errorf("failed to renew the lease")
 	}
@@ -210,12 +210,12 @@ func TestLessorRecover(t *testing.T) {
 
 	// Create a new lessor with the same backend
 	nle := newLessor(be, minLeaseTTL)
-	nl1 := nle.get(l1.ID)
+	nl1 := nle.Lookup(l1.ID)
 	if nl1 == nil || nl1.TTL != l1.TTL {
 		t.Errorf("nl1 = %v, want nl1.TTL= %d", nl1.TTL, l1.TTL)
 	}
 
-	nl2 := nle.get(l2.ID)
+	nl2 := nle.Lookup(l2.ID)
 	if nl2 == nil || nl2.TTL != l2.TTL {
 		t.Errorf("nl2 = %v, want nl2.TTL= %d", nl2.TTL, l2.TTL)
 	}