fix create peer handler

PeerDB-io · Mar 12, 2024 · db6b2b3 · db6b2b3
1 parent 19f5750
commit db6b2b3
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/flow/cmd/handler.go b/flow/cmd/handler.go
@@ -533,9 +533,7 @@ func (h *FlowRequestHandler) CreatePeer(
 			return wrongConfigResponse, nil
 		}
 		pgConfig := pgConfigObject.PostgresConfig
-
 		encodedConfig, encodingErr = proto.Marshal(pgConfig)
-
 	case protos.DBType_SNOWFLAKE:
 		sfConfigObject, ok := config.(*protos.Peer_SnowflakeConfig)
 		if !ok {
@@ -566,13 +564,18 @@ func (h *FlowRequestHandler) CreatePeer(
 		encodedConfig, encodingErr = proto.Marshal(s3Config)
 	case protos.DBType_CLICKHOUSE:
 		chConfigObject, ok := config.(*protos.Peer_ClickhouseConfig)
-
 		if !ok {
 			return wrongConfigResponse, nil
 		}
-
 		chConfig := chConfigObject.ClickhouseConfig
 		encodedConfig, encodingErr = proto.Marshal(chConfig)
+	case protos.DBType_KAFKA:
+		kaConfigObject, ok := config.(*protos.Peer_KafkaConfig)
+		if !ok {
+			return wrongConfigResponse, nil
+		}
+		kaConfig := kaConfigObject.KafkaConfig
+		encodedConfig, encodingErr = proto.Marshal(kaConfig)
 	default:
 		return wrongConfigResponse, nil
 	}