refactor: remove unnessary bool return

siyul-park · Oct 3, 2024 · 912997b · 912997b
1 parent 8dd63a0
commit 912997b
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 29 deletions.
diff --git a/pkg/symbol/loader.go b/pkg/symbol/loader.go
@@ -85,8 +85,8 @@ func (l *Loader) Load(ctx context.Context, specs ...spec.Spec) error {
 			continue
 		}
 
-		sb, ok := l.table.Lookup(decode.GetID())
-		if !ok || !reflect.DeepEqual(sb.Spec, decode) {
+		sb := l.table.Lookup(decode.GetID())
+		if sb == nil || !reflect.DeepEqual(sb.Spec, decode) {
 			n, err := l.scheme.Compile(decode)
 			if err != nil {
 				errs = append(errs, err)
@@ -108,8 +108,8 @@ func (l *Loader) Load(ctx context.Context, specs ...spec.Spec) error {
 	}
 
 	for _, id := range l.table.Keys() {
-		sb, ok := l.table.Lookup(id)
-		if ok && len(resource.Match(sb.Spec, examples...)) > 0 {
+		sb := l.table.Lookup(id)
+		if sb != nil && len(resource.Match(sb.Spec, examples...)) > 0 {
 			ok := false
 			for _, s := range symbols {
 				if s.ID() == id {

diff --git a/pkg/symbol/loader_test.go b/pkg/symbol/loader_test.go
@@ -77,15 +77,9 @@ func TestLoader_Load(t *testing.T) {
 
 		err := loader.Load(ctx, meta1, meta2, meta3)
 		assert.NoError(t, err)
-
-		_, ok := table.Lookup(meta1.GetID())
-		assert.True(t, ok)
-
-		_, ok = table.Lookup(meta2.GetID())
-		assert.True(t, ok)
-
-		_, ok = table.Lookup(meta3.GetID())
-		assert.True(t, ok)
+		assert.NotNil(t, table.Lookup(meta1.GetID()))
+		assert.NotNil(t, table.Lookup(meta2.GetID()))
+		assert.NotNil(t, table.Lookup(meta3.GetID()))
 	})
 
 	t.Run("ReloadWithSameID", func(t *testing.T) {
@@ -166,9 +160,7 @@ func TestLoader_Load(t *testing.T) {
 
 		err = loader.Load(ctx, meta)
 		assert.NoError(t, err)
-
-		_, ok := table.Lookup(meta.GetID())
-		assert.False(t, ok)
+		assert.NotNil(t, table.Lookup(meta.GetID()))
 	})
 
 	t.Run("LoadMultipleSecrets", func(t *testing.T) {
@@ -214,9 +206,7 @@ func TestLoader_Load(t *testing.T) {
 
 		err := loader.Load(ctx, meta)
 		assert.NoError(t, err)
-
-		_, ok := table.Lookup(meta.GetID())
-		assert.True(t, ok)
+		assert.NotNil(t, table.Lookup(meta.GetID()))
 	})
 
 	t.Run("LoadNonExistSecret", func(t *testing.T) {
@@ -252,8 +242,6 @@ func TestLoader_Load(t *testing.T) {
 
 		err := loader.Load(ctx, meta)
 		assert.Error(t, err)
-
-		_, ok := table.Lookup(meta.GetID())
-		assert.False(t, ok)
+		assert.NotNil(t, table.Lookup(meta.GetID()))
 	})
 }
diff --git a/pkg/symbol/table.go b/pkg/symbol/table.go
@@ -74,12 +74,11 @@ func (t *Table) Free(id uuid.UUID) (bool, error) {
 }
 
 // Lookup retrieves a symbol from the table by its ID.
-func (t *Table) Lookup(id uuid.UUID) (*Symbol, bool) {
+func (t *Table) Lookup(id uuid.UUID) *Symbol {
 	t.mu.RLock()
 	defer t.mu.RUnlock()
 
-	sb, ok := t.symbols[id]
-	return sb, ok
+	return t.symbols[id]
 }
 
 // Keys returns all IDs of symbols in the table.

diff --git a/pkg/symbol/table_test.go b/pkg/symbol/table_test.go
@@ -396,12 +396,10 @@ func TestTable_LookupByID(t *testing.T) {
 	}
 
 	sb := &Symbol{Spec: meta, Node: node.NewOneToOneNode(nil)}
+
 	err := tb.Insert(sb)
 	assert.NoError(t, err)
-
-	r, ok := tb.Lookup(sb.ID())
-	assert.True(t, ok)
-	assert.Equal(t, sb, r)
+	assert.Equal(t, sb, tb.Lookup(sb.ID()))
 }
 
 func TestTable_Keys(t *testing.T) {