Gleipnir/jet
4
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Small optimizations.

Browse source
This commit is contained in:
go-jet 2023-04-14 12:20:36 +02:00
parent 73d7e4823c
commit 65e02fa87d
2 changed files with 22 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

16
qrm/scan_context.go
View file

@ -123,7 +123,7 @@ func (s *ScanContext) getTypeInfo(structType reflect.Type, parentField *reflect.
type groupKeyInfo struct {
typeName string
indexes []int
pkIndexes []int
subTypes []groupKeyInfo
}
@ -148,13 +148,13 @@ func (s *ScanContext) getGroupKey(structType reflect.Type, structField *reflect.
}
func (s *ScanContext) constructGroupKey(groupKeyInfo groupKeyInfo) string {
if len(groupKeyInfo.indexes) == 0 && len(groupKeyInfo.subTypes) == 0 {
if len(groupKeyInfo.pkIndexes) == 0 && len(groupKeyInfo.subTypes) == 0 {
return fmt.Sprintf("|ROW:%d|", s.rowNum)
}
var groupKeys []string
for _, index := range groupKeyInfo.indexes {
for _, index := range groupKeyInfo.pkIndexes {
groupKeys = append(groupKeys, s.rowElemToString(index))
}
@ -190,19 +190,19 @@ func (s *ScanContext) getGroupKeyInfo(
if isPrimaryKey(field, primaryKeyOverwrites) {
newTypeName, fieldName := getTypeAndFieldName(typeName, field)
index := s.typeToColumnIndex(newTypeName, fieldName)
pkIndex := s.typeToColumnIndex(newTypeName, fieldName)
if index < 0 {
if pkIndex < 0 {
continue
}
ret.indexes = append(ret.indexes, index)
ret.pkIndexes = append(ret.pkIndexes, pkIndex)
} else if fieldType.Kind() == reflect.Struct {
} else if fieldType.Kind() == reflect.Struct && fieldType != timeType {
subType := s.getGroupKeyInfo(fieldType, &field, typeVisited)
if len(subType.indexes) != 0 || len(subType.subTypes) != 0 {
if len(subType.pkIndexes) != 0 || len(subType.subTypes) != 0 {
ret.subTypes = append(ret.subTypes, subType)
}
}

12
qrm/utill.go
View file

@ -360,8 +360,20 @@ func cloneBytes(b []byte) []byte {
func concat(stringList ...string) string {
var b strings.Builder
b.Grow(length(stringList))
for _, str := range stringList {
b.WriteString(str)
}
return b.String()
}
func length(strings []string) int {
var ret int
for i := 0; i < len(strings); i++ {
ret += len(strings[i])
}
return ret
}