mirror.go 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202
  1. package av
  2. import (
  3. "os"
  4. "path/filepath"
  5. "sync"
  6. "github.com/88250/gulu"
  7. "github.com/88250/lute/ast"
  8. "github.com/siyuan-note/filelock"
  9. "github.com/siyuan-note/logging"
  10. "github.com/siyuan-note/siyuan/kernel/util"
  11. "github.com/vmihailenco/msgpack/v5"
  12. )
  13. var (
  14. AttributeViewBlocksLock = sync.Mutex{}
  15. )
  16. func GetBlockRels() (ret map[string][]string) {
  17. AttributeViewBlocksLock.Lock()
  18. defer AttributeViewBlocksLock.Unlock()
  19. ret = map[string][]string{}
  20. blocks := filepath.Join(util.DataDir, "storage", "av", "blocks.msgpack")
  21. if !filelock.IsExist(blocks) {
  22. return
  23. }
  24. data, err := filelock.ReadFile(blocks)
  25. if nil != err {
  26. logging.LogErrorf("read attribute view blocks failed: %s", err)
  27. return
  28. }
  29. if err = msgpack.Unmarshal(data, &ret); nil != err {
  30. logging.LogErrorf("unmarshal attribute view blocks failed: %s", err)
  31. return
  32. }
  33. return
  34. }
  35. func IsMirror(avID string) bool {
  36. AttributeViewBlocksLock.Lock()
  37. defer AttributeViewBlocksLock.Unlock()
  38. blocks := filepath.Join(util.DataDir, "storage", "av", "blocks.msgpack")
  39. if !filelock.IsExist(blocks) {
  40. return false
  41. }
  42. data, err := filelock.ReadFile(blocks)
  43. if nil != err {
  44. logging.LogErrorf("read attribute view blocks failed: %s", err)
  45. return false
  46. }
  47. avBlocks := map[string][]string{}
  48. if err = msgpack.Unmarshal(data, &avBlocks); nil != err {
  49. logging.LogErrorf("unmarshal attribute view blocks failed: %s", err)
  50. return false
  51. }
  52. blockIDs := avBlocks[avID]
  53. return nil != blockIDs && 1 < len(blockIDs)
  54. }
  55. func RemoveBlockRel(avID, blockID string) {
  56. AttributeViewBlocksLock.Lock()
  57. defer AttributeViewBlocksLock.Unlock()
  58. blocks := filepath.Join(util.DataDir, "storage", "av", "blocks.msgpack")
  59. if !filelock.IsExist(blocks) {
  60. return
  61. }
  62. data, err := filelock.ReadFile(blocks)
  63. if nil != err {
  64. logging.LogErrorf("read attribute view blocks failed: %s", err)
  65. return
  66. }
  67. avBlocks := map[string][]string{}
  68. if err = msgpack.Unmarshal(data, &avBlocks); nil != err {
  69. logging.LogErrorf("unmarshal attribute view blocks failed: %s", err)
  70. return
  71. }
  72. blockIDs := avBlocks[avID]
  73. if nil == blockIDs {
  74. return
  75. }
  76. var newBlockIDs []string
  77. for _, v := range blockIDs {
  78. if v != blockID {
  79. newBlockIDs = append(newBlockIDs, v)
  80. }
  81. }
  82. avBlocks[avID] = newBlockIDs
  83. data, err = msgpack.Marshal(avBlocks)
  84. if nil != err {
  85. logging.LogErrorf("marshal attribute view blocks failed: %s", err)
  86. return
  87. }
  88. if err = filelock.WriteFile(blocks, data); nil != err {
  89. logging.LogErrorf("write attribute view blocks failed: %s", err)
  90. return
  91. }
  92. }
  93. func BatchUpsertBlockRel(nodes []*ast.Node) {
  94. AttributeViewBlocksLock.Lock()
  95. defer AttributeViewBlocksLock.Unlock()
  96. avBlocks := map[string][]string{}
  97. blocks := filepath.Join(util.DataDir, "storage", "av", "blocks.msgpack")
  98. if !filelock.IsExist(blocks) {
  99. if err := os.MkdirAll(filepath.Dir(blocks), 0755); nil != err {
  100. logging.LogErrorf("create attribute view dir failed: %s", err)
  101. return
  102. }
  103. } else {
  104. data, err := filelock.ReadFile(blocks)
  105. if nil != err {
  106. logging.LogErrorf("read attribute view blocks failed: %s", err)
  107. return
  108. }
  109. if err = msgpack.Unmarshal(data, &avBlocks); nil != err {
  110. logging.LogErrorf("unmarshal attribute view blocks failed: %s", err)
  111. return
  112. }
  113. }
  114. for _, n := range nodes {
  115. if ast.NodeAttributeView != n.Type {
  116. continue
  117. }
  118. if "" == n.AttributeViewID || "" == n.ID {
  119. continue
  120. }
  121. blockIDs := avBlocks[n.AttributeViewID]
  122. blockIDs = append(blockIDs, n.ID)
  123. blockIDs = gulu.Str.RemoveDuplicatedElem(blockIDs)
  124. avBlocks[n.AttributeViewID] = blockIDs
  125. }
  126. data, err := msgpack.Marshal(avBlocks)
  127. if nil != err {
  128. logging.LogErrorf("marshal attribute view blocks failed: %s", err)
  129. return
  130. }
  131. if err = filelock.WriteFile(blocks, data); nil != err {
  132. logging.LogErrorf("write attribute view blocks failed: %s", err)
  133. return
  134. }
  135. }
  136. func UpsertBlockRel(avID, blockID string) {
  137. AttributeViewBlocksLock.Lock()
  138. defer AttributeViewBlocksLock.Unlock()
  139. avBlocks := map[string][]string{}
  140. blocks := filepath.Join(util.DataDir, "storage", "av", "blocks.msgpack")
  141. if !filelock.IsExist(blocks) {
  142. if err := os.MkdirAll(filepath.Dir(blocks), 0755); nil != err {
  143. logging.LogErrorf("create attribute view dir failed: %s", err)
  144. return
  145. }
  146. } else {
  147. data, err := filelock.ReadFile(blocks)
  148. if nil != err {
  149. logging.LogErrorf("read attribute view blocks failed: %s", err)
  150. return
  151. }
  152. if err = msgpack.Unmarshal(data, &avBlocks); nil != err {
  153. logging.LogErrorf("unmarshal attribute view blocks failed: %s", err)
  154. return
  155. }
  156. }
  157. blockIDs := avBlocks[avID]
  158. blockIDs = append(blockIDs, blockID)
  159. blockIDs = gulu.Str.RemoveDuplicatedElem(blockIDs)
  160. avBlocks[avID] = blockIDs
  161. data, err := msgpack.Marshal(avBlocks)
  162. if nil != err {
  163. logging.LogErrorf("marshal attribute view blocks failed: %s", err)
  164. return
  165. }
  166. if err = filelock.WriteFile(blocks, data); nil != err {
  167. logging.LogErrorf("write attribute view blocks failed: %s", err)
  168. return
  169. }
  170. }