From 0ce8b58a79283706ec4fec74dfded99b64b13d03 Mon Sep 17 00:00:00 2001 From: Daniel <845765@qq.com> Date: Sun, 24 Dec 2023 22:27:38 +0800 Subject: [PATCH] :art: Add Rollup column to database table view https://github.com/siyuan-note/siyuan/issues/9958 --- kernel/av/av.go | 24 +++++++++++ kernel/model/attribute_view.go | 75 ++++++++++++++++++++++++++++------ kernel/treenode/node.go | 24 +++++++++++ 3 files changed, 110 insertions(+), 13 deletions(-) diff --git a/kernel/av/av.go b/kernel/av/av.go index 348366ce6..c2999ca16 100644 --- a/kernel/av/av.go +++ b/kernel/av/av.go @@ -49,6 +49,16 @@ type KeyValues struct { Values []*Value `json:"values,omitempty"` // 属性视图属性列值 } +func (kValues *KeyValues) GetValue(blockID string) (ret *Value) { + for _, v := range kValues.Values { + if v.BlockID == blockID { + ret = v + return + } + } + return +} + type KeyType string const ( @@ -306,6 +316,20 @@ func (av *AttributeView) GetCurrentView() (ret *View, err error) { return } +func (av *AttributeView) GetValue(keyID, blockID string) (ret *Value) { + for _, kv := range av.KeyValues { + if kv.Key.ID == keyID { + for _, v := range kv.Values { + if v.BlockID == blockID { + ret = v + return + } + } + } + } + return +} + func (av *AttributeView) GetKey(keyID string) (ret *Key, err error) { for _, kv := range av.KeyValues { if kv.Key.ID == keyID { diff --git a/kernel/model/attribute_view.go b/kernel/model/attribute_view.go index 581da4d3d..542131929 100644 --- a/kernel/model/attribute_view.go +++ b/kernel/model/attribute_view.go @@ -180,23 +180,48 @@ func GetBlockAttributeViewKeys(blockID string) (ret []*BlockAttributeViewKeys) { } // 渲染自动生成的列值，比如模板列、关联列、汇总列、创建时间列和更新时间列 - // 先处理创建时间和更新时间 + // 先处理关联列、汇总列、创建时间列和更新时间列 for _, kv := range keyValues { switch kv.Key.Type { - case av.KeyTypeRelation: - relKey, _ := attrView.GetKey(kv.Values[0].KeyID) - if nil != relKey && nil != relKey.Relation { - destAv, _ := av.ParseAttributeView(relKey.Relation.AvID) - if nil != destAv { - blocks := map[string]string{} - for _, blockValue := range destAv.GetBlockKeyValues().Values { - blocks[blockValue.BlockID] = blockValue.Block.Content - } - for _, blockID := range kv.Values[0].Relation.BlockIDs { - kv.Values[0].Relation.Contents = append(kv.Values[0].Relation.Contents, blocks[blockID]) - } + case av.KeyTypeRollup: + if nil == kv.Key.Rollup { + break + } + + relKey, _ := attrView.GetKey(kv.Key.Rollup.RelationKeyID) + if nil == relKey { + break + } + + var blockIDs []string + relVal := attrView.GetValue(kv.Key.Rollup.RelationKeyID, kv.Values[0].BlockID) + if nil != relVal && nil != relVal.Relation { + blockIDs = relVal.Relation.BlockIDs + } + + destAv, _ := av.ParseAttributeView(relKey.Relation.AvID) + if nil != destAv { + for _, bID := range blockIDs { + kv.Values[0].Rollup.Contents = append(kv.Values[0].Rollup.Contents, destAv.GetValue(kv.Key.Rollup.KeyID, bID).String()) } } + case av.KeyTypeRelation: + if nil == kv.Key.Relation { + break + } + + destAv, _ := av.ParseAttributeView(kv.Key.Relation.AvID) + if nil == destAv { + break + } + + blocks := map[string]string{} + for _, blockValue := range destAv.GetBlockKeyValues().Values { + blocks[blockValue.BlockID] = blockValue.Block.Content + } + for _, bID := range kv.Values[0].Relation.BlockIDs { + kv.Values[0].Relation.Contents = append(kv.Values[0].Relation.Contents, blocks[bID]) + } case av.KeyTypeCreated: createdStr := blockID[:len("20060102150405")] created, parseErr := time.ParseInLocation("20060102150405", createdStr, time.Local) @@ -699,6 +724,30 @@ func renderAttributeViewTable(attrView *av.AttributeView, view *av.View) (ret *a } content := renderTemplateCol(ial, cell.Value.Template.Content, keyValues) cell.Value.Template.Content = content + case av.KeyTypeRollup: // 渲染汇总列 + rollupKey, _ := attrView.GetKey(cell.Value.KeyID) + if nil == rollupKey || nil == rollupKey.Rollup { + break + } + + relKey, _ := attrView.GetKey(rollupKey.Rollup.RelationKeyID) + if nil == relKey || nil == relKey.Relation { + break + } + + relVal := attrView.GetValue(relKey.ID, row.ID) + if nil == relVal || nil == relVal.Relation { + break + } + + destAv, _ := av.ParseAttributeView(relKey.Relation.AvID) + if nil == destAv { + break + } + + for _, blockID := range relVal.Relation.BlockIDs { + cell.Value.Rollup.Contents = append(cell.Value.Rollup.Contents, destAv.GetValue(rollupKey.Rollup.KeyID, blockID).String()) + } case av.KeyTypeRelation: // 渲染关联列 relKey, _ := attrView.GetKey(cell.Value.KeyID) if nil != relKey && nil != relKey.Relation { diff --git a/kernel/treenode/node.go b/kernel/treenode/node.go index 7e80aa0e2..a316b82b3 100644 --- a/kernel/treenode/node.go +++ b/kernel/treenode/node.go @@ -723,6 +723,30 @@ func renderAttributeViewTable(attrView *av.AttributeView, view *av.View) (ret *a } content := renderTemplateCol(ial, cell.Value.Template.Content, keyValues) cell.Value.Template.Content = content + case av.KeyTypeRollup: // 渲染汇总列 + rollupKey, _ := attrView.GetKey(cell.Value.KeyID) + if nil == rollupKey || nil == rollupKey.Rollup { + break + } + + relKey, _ := attrView.GetKey(rollupKey.Rollup.RelationKeyID) + if nil == relKey || nil == relKey.Relation { + break + } + + relVal := attrView.GetValue(relKey.ID, row.ID) + if nil == relVal || nil == relVal.Relation { + break + } + + destAv, _ := av.ParseAttributeView(relKey.Relation.AvID) + if nil == destAv { + break + } + + for _, blockID := range relVal.Relation.BlockIDs { + cell.Value.Rollup.Contents = append(cell.Value.Rollup.Contents, destAv.GetValue(rollupKey.Rollup.KeyID, blockID).String()) + } case av.KeyTypeRelation: // 渲染关联列 relKey, _ := attrView.GetKey(cell.Value.KeyID) if nil != relKey && nil != relKey.Relation {