fix: expose svc for sharding (#8808)

apecloud · Jan 15, 2025 · 5be65e3 · 5be65e3
1 parent 982672d
commit 5be65e3
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pkg/operations/expose.go b/pkg/operations/expose.go
@@ -62,7 +62,9 @@ func (e ExposeOpsHandler) Action(reqCtx intctrlutil.RequestCtx, cli client.Clien
 	for _, comp := range opsRes.Cluster.Spec.ComponentSpecs {
 		compMap[comp.Name] = comp
 	}
-
+	for _, shard := range opsRes.Cluster.Spec.Shardings {
+		compMap[shard.Name] = shard.Template
+	}
 	for _, expose := range exposeMap {
 		clusterCompSpecName := ""
 		compDef := ""
@@ -71,7 +73,8 @@ func (e ExposeOpsHandler) Action(reqCtx intctrlutil.RequestCtx, cli client.Clien
 			if !ok {
 				return fmt.Errorf("component spec not found: %s", expose.ComponentName)
 			}
-			clusterCompSpecName = clusterCompSpec.Name
+			// shardName or compName
+			clusterCompSpecName = expose.ComponentName
 			compDef = clusterCompSpec.ComponentDef
 		}