snapshot_flow: use table schema utils

flow/workflows/snapshot_flow.go

@@ -16,7 +16,6 @@ import (
 	"go.temporal.io/sdk/log"
 	"go.temporal.io/sdk/temporal"
 	"go.temporal.io/sdk/workflow"
-	"golang.org/x/exp/maps"
 )
 
 type SnapshotFlowExecution struct {
@@ -141,17 +140,12 @@ func (s *SnapshotFlowExecution) cloneTable(
 	if len(mapping.Exclude) != 0 {
 		for _, v := range s.config.TableNameSchemaMapping {
 			if v.TableIdentifier == srcName {
-				if v.Columns != nil {
-					cols := maps.Keys(v.Columns)
-					for i, col := range cols {
-						cols[i] = fmt.Sprintf(`"%s"`, col)
-					}
-					from = strings.Join(cols, ",")
-					break
-				} else {
-					from = strings.Join(v.ColumnNames, ",")
-					break
+				colNames := utils.TableSchemaColumnNames(v)
+				for i, colName := range colNames {
+					colNames[i] = fmt.Sprintf(`"%s"`, colName)
 				}
+				from = strings.Join(colNames, ",")
+				break
 			}
 		}
 	}