You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

clean_pinned.go 27 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888
  1. package event
  2. import (
  3. "fmt"
  4. "math"
  5. "math/rand"
  6. "sync"
  7. "time"
  8. "github.com/samber/lo"
  9. "gitlink.org.cn/cloudream/common/pkgs/bitmap"
  10. "gitlink.org.cn/cloudream/common/pkgs/logger"
  11. cdssdk "gitlink.org.cn/cloudream/common/sdks/storage"
  12. "gitlink.org.cn/cloudream/common/utils/lo2"
  13. "gitlink.org.cn/cloudream/common/utils/math2"
  14. "gitlink.org.cn/cloudream/common/utils/sort2"
  15. "gitlink.org.cn/cloudream/storage/common/consts"
  16. stgglb "gitlink.org.cn/cloudream/storage/common/globals"
  17. stgmod "gitlink.org.cn/cloudream/storage/common/models"
  18. "gitlink.org.cn/cloudream/storage/common/pkgs/distlock/reqbuilder"
  19. "gitlink.org.cn/cloudream/storage/common/pkgs/ioswitch/plans"
  20. agtmq "gitlink.org.cn/cloudream/storage/common/pkgs/mq/agent"
  21. coormq "gitlink.org.cn/cloudream/storage/common/pkgs/mq/coordinator"
  22. scevt "gitlink.org.cn/cloudream/storage/common/pkgs/mq/scanner/event"
  23. )
  24. type CleanPinned struct {
  25. *scevt.CleanPinned
  26. }
  27. func NewCleanPinned(evt *scevt.CleanPinned) *CleanPinned {
  28. return &CleanPinned{
  29. CleanPinned: evt,
  30. }
  31. }
  32. func (t *CleanPinned) TryMerge(other Event) bool {
  33. event, ok := other.(*CleanPinned)
  34. if !ok {
  35. return false
  36. }
  37. return t.PackageID == event.PackageID
  38. }
  39. func (t *CleanPinned) Execute(execCtx ExecuteContext) {
  40. log := logger.WithType[CleanPinned]("Event")
  41. startTime := time.Now()
  42. log.Debugf("begin with %v", logger.FormatStruct(t.CleanPinned))
  43. defer func() {
  44. log.Debugf("end, time: %v", time.Since(startTime))
  45. }()
  46. coorCli, err := stgglb.CoordinatorMQPool.Acquire()
  47. if err != nil {
  48. log.Warnf("new coordinator client: %s", err.Error())
  49. return
  50. }
  51. defer stgglb.CoordinatorMQPool.Release(coorCli)
  52. getObjs, err := coorCli.GetPackageObjectDetails(coormq.ReqGetPackageObjectDetails(t.PackageID))
  53. if err != nil {
  54. log.Warnf("getting package objects: %s", err.Error())
  55. return
  56. }
  57. getLoadLog, err := coorCli.GetPackageLoadLogDetails(coormq.ReqGetPackageLoadLogDetails(t.PackageID))
  58. if err != nil {
  59. log.Warnf("getting package load log details: %s", err.Error())
  60. return
  61. }
  62. readerNodeIDs := lo.Map(getLoadLog.Logs, func(item coormq.PackageLoadLogDetail, idx int) cdssdk.NodeID { return item.Storage.NodeID })
  63. // 注意!需要保证allNodeID包含所有之后可能用到的节点ID
  64. // TOOD 可以考虑设计Cache机制
  65. var allNodeID []cdssdk.NodeID
  66. for _, obj := range getObjs.Objects {
  67. for _, block := range obj.Blocks {
  68. allNodeID = append(allNodeID, block.NodeID)
  69. }
  70. allNodeID = append(allNodeID, obj.PinnedAt...)
  71. }
  72. allNodeID = append(allNodeID, readerNodeIDs...)
  73. getNodeResp, err := coorCli.GetNodes(coormq.NewGetNodes(lo.Union(allNodeID)))
  74. if err != nil {
  75. log.Warnf("getting nodes: %s", err.Error())
  76. return
  77. }
  78. allNodeInfos := make(map[cdssdk.NodeID]*cdssdk.Node)
  79. for _, node := range getNodeResp.Nodes {
  80. n := node
  81. allNodeInfos[node.NodeID] = &n
  82. }
  83. // 只对ec和rep对象进行处理
  84. var ecObjects []stgmod.ObjectDetail
  85. var repObjects []stgmod.ObjectDetail
  86. for _, obj := range getObjs.Objects {
  87. if _, ok := obj.Object.Redundancy.(*cdssdk.ECRedundancy); ok {
  88. ecObjects = append(ecObjects, obj)
  89. } else if _, ok := obj.Object.Redundancy.(*cdssdk.RepRedundancy); ok {
  90. repObjects = append(repObjects, obj)
  91. }
  92. }
  93. planBld := plans.NewPlanBuilder()
  94. pinPlans := make(map[cdssdk.NodeID]*[]string)
  95. // 对于rep对象,统计出所有对象块分布最多的两个节点,用这两个节点代表所有rep对象块的分布,去进行退火算法
  96. var repObjectsUpdating []coormq.UpdatingObjectRedundancy
  97. repMostNodeIDs := t.summaryRepObjectBlockNodes(repObjects)
  98. solu := t.startAnnealing(allNodeInfos, readerNodeIDs, annealingObject{
  99. totalBlockCount: 1,
  100. minBlockCnt: 1,
  101. pinnedAt: repMostNodeIDs,
  102. blocks: nil,
  103. })
  104. for _, obj := range repObjects {
  105. repObjectsUpdating = append(repObjectsUpdating, t.makePlansForRepObject(solu, obj, pinPlans))
  106. }
  107. // 对于ec对象,则每个对象单独进行退火算法
  108. var ecObjectsUpdating []coormq.UpdatingObjectRedundancy
  109. for _, obj := range ecObjects {
  110. ecRed := obj.Object.Redundancy.(*cdssdk.ECRedundancy)
  111. solu := t.startAnnealing(allNodeInfos, readerNodeIDs, annealingObject{
  112. totalBlockCount: ecRed.N,
  113. minBlockCnt: ecRed.K,
  114. pinnedAt: obj.PinnedAt,
  115. blocks: obj.Blocks,
  116. })
  117. ecObjectsUpdating = append(ecObjectsUpdating, t.makePlansForECObject(allNodeInfos, solu, obj, &planBld))
  118. }
  119. ioSwRets, err := t.executePlans(execCtx, pinPlans, &planBld)
  120. if err != nil {
  121. log.Warn(err.Error())
  122. return
  123. }
  124. // 根据按照方案进行调整的结果,填充更新元数据的命令
  125. for i := range ecObjectsUpdating {
  126. t.populateECObjectEntry(&ecObjectsUpdating[i], ecObjects[i], ioSwRets)
  127. }
  128. finalEntries := append(repObjectsUpdating, ecObjectsUpdating...)
  129. if len(finalEntries) > 0 {
  130. _, err = coorCli.UpdateObjectRedundancy(coormq.ReqUpdateObjectRedundancy(finalEntries))
  131. if err != nil {
  132. log.Warnf("changing object redundancy: %s", err.Error())
  133. return
  134. }
  135. }
  136. }
  137. func (t *CleanPinned) summaryRepObjectBlockNodes(objs []stgmod.ObjectDetail) []cdssdk.NodeID {
  138. type nodeBlocks struct {
  139. NodeID cdssdk.NodeID
  140. Count int
  141. }
  142. nodeBlocksMap := make(map[cdssdk.NodeID]*nodeBlocks)
  143. for _, obj := range objs {
  144. cacheBlockNodes := make(map[cdssdk.NodeID]bool)
  145. for _, block := range obj.Blocks {
  146. if _, ok := nodeBlocksMap[block.NodeID]; !ok {
  147. nodeBlocksMap[block.NodeID] = &nodeBlocks{
  148. NodeID: block.NodeID,
  149. Count: 0,
  150. }
  151. }
  152. nodeBlocksMap[block.NodeID].Count++
  153. cacheBlockNodes[block.NodeID] = true
  154. }
  155. for _, nodeID := range obj.PinnedAt {
  156. if cacheBlockNodes[nodeID] {
  157. continue
  158. }
  159. if _, ok := nodeBlocksMap[nodeID]; !ok {
  160. nodeBlocksMap[nodeID] = &nodeBlocks{
  161. NodeID: nodeID,
  162. Count: 0,
  163. }
  164. }
  165. nodeBlocksMap[nodeID].Count++
  166. }
  167. }
  168. nodes := lo.Values(nodeBlocksMap)
  169. sort2.Sort(nodes, func(left *nodeBlocks, right *nodeBlocks) int {
  170. return right.Count - left.Count
  171. })
  172. // 只选出块数超过一半的节点,但要保证至少有两个节点
  173. for i := 2; i < len(nodes); i++ {
  174. if nodes[i].Count < len(objs)/2 {
  175. nodes = nodes[:i]
  176. break
  177. }
  178. }
  179. return lo.Map(nodes, func(item *nodeBlocks, idx int) cdssdk.NodeID { return item.NodeID })
  180. }
  181. type annealingState struct {
  182. allNodeInfos map[cdssdk.NodeID]*cdssdk.Node // 所有节点的信息
  183. readerNodeIDs []cdssdk.NodeID // 近期可能访问此对象的节点
  184. nodesSortedByReader map[cdssdk.NodeID][]nodeDist // 拥有数据的节点到每个可能访问对象的节点按距离排序
  185. object annealingObject // 进行退火的对象
  186. blockList []objectBlock // 排序后的块分布情况
  187. nodeBlockBitmaps map[cdssdk.NodeID]*bitmap.Bitmap64 // 用位图的形式表示每一个节点上有哪些块
  188. nodeCombTree combinatorialTree // 节点组合树,用于加速计算容灾度
  189. maxScore float64 // 搜索过程中得到过的最大分数
  190. maxScoreRmBlocks []bool // 最大分数对应的删除方案
  191. rmBlocks []bool // 当前删除方案
  192. inversedIndex int // 当前删除方案是从上一次的方案改动哪个flag而来的
  193. lastScore float64 // 上一次方案的分数
  194. }
  195. type objectBlock struct {
  196. Index int
  197. NodeID cdssdk.NodeID
  198. HasEntity bool // 节点拥有实际的文件数据块
  199. HasShadow bool // 如果节点拥有完整文件数据,那么认为这个节点拥有所有块,这些块被称为影子块
  200. FileHash string // 只有在拥有实际文件数据块时,这个字段才有值
  201. }
  202. type nodeDist struct {
  203. NodeID cdssdk.NodeID
  204. Distance float64
  205. }
  206. type combinatorialTree struct {
  207. nodes []combinatorialTreeNode
  208. blocksMaps map[int]bitmap.Bitmap64
  209. nodeIDToLocalNodeID map[cdssdk.NodeID]int
  210. localNodeIDToNodeID []cdssdk.NodeID
  211. }
  212. type annealingObject struct {
  213. totalBlockCount int
  214. minBlockCnt int
  215. pinnedAt []cdssdk.NodeID
  216. blocks []stgmod.ObjectBlock
  217. }
  218. const (
  219. iterActionNone = 0
  220. iterActionSkip = 1
  221. iterActionBreak = 2
  222. )
  223. func newCombinatorialTree(nodeBlocksMaps map[cdssdk.NodeID]*bitmap.Bitmap64) combinatorialTree {
  224. tree := combinatorialTree{
  225. blocksMaps: make(map[int]bitmap.Bitmap64),
  226. nodeIDToLocalNodeID: make(map[cdssdk.NodeID]int),
  227. }
  228. tree.nodes = make([]combinatorialTreeNode, (1 << len(nodeBlocksMaps)))
  229. for id, mp := range nodeBlocksMaps {
  230. tree.nodeIDToLocalNodeID[id] = len(tree.localNodeIDToNodeID)
  231. tree.blocksMaps[len(tree.localNodeIDToNodeID)] = *mp
  232. tree.localNodeIDToNodeID = append(tree.localNodeIDToNodeID, id)
  233. }
  234. tree.nodes[0].localNodeID = -1
  235. index := 1
  236. tree.initNode(0, &tree.nodes[0], &index)
  237. return tree
  238. }
  239. func (t *combinatorialTree) initNode(minAvaiLocalNodeID int, parent *combinatorialTreeNode, index *int) {
  240. for i := minAvaiLocalNodeID; i < len(t.nodeIDToLocalNodeID); i++ {
  241. curIndex := *index
  242. *index++
  243. bitMp := t.blocksMaps[i]
  244. bitMp.Or(&parent.blocksBitmap)
  245. t.nodes[curIndex] = combinatorialTreeNode{
  246. localNodeID: i,
  247. parent: parent,
  248. blocksBitmap: bitMp,
  249. }
  250. t.initNode(i+1, &t.nodes[curIndex], index)
  251. }
  252. }
  253. // 获得索引指定的节点所在的层
  254. func (t *combinatorialTree) GetDepth(index int) int {
  255. depth := 0
  256. // 反复判断节点在哪个子树。从左到右,子树节点的数量呈现8 4 2的变化,由此可以得到每个子树的索引值的范围
  257. subTreeCount := 1 << len(t.nodeIDToLocalNodeID)
  258. for index > 0 {
  259. if index < subTreeCount {
  260. // 定位到一个子树后,深度+1,然后进入这个子树,使用同样的方法再进行定位。
  261. // 进入子树后需要将索引值-1,因为要去掉子树的根节点
  262. index--
  263. depth++
  264. } else {
  265. // 如果索引值不在这个子树范围内,则将值减去子树的节点数量,
  266. // 这样每一次都可以视为使用同样的逻辑对不同大小的树进行判断。
  267. index -= subTreeCount
  268. }
  269. subTreeCount >>= 1
  270. }
  271. return depth
  272. }
  273. // 更新某一个算力中心节点的块分布位图,同时更新它对应组合树节点的所有子节点。
  274. // 如果更新到某个节点时,已有K个块,那么就不会再更新它的子节点
  275. func (t *combinatorialTree) UpdateBitmap(nodeID cdssdk.NodeID, mp bitmap.Bitmap64, k int) {
  276. t.blocksMaps[t.nodeIDToLocalNodeID[nodeID]] = mp
  277. // 首先定义两种遍历树节点时的移动方式:
  278. // 1. 竖直移动(深度增加):从一个节点移动到它最左边的子节点。每移动一步,index+1
  279. // 2. 水平移动:从一个节点移动到它右边的兄弟节点。每移动一步,根据它所在的深度,index+8,+4,+2
  280. // LocalNodeID从0开始,将其+1后得到移动步数steps。
  281. // 将移动步数拆成多部分,分配到上述的两种移动方式上,并进行任意组合,且保证第一次为至少进行一次的竖直移动,移动之后的节点都会是同一个计算中心节点。
  282. steps := t.nodeIDToLocalNodeID[nodeID] + 1
  283. for d := 1; d <= steps; d++ {
  284. t.iterCombBits(len(t.nodeIDToLocalNodeID)-1, steps-d, 0, func(i int) {
  285. index := d + i
  286. node := &t.nodes[index]
  287. newMp := t.blocksMaps[node.localNodeID]
  288. newMp.Or(&node.parent.blocksBitmap)
  289. node.blocksBitmap = newMp
  290. if newMp.Weight() >= k {
  291. return
  292. }
  293. t.iterChildren(index, func(index, parentIndex, depth int) int {
  294. curNode := &t.nodes[index]
  295. parentNode := t.nodes[parentIndex]
  296. newMp := t.blocksMaps[curNode.localNodeID]
  297. newMp.Or(&parentNode.blocksBitmap)
  298. curNode.blocksBitmap = newMp
  299. if newMp.Weight() >= k {
  300. return iterActionSkip
  301. }
  302. return iterActionNone
  303. })
  304. })
  305. }
  306. }
  307. // 遍历树,找到至少拥有K个块的树节点的最大深度
  308. func (t *combinatorialTree) FindKBlocksMaxDepth(k int) int {
  309. maxDepth := -1
  310. t.iterChildren(0, func(index, parentIndex, depth int) int {
  311. if t.nodes[index].blocksBitmap.Weight() >= k {
  312. if maxDepth < depth {
  313. maxDepth = depth
  314. }
  315. return iterActionSkip
  316. }
  317. // 如果到了叶子节点,还没有找到K个块,那就认为要满足K个块,至少需要再多一个节点,即深度+1。
  318. // 由于遍历时采用的是深度优先的算法,因此遍历到这个叶子节点时,叶子节点再加一个节点的组合已经在前面搜索过,
  319. // 所以用当前叶子节点深度+1来作为当前分支的结果就可以,即使当前情况下增加任意一个节点依然不够K块,
  320. // 可以使用同样的思路去递推到当前叶子节点增加两个块的情况。
  321. if t.nodes[index].localNodeID == len(t.nodeIDToLocalNodeID)-1 {
  322. if maxDepth < depth+1 {
  323. maxDepth = depth + 1
  324. }
  325. }
  326. return iterActionNone
  327. })
  328. if maxDepth == -1 || maxDepth > len(t.nodeIDToLocalNodeID) {
  329. return len(t.nodeIDToLocalNodeID)
  330. }
  331. return maxDepth
  332. }
  333. func (t *combinatorialTree) iterCombBits(width int, count int, offset int, callback func(int)) {
  334. if count == 0 {
  335. callback(offset)
  336. return
  337. }
  338. for b := width; b >= count; b-- {
  339. t.iterCombBits(b-1, count-1, offset+(1<<b), callback)
  340. }
  341. }
  342. func (t *combinatorialTree) iterChildren(index int, do func(index int, parentIndex int, depth int) int) {
  343. curNode := &t.nodes[index]
  344. childIndex := index + 1
  345. curDepth := t.GetDepth(index)
  346. childCounts := len(t.nodeIDToLocalNodeID) - 1 - curNode.localNodeID
  347. if childCounts == 0 {
  348. return
  349. }
  350. childTreeNodeCnt := 1 << (childCounts - 1)
  351. for c := 0; c < childCounts; c++ {
  352. act := t.itering(childIndex, index, curDepth+1, do)
  353. if act == iterActionBreak {
  354. return
  355. }
  356. childIndex += childTreeNodeCnt
  357. childTreeNodeCnt >>= 1
  358. }
  359. }
  360. func (t *combinatorialTree) itering(index int, parentIndex int, depth int, do func(index int, parentIndex int, depth int) int) int {
  361. act := do(index, parentIndex, depth)
  362. if act == iterActionBreak {
  363. return act
  364. }
  365. if act == iterActionSkip {
  366. return iterActionNone
  367. }
  368. curNode := &t.nodes[index]
  369. childIndex := index + 1
  370. childCounts := len(t.nodeIDToLocalNodeID) - 1 - curNode.localNodeID
  371. if childCounts == 0 {
  372. return iterActionNone
  373. }
  374. childTreeNodeCnt := 1 << (childCounts - 1)
  375. for c := 0; c < childCounts; c++ {
  376. act = t.itering(childIndex, index, depth+1, do)
  377. if act == iterActionBreak {
  378. return act
  379. }
  380. childIndex += childTreeNodeCnt
  381. childTreeNodeCnt >>= 1
  382. }
  383. return iterActionNone
  384. }
  385. type combinatorialTreeNode struct {
  386. localNodeID int
  387. parent *combinatorialTreeNode
  388. blocksBitmap bitmap.Bitmap64 // 选择了这个中心之后,所有中心一共包含多少种块
  389. }
  390. type annealingSolution struct {
  391. blockList []objectBlock // 所有节点的块分布情况
  392. rmBlocks []bool // 要删除哪些块
  393. }
  394. func (t *CleanPinned) startAnnealing(allNodeInfos map[cdssdk.NodeID]*cdssdk.Node, readerNodeIDs []cdssdk.NodeID, object annealingObject) annealingSolution {
  395. state := &annealingState{
  396. allNodeInfos: allNodeInfos,
  397. readerNodeIDs: readerNodeIDs,
  398. nodesSortedByReader: make(map[cdssdk.NodeID][]nodeDist),
  399. object: object,
  400. nodeBlockBitmaps: make(map[cdssdk.NodeID]*bitmap.Bitmap64),
  401. }
  402. t.initBlockList(state)
  403. if state.blockList == nil {
  404. return annealingSolution{}
  405. }
  406. t.initNodeBlockBitmap(state)
  407. t.sortNodeByReaderDistance(state)
  408. state.rmBlocks = make([]bool, len(state.blockList))
  409. state.inversedIndex = -1
  410. state.nodeCombTree = newCombinatorialTree(state.nodeBlockBitmaps)
  411. state.lastScore = t.calcScore(state)
  412. state.maxScore = state.lastScore
  413. state.maxScoreRmBlocks = lo2.ArrayClone(state.rmBlocks)
  414. // 模拟退火算法的温度
  415. curTemp := state.lastScore
  416. // 结束温度
  417. finalTemp := curTemp * 0.2
  418. // 冷却率
  419. coolingRate := 0.95
  420. for curTemp > finalTemp {
  421. state.inversedIndex = rand.Intn(len(state.rmBlocks))
  422. block := state.blockList[state.inversedIndex]
  423. state.rmBlocks[state.inversedIndex] = !state.rmBlocks[state.inversedIndex]
  424. state.nodeBlockBitmaps[block.NodeID].Set(block.Index, !state.rmBlocks[state.inversedIndex])
  425. state.nodeCombTree.UpdateBitmap(block.NodeID, *state.nodeBlockBitmaps[block.NodeID], state.object.minBlockCnt)
  426. curScore := t.calcScore(state)
  427. dScore := curScore - state.lastScore
  428. // 如果新方案比旧方案得分低,且没有要求强制接受新方案,那么就将变化改回去
  429. if curScore == 0 || (dScore < 0 && !t.alwaysAccept(curTemp, dScore, coolingRate)) {
  430. state.rmBlocks[state.inversedIndex] = !state.rmBlocks[state.inversedIndex]
  431. state.nodeBlockBitmaps[block.NodeID].Set(block.Index, !state.rmBlocks[state.inversedIndex])
  432. state.nodeCombTree.UpdateBitmap(block.NodeID, *state.nodeBlockBitmaps[block.NodeID], state.object.minBlockCnt)
  433. // fmt.Printf("\n")
  434. } else {
  435. // fmt.Printf(" accept!\n")
  436. state.lastScore = curScore
  437. if state.maxScore < curScore {
  438. state.maxScore = state.lastScore
  439. state.maxScoreRmBlocks = lo2.ArrayClone(state.rmBlocks)
  440. }
  441. }
  442. curTemp *= coolingRate
  443. }
  444. // fmt.Printf("final: %v\n", state.maxScoreRmBlocks)
  445. return annealingSolution{
  446. blockList: state.blockList,
  447. rmBlocks: state.maxScoreRmBlocks,
  448. }
  449. }
  450. func (t *CleanPinned) initBlockList(ctx *annealingState) {
  451. blocksMap := make(map[cdssdk.NodeID][]objectBlock)
  452. // 先生成所有的影子块
  453. for _, pinned := range ctx.object.pinnedAt {
  454. blocks := make([]objectBlock, 0, ctx.object.totalBlockCount)
  455. for i := 0; i < ctx.object.totalBlockCount; i++ {
  456. blocks = append(blocks, objectBlock{
  457. Index: i,
  458. NodeID: pinned,
  459. HasShadow: true,
  460. })
  461. }
  462. blocksMap[pinned] = blocks
  463. }
  464. // 再填充实际块
  465. for _, b := range ctx.object.blocks {
  466. blocks := blocksMap[b.NodeID]
  467. has := false
  468. for i := range blocks {
  469. if blocks[i].Index == b.Index {
  470. blocks[i].HasEntity = true
  471. blocks[i].FileHash = b.FileHash
  472. has = true
  473. break
  474. }
  475. }
  476. if has {
  477. continue
  478. }
  479. blocks = append(blocks, objectBlock{
  480. Index: b.Index,
  481. NodeID: b.NodeID,
  482. HasEntity: true,
  483. FileHash: b.FileHash,
  484. })
  485. blocksMap[b.NodeID] = blocks
  486. }
  487. var sortedBlocks []objectBlock
  488. for _, bs := range blocksMap {
  489. sortedBlocks = append(sortedBlocks, bs...)
  490. }
  491. sortedBlocks = sort2.Sort(sortedBlocks, func(left objectBlock, right objectBlock) int {
  492. d := left.NodeID - right.NodeID
  493. if d != 0 {
  494. return int(d)
  495. }
  496. return left.Index - right.Index
  497. })
  498. ctx.blockList = sortedBlocks
  499. }
  500. func (t *CleanPinned) initNodeBlockBitmap(state *annealingState) {
  501. for _, b := range state.blockList {
  502. mp, ok := state.nodeBlockBitmaps[b.NodeID]
  503. if !ok {
  504. nb := bitmap.Bitmap64(0)
  505. mp = &nb
  506. state.nodeBlockBitmaps[b.NodeID] = mp
  507. }
  508. mp.Set(b.Index, true)
  509. }
  510. }
  511. func (t *CleanPinned) sortNodeByReaderDistance(state *annealingState) {
  512. for _, r := range state.readerNodeIDs {
  513. var nodeDists []nodeDist
  514. for n := range state.nodeBlockBitmaps {
  515. if r == n {
  516. // 同节点时距离视为0.1
  517. nodeDists = append(nodeDists, nodeDist{
  518. NodeID: n,
  519. Distance: consts.NodeDistanceSameNode,
  520. })
  521. } else if state.allNodeInfos[r].LocationID == state.allNodeInfos[n].LocationID {
  522. // 同地区时距离视为1
  523. nodeDists = append(nodeDists, nodeDist{
  524. NodeID: n,
  525. Distance: consts.NodeDistanceSameLocation,
  526. })
  527. } else {
  528. // 不同地区时距离视为5
  529. nodeDists = append(nodeDists, nodeDist{
  530. NodeID: n,
  531. Distance: consts.NodeDistanceOther,
  532. })
  533. }
  534. }
  535. state.nodesSortedByReader[r] = sort2.Sort(nodeDists, func(left, right nodeDist) int { return sort2.Cmp(left.Distance, right.Distance) })
  536. }
  537. }
  538. func (t *CleanPinned) calcScore(state *annealingState) float64 {
  539. dt := t.calcDisasterTolerance(state)
  540. ac := t.calcMinAccessCost(state)
  541. sc := t.calcSpaceCost(state)
  542. dtSc := 1.0
  543. if dt < 1 {
  544. dtSc = 0
  545. } else if dt >= 2 {
  546. dtSc = 1.5
  547. }
  548. newSc := 0.0
  549. if dt == 0 || ac == 0 {
  550. newSc = 0
  551. } else {
  552. newSc = dtSc / (sc * ac)
  553. }
  554. // fmt.Printf("solu: %v, cur: %v, dt: %v, ac: %v, sc: %v \n", state.rmBlocks, newSc, dt, ac, sc)
  555. return newSc
  556. }
  557. // 计算容灾度
  558. func (t *CleanPinned) calcDisasterTolerance(state *annealingState) float64 {
  559. if state.inversedIndex != -1 {
  560. node := state.blockList[state.inversedIndex]
  561. state.nodeCombTree.UpdateBitmap(node.NodeID, *state.nodeBlockBitmaps[node.NodeID], state.object.minBlockCnt)
  562. }
  563. return float64(len(state.nodeBlockBitmaps) - state.nodeCombTree.FindKBlocksMaxDepth(state.object.minBlockCnt))
  564. }
  565. // 计算最小访问数据的代价
  566. func (t *CleanPinned) calcMinAccessCost(state *annealingState) float64 {
  567. cost := math.MaxFloat64
  568. for _, reader := range state.readerNodeIDs {
  569. tarNodes := state.nodesSortedByReader[reader]
  570. gotBlocks := bitmap.Bitmap64(0)
  571. thisCost := 0.0
  572. for _, tar := range tarNodes {
  573. tarNodeMp := state.nodeBlockBitmaps[tar.NodeID]
  574. // 只需要从目的节点上获得缺少的块
  575. curWeigth := gotBlocks.Weight()
  576. // 下面的if会在拿到k个块之后跳出循环,所以or多了块也没关系
  577. gotBlocks.Or(tarNodeMp)
  578. // 但是算读取块的消耗时,不能多算,最多算读了k个块的消耗
  579. willGetBlocks := math2.Min(gotBlocks.Weight()-curWeigth, state.object.minBlockCnt-curWeigth)
  580. thisCost += float64(willGetBlocks) * float64(tar.Distance)
  581. if gotBlocks.Weight() >= state.object.minBlockCnt {
  582. break
  583. }
  584. }
  585. if gotBlocks.Weight() >= state.object.minBlockCnt {
  586. cost = math.Min(cost, thisCost)
  587. }
  588. }
  589. return cost
  590. }
  591. // 计算冗余度
  592. func (t *CleanPinned) calcSpaceCost(ctx *annealingState) float64 {
  593. blockCount := 0
  594. for i, b := range ctx.blockList {
  595. if ctx.rmBlocks[i] {
  596. continue
  597. }
  598. if b.HasEntity {
  599. blockCount++
  600. }
  601. if b.HasShadow {
  602. blockCount++
  603. }
  604. }
  605. // 所有算力中心上拥有的块的总数 / 一个对象被分成了几个块
  606. return float64(blockCount) / float64(ctx.object.minBlockCnt)
  607. }
  608. // 如果新方案得分比旧方案小,那么在一定概率内也接受新方案
  609. func (t *CleanPinned) alwaysAccept(curTemp float64, dScore float64, coolingRate float64) bool {
  610. v := math.Exp(dScore / curTemp / coolingRate)
  611. // fmt.Printf(" -- chance: %v, temp: %v", v, curTemp)
  612. return v > rand.Float64()
  613. }
  614. func (t *CleanPinned) makePlansForRepObject(solu annealingSolution, obj stgmod.ObjectDetail, pinPlans map[cdssdk.NodeID]*[]string) coormq.UpdatingObjectRedundancy {
  615. entry := coormq.UpdatingObjectRedundancy{
  616. ObjectID: obj.Object.ObjectID,
  617. Redundancy: obj.Object.Redundancy,
  618. }
  619. for i, f := range solu.rmBlocks {
  620. hasCache := lo.ContainsBy(obj.Blocks, func(b stgmod.ObjectBlock) bool { return b.NodeID == solu.blockList[i].NodeID }) ||
  621. lo.ContainsBy(obj.PinnedAt, func(n cdssdk.NodeID) bool { return n == solu.blockList[i].NodeID })
  622. willRm := f
  623. if !willRm {
  624. // 如果对象在退火后要保留副本的节点没有副本,则需要在这个节点创建副本
  625. if !hasCache {
  626. pinPlan, ok := pinPlans[solu.blockList[i].NodeID]
  627. if !ok {
  628. pinPlan = &[]string{}
  629. pinPlans[solu.blockList[i].NodeID] = pinPlan
  630. }
  631. *pinPlan = append(*pinPlan, obj.Object.FileHash)
  632. }
  633. entry.Blocks = append(entry.Blocks, stgmod.ObjectBlock{
  634. ObjectID: obj.Object.ObjectID,
  635. Index: solu.blockList[i].Index,
  636. NodeID: solu.blockList[i].NodeID,
  637. FileHash: obj.Object.FileHash,
  638. })
  639. }
  640. }
  641. return entry
  642. }
  643. func (t *CleanPinned) makePlansForECObject(allNodeInfos map[cdssdk.NodeID]*cdssdk.Node, solu annealingSolution, obj stgmod.ObjectDetail, planBld *plans.PlanBuilder) coormq.UpdatingObjectRedundancy {
  644. entry := coormq.UpdatingObjectRedundancy{
  645. ObjectID: obj.Object.ObjectID,
  646. Redundancy: obj.Object.Redundancy,
  647. }
  648. reconstrct := make(map[cdssdk.NodeID]*[]int)
  649. for i, f := range solu.rmBlocks {
  650. block := solu.blockList[i]
  651. if !f {
  652. entry.Blocks = append(entry.Blocks, stgmod.ObjectBlock{
  653. ObjectID: obj.Object.ObjectID,
  654. Index: block.Index,
  655. NodeID: block.NodeID,
  656. FileHash: block.FileHash,
  657. })
  658. // 如果这个块是影子块,那么就要从完整对象里重建这个块
  659. if !block.HasEntity {
  660. re, ok := reconstrct[block.NodeID]
  661. if !ok {
  662. re = &[]int{}
  663. reconstrct[block.NodeID] = re
  664. }
  665. *re = append(*re, block.Index)
  666. }
  667. }
  668. }
  669. ecRed := obj.Object.Redundancy.(*cdssdk.ECRedundancy)
  670. for id, idxs := range reconstrct {
  671. agt := planBld.AtAgent(*allNodeInfos[id])
  672. strs := agt.IPFSRead(obj.Object.FileHash).ChunkedSplit(ecRed.ChunkSize, ecRed.K, true)
  673. ss := agt.ECReconstructAny(*ecRed, lo.Range(ecRed.K), *idxs, strs.Streams...)
  674. for i, s := range ss.Streams {
  675. s.IPFSWrite(fmt.Sprintf("%d.%d", obj.Object.ObjectID, (*idxs)[i]))
  676. }
  677. }
  678. return entry
  679. }
  680. func (t *CleanPinned) executePlans(execCtx ExecuteContext, pinPlans map[cdssdk.NodeID]*[]string, planBld *plans.PlanBuilder) (map[string]any, error) {
  681. log := logger.WithType[CleanPinned]("Event")
  682. ioPlan, err := planBld.Build()
  683. if err != nil {
  684. return nil, fmt.Errorf("building io switch plan: %w", err)
  685. }
  686. // 统一加锁,有重复也没关系
  687. lockBld := reqbuilder.NewBuilder()
  688. for nodeID := range pinPlans {
  689. lockBld.IPFS().Buzy(nodeID)
  690. }
  691. for _, plan := range ioPlan.AgentPlans {
  692. lockBld.IPFS().Buzy(plan.Node.NodeID)
  693. }
  694. lock, err := lockBld.MutexLock(execCtx.Args.DistLock)
  695. if err != nil {
  696. return nil, fmt.Errorf("acquiring distlock: %w", err)
  697. }
  698. defer lock.Unlock()
  699. wg := sync.WaitGroup{}
  700. // 执行pin操作
  701. var anyPinErr error
  702. for nodeID, pin := range pinPlans {
  703. wg.Add(1)
  704. go func(nodeID cdssdk.NodeID, pin *[]string) {
  705. defer wg.Done()
  706. agtCli, err := stgglb.AgentMQPool.Acquire(nodeID)
  707. if err != nil {
  708. log.Warnf("new agent client: %s", err.Error())
  709. return
  710. }
  711. defer stgglb.AgentMQPool.Release(agtCli)
  712. _, err = agtCli.PinObject(agtmq.ReqPinObject(*pin, false))
  713. if err != nil {
  714. log.Warnf("pinning object: %s", err.Error())
  715. anyPinErr = err
  716. }
  717. }(nodeID, pin)
  718. }
  719. // 执行IO计划
  720. var ioSwRets map[string]any
  721. var ioSwErr error
  722. wg.Add(1)
  723. go func() {
  724. defer wg.Done()
  725. exec, err := plans.Execute(*ioPlan)
  726. if err != nil {
  727. ioSwErr = fmt.Errorf("executing io switch plan: %w", err)
  728. return
  729. }
  730. ret, err := exec.Wait()
  731. if err != nil {
  732. ioSwErr = fmt.Errorf("waiting io switch plan: %w", err)
  733. return
  734. }
  735. ioSwRets = ret.ResultValues
  736. }()
  737. wg.Wait()
  738. if anyPinErr != nil {
  739. return nil, anyPinErr
  740. }
  741. if ioSwErr != nil {
  742. return nil, ioSwErr
  743. }
  744. return ioSwRets, nil
  745. }
  746. func (t *CleanPinned) populateECObjectEntry(entry *coormq.UpdatingObjectRedundancy, obj stgmod.ObjectDetail, ioRets map[string]any) {
  747. for i := range entry.Blocks {
  748. if entry.Blocks[i].FileHash != "" {
  749. continue
  750. }
  751. key := fmt.Sprintf("%d.%d", obj.Object.ObjectID, entry.Blocks[i].Index)
  752. // 不应该出现key不存在的情况
  753. entry.Blocks[i].FileHash = ioRets[key].(string)
  754. }
  755. }
  756. func init() {
  757. RegisterMessageConvertor(NewCleanPinned)
  758. }

本项目旨在将云际存储公共基础设施化,使个人及企业可低门槛使用高效的云际存储服务(安装开箱即用云际存储客户端即可,无需关注其他组件的部署),同时支持用户灵活便捷定制云际存储的功能细节。