github.com/whtcorpsinc/milevadb-prod@v0.0.0-20211104133533-f57f4be3b597/causetstore/petri/acyclic/causet/embedded/stringer.go

// Copyright 2020 WHTCORPS INC, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// See the License for the specific language governing permissions and
// limitations under the License.

package cascades

import (
	"bytes"
	"fmt"
	"strings"

	"github.com/whtcorpsinc/milevadb/causet/memo"
)

// ToString stringifies a Group Tree.
func ToString(g *memo.Group) []string {
	idMap := make(map[*memo.Group]int)
	idMap[g] = 0
	return toString(g, idMap, map[*memo.Group]struct{}{}, []string{})
}

// toString recursively stringifies a Group Tree using a preorder traversal method.
func toString(g *memo.Group, idMap map[*memo.Group]int, visited map[*memo.Group]struct{}, strs []string) []string {
	if _, exists := visited[g]; exists {
		return strs
	}
	visited[g] = struct{}{}
	// Add new Groups to idMap.
	for item := g.Equivalents.Front(); item != nil; item = item.Next() {
		expr := item.Value.(*memo.GroupExpr)
		for _, childGroup := range expr.Children {
			if _, exists := idMap[childGroup]; !exists {
				idMap[childGroup] = len(idMap)
			}
		}
	}
	// Visit self first.
	strs = append(strs, groupToString(g, idMap)...)
	// Visit children then.
	for item := g.Equivalents.Front(); item != nil; item = item.Next() {
		expr := item.Value.(*memo.GroupExpr)
		for _, childGroup := range expr.Children {
			strs = toString(childGroup, idMap, visited, strs)
		}
	}
	return strs
}

// groupToString only stringifies a single Group.
// Format:
// Group#1 DeferredCauset: [DeferredCauset#1,DeferredCauset#2,DeferredCauset#13] Unique key: []
//     Selection_4 input:[Group#2], eq(DeferredCauset#13, DeferredCauset#2), gt(DeferredCauset#1, 10)
//     Projection_15 input:Group#3 DeferredCauset#1, DeferredCauset#2
func groupToString(g *memo.Group, idMap map[*memo.Group]int) []string {
	schemaReplicant := g.Prop.Schema
	defCausStrs := make([]string, 0, len(schemaReplicant.DeferredCausets))
	for _, defCaus := range schemaReplicant.DeferredCausets {
		defCausStrs = append(defCausStrs, defCaus.String())
	}

	groupLine := bytes.NewBufferString("")
	fmt.Fprintf(groupLine, "Group#%d Schema:[%s]", idMap[g], strings.Join(defCausStrs, ","))

	if len(g.Prop.Schema.Keys) > 0 {
		ukStrs := make([]string, 0, len(schemaReplicant.Keys))
		for _, key := range schemaReplicant.Keys {
			ukDefCausStrs := make([]string, 0, len(key))
			for _, defCaus := range key {
				ukDefCausStrs = append(ukDefCausStrs, defCaus.String())
			}
			ukStrs = append(ukStrs, strings.Join(ukDefCausStrs, ","))
		}
		fmt.Fprintf(groupLine, ", UniqueKey:[%s]", strings.Join(ukStrs, ","))
	}

	result := make([]string, 0, g.Equivalents.Len()+1)
	result = append(result, groupLine.String())
	for item := g.Equivalents.Front(); item != nil; item = item.Next() {
		expr := item.Value.(*memo.GroupExpr)
		result = append(result, "    "+groupExprToString(expr, idMap))
	}
	return result
}

// groupExprToString stringifies a groupExpr(or a LogicalCauset).
// Format:
// Selection_13 input:Group#2 gt(DeferredCauset#1, DeferredCauset#4)
func groupExprToString(expr *memo.GroupExpr, idMap map[*memo.Group]int) string {
	buffer := bytes.NewBufferString(expr.ExprNode.ExplainID().String())
	if len(expr.Children) == 0 {
		fmt.Fprintf(buffer, " %s", expr.ExprNode.ExplainInfo())
	} else {
		fmt.Fprintf(buffer, " %s", getChildrenGroupID(expr, idMap))
		explainInfo := expr.ExprNode.ExplainInfo()
		if len(explainInfo) != 0 {
			fmt.Fprintf(buffer, ", %s", explainInfo)
		}
	}
	return buffer.String()
}

func getChildrenGroupID(expr *memo.GroupExpr, idMap map[*memo.Group]int) string {
	children := make([]string, 0, len(expr.Children))
	for _, child := range expr.Children {
		children = append(children, fmt.Sprintf("Group#%d", idMap[child]))
	}
	return "input:[" + strings.Join(children, ",") + "]"
}