feat(go): unify index logic

Author: Kybxd

cmd/protoc-gen-go-tableau-loader/leveledindex/leveled_index.go renamed to cmd/protoc-gen-go-tableau-loader/indexes/generator.go

@@ -1,11 +1,10 @@
-package leveledindex
+package indexes
 
 import (
 	"fmt"
 
 	"github.com/tableauio/loader/cmd/protoc-gen-go-tableau-loader/helper"
 	"github.com/tableauio/loader/internal/index"
-	"github.com/tableauio/loader/internal/options"
 	"google.golang.org/protobuf/compiler/protogen"
 	"google.golang.org/protobuf/reflect/protoreflect"
 )
@@ -16,9 +15,10 @@ type Generator struct {
 	descriptor *index.IndexDescriptor
 	message    *protogen.Message
 
+	// level message
 	maxDepth int
-	Keys     helper.MapKeys
-	MapFds   []protoreflect.FieldDescriptor
+	keys     helper.MapKeys
+	mapFds   []protoreflect.FieldDescriptor
 }
 
 func NewGenerator(gen *protogen.Plugin, g *protogen.GeneratedFile, descriptor *index.IndexDescriptor, message *protogen.Message) *Generator {
@@ -28,19 +28,19 @@ func NewGenerator(gen *protogen.Plugin, g *protogen.GeneratedFile, descriptor *i
 		descriptor: descriptor,
 		message:    message,
 	}
-	generator.init()
+	generator.initLevelMessage()
 	return generator
 }
 
-func (x *Generator) init() {
+func (x *Generator) initLevelMessage() {
 	for levelMessage := x.descriptor.LevelMessage; levelMessage != nil; levelMessage = levelMessage.NextLevel {
 		if fd := levelMessage.FD; fd != nil && fd.IsMap() {
-			x.Keys = x.Keys.AddMapKey(helper.MapKey{
+			x.keys = x.keys.AddMapKey(helper.MapKey{
 				Type:      helper.ParseMapKeyType(fd.MapKey()),
 				Name:      helper.ParseMapFieldNameAsFuncParam(fd),
 				FieldName: helper.ParseMapFieldNameAsKeyStructFieldName(fd),
 			})
-			x.MapFds = append(x.MapFds, fd)
+			x.mapFds = append(x.mapFds, fd)
 		}
 		if len(levelMessage.Indexes) != 0 || len(levelMessage.OrderedIndexes) != 0 {
 			x.maxDepth = levelMessage.Depth
@@ -49,33 +49,73 @@ func (x *Generator) init() {
 }
 
 func (x *Generator) NeedGenerate() bool {
-	return options.NeedGenIndex(x.message.Desc, options.LangCPP) || options.NeedGenOrderedIndex(x.message.Desc, options.LangCPP)
+	return x.needGenerateIndex() || x.needGenerateOrderedIndex()
 }
 
 func (x *Generator) messagerName() string {
 	return string(x.message.Desc.Name())
 }
 
-func (x *Generator) KeyType(mapFd protoreflect.FieldDescriptor) string {
+func (x *Generator) levelKeyType(mapFd protoreflect.FieldDescriptor) string {
 	return fmt.Sprintf("%s_LeveledIndex_%sKey", x.messagerName(), helper.ParseLeveledMapPrefix(x.message.Desc, mapFd))
 }
 
-func (x *Generator) GenLeveledIndexTypeDef() {
+func (x *Generator) mapValueType(index *index.LevelIndex) protogen.GoIdent {
+	return helper.FindMessageGoIdent(x.gen, index.MD)
+}
+
+func (x *Generator) fieldGetter(fd protoreflect.FieldDescriptor) string {
+	return fmt.Sprintf(".Get%s()", helper.ParseIndexFieldName(x.gen, fd))
+}
+
+func (x *Generator) parseKeyFieldNameAndSuffix(field *index.LevelField) (string, string) {
+	var fieldName, suffix string
+	for i, leveledFd := range field.LeveledFDList {
+		fieldName += x.fieldGetter(leveledFd)
+		if i == len(field.LeveledFDList)-1 && leveledFd.Message() != nil {
+			switch leveledFd.Message().FullName() {
+			case "google.protobuf.Timestamp", "google.protobuf.Duration":
+				suffix = ".GetSeconds()"
+			default:
+			}
+		}
+	}
+	return fieldName, suffix
+}
+
+func (x *Generator) GenIndexTypeDef() {
 	if !x.NeedGenerate() {
 		return
 	}
-	for i := 1; i <= x.maxDepth-3 && i <= len(x.MapFds)-1; i++ {
+	for i := 1; i <= x.maxDepth-3 && i <= len(x.mapFds)-1; i++ {
 		if i == 1 {
 			x.g.P()
 			x.g.P("// LeveledIndex keys.")
 		}
-		fd := x.MapFds[i]
-		keyType := x.KeyType(fd)
-		keys := x.Keys[:i+1]
+		fd := x.mapFds[i]
+		keyType := x.levelKeyType(fd)
+		keys := x.keys[:i+1]
 		x.g.P("type ", keyType, " struct {")
 		for _, key := range keys {
 			x.g.P(key.FieldName, " ", key.Type)
 		}
 		x.g.P("}")
 	}
+	x.genIndexTypeDef()
+	x.genOrderedIndexTypeDef()
+}
+
+func (x *Generator) GenIndexField() {
+	x.genIndexField()
+	x.genOrderedIndexField()
+}
+
+func (x *Generator) GenIndexLoader() {
+	x.genIndexLoader()
+	x.genOrderedIndexLoader()
+}
+
+func (x *Generator) GenIndexFinders() {
+	x.genIndexFinders()
+	x.genOrderedIndexFinders()
 }