From d6d90bf28e76507366f70efe3dc54093e998c49f Mon Sep 17 00:00:00 2001
From: Robin Tang <rtang.cs@gmail.com>
Date: Thu, 17 Oct 2024 08:39:20 -0700
Subject: [PATCH] Clean up.

---
 lib/cdc/util/relational_event.go | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/lib/cdc/util/relational_event.go b/lib/cdc/util/relational_event.go
index 2c500f412..bd58fc708 100644
--- a/lib/cdc/util/relational_event.go
+++ b/lib/cdc/util/relational_event.go
@@ -77,10 +77,11 @@ func (s *SchemaEventPayload) GetTableName() string {
 }
 
 func (s *SchemaEventPayload) GetData(pkMap map[string]any, tc kafkalib.TopicConfig) (map[string]any, error) {
+	var err error
 	var retMap map[string]any
-	if len(s.Payload.After) == 0 {
+	switch s.Operation() {
+	case "d":
 		if len(s.Payload.Before) > 0 {
-			var err error
 			retMap, err = s.parseAndMutateMapInPlace(s.Payload.Before, debezium.Before)
 			if err != nil {
 				return nil, err
@@ -100,8 +101,7 @@ func (s *SchemaEventPayload) GetData(pkMap map[string]any, tc kafkalib.TopicConf
 		for k, v := range pkMap {
 			retMap[k] = v
 		}
-	} else {
-		var err error
+	case "r", "u", "c":
 		retMap, err = s.parseAndMutateMapInPlace(s.Payload.After, debezium.After)
 		if err != nil {
 			return nil, err