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.

cache.go 26 kB

8 months ago
8 months ago
8 months ago
8 months ago
8 months ago
8 months ago
8 months ago
8 months ago
7 months ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104
  1. package cache
  2. import (
  3. "errors"
  4. "io"
  5. "os"
  6. "path/filepath"
  7. "sync"
  8. "syscall"
  9. "time"
  10. "github.com/inhies/go-bytesize"
  11. "github.com/samber/lo"
  12. "gitlink.org.cn/cloudream/common/pkgs/logger"
  13. "gitlink.org.cn/cloudream/common/pkgs/trie"
  14. "gitlink.org.cn/cloudream/common/utils/io2"
  15. "gitlink.org.cn/cloudream/common/utils/lo2"
  16. "gitlink.org.cn/cloudream/common/utils/sort2"
  17. "gitlink.org.cn/cloudream/jcs-pub/client/internal/db"
  18. "gitlink.org.cn/cloudream/jcs-pub/client/internal/downloader"
  19. "gitlink.org.cn/cloudream/jcs-pub/client/internal/mount/config"
  20. "gitlink.org.cn/cloudream/jcs-pub/client/internal/mount/fuse"
  21. "gitlink.org.cn/cloudream/jcs-pub/client/internal/uploader"
  22. clitypes "gitlink.org.cn/cloudream/jcs-pub/client/types"
  23. )
  24. type CacheEntry interface {
  25. fuse.FsEntry
  26. // 在虚拟文件系统中的路径,即不包含缓存目录的路径
  27. PathComps() []string
  28. }
  29. type CacheEntryInfo struct {
  30. PathComps []string
  31. Size int64
  32. Perm os.FileMode
  33. ModTime time.Time
  34. IsDir bool
  35. // 元数据版本号
  36. MetaRevision int
  37. // 文件数据版本号
  38. DataRevision int
  39. // 引用计数
  40. RefCount int
  41. // 上次引用计数归零的时间,也即上次使用时间
  42. FreeTime time.Time
  43. // 缓存等级
  44. Level CacheLevel
  45. // 缓存等级改变时间
  46. ChangeLevelTime time.Time
  47. }
  48. type Cache struct {
  49. cfg *config.Config
  50. db *db.DB
  51. uploader *uploader.Uploader
  52. downloader *downloader.Downloader
  53. lock *sync.RWMutex
  54. cacheDone chan any
  55. doFullScan chan any
  56. activeCache *trie.Trie[*CacheFile]
  57. }
  58. func NewCache(cfg *config.Config, db *db.DB, uploader *uploader.Uploader, downloader *downloader.Downloader) *Cache {
  59. return &Cache{
  60. cfg: cfg,
  61. db: db,
  62. uploader: uploader,
  63. downloader: downloader,
  64. lock: &sync.RWMutex{},
  65. cacheDone: make(chan any, 1),
  66. doFullScan: make(chan any, 1),
  67. activeCache: trie.NewTrie[*CacheFile](),
  68. }
  69. }
  70. func (c *Cache) Start() {
  71. go c.scanningCache()
  72. go c.scanningData()
  73. }
  74. func (c *Cache) Stop() {
  75. close(c.cacheDone)
  76. }
  77. func (c *Cache) GetCacheDataPath(comps ...string) string {
  78. comps2 := make([]string, len(comps)+1)
  79. comps2[0] = c.cfg.DataDir
  80. copy(comps2[1:], comps)
  81. return filepath.Join(comps2...)
  82. }
  83. func (c *Cache) GetCacheMetaPath(comps ...string) string {
  84. comps2 := make([]string, len(comps)+1)
  85. comps2[0] = c.cfg.MetaDir
  86. copy(comps2[1:], comps)
  87. return filepath.Join(comps2...)
  88. }
  89. func (c *Cache) Dump() CacheStatus {
  90. c.lock.RLock()
  91. defer c.lock.RUnlock()
  92. var activeFiles []CacheFileStatus
  93. c.activeCache.Iterate(func(path []string, node *trie.Node[*CacheFile], isWordNode bool) trie.VisitCtrl {
  94. if node.Value == nil {
  95. return trie.VisitContinue
  96. }
  97. info := node.Value.Info()
  98. activeFiles = append(activeFiles, CacheFileStatus{
  99. Path: filepath.Join(path...),
  100. RefCount: info.RefCount,
  101. Level: info.Level.String(),
  102. IsUploading: node.Value.state.uploading != nil,
  103. })
  104. return trie.VisitContinue
  105. })
  106. return CacheStatus{
  107. ActiveFiles: activeFiles,
  108. }
  109. }
  110. func (c *Cache) ReclaimSpace() {
  111. select {
  112. case c.doFullScan <- nil:
  113. default:
  114. }
  115. }
  116. // 获取指定位置的缓存条目信息。如果路径不存在,则返回nil。
  117. func (c *Cache) Stat(pathComps []string) *CacheEntryInfo {
  118. c.lock.RLock()
  119. defer c.lock.RUnlock()
  120. node, ok := c.activeCache.WalkEnd(pathComps)
  121. if ok && node.Value != nil {
  122. info := node.Value.Info()
  123. return &info
  124. }
  125. dataPath := c.GetCacheDataPath(pathComps...)
  126. stat, err := os.Stat(dataPath)
  127. if err != nil {
  128. // TODO 日志记录
  129. return nil
  130. }
  131. if stat.IsDir() {
  132. info, err := loadCacheDirInfo(c, pathComps, stat)
  133. if err != nil {
  134. return nil
  135. }
  136. return info
  137. }
  138. info, err := loadCacheFileInfo(c, pathComps, stat)
  139. if err != nil {
  140. return nil
  141. }
  142. return info
  143. }
  144. // 创建一个缓存文件。如果文件已经存在,则会覆盖已有文件。如果加载过程中发生了错误,或者目标位置是一个目录,则会返回nil。
  145. //
  146. // 记得使用Release减少引用计数
  147. func (c *Cache) CreateFile(pathComps []string) *CacheFile {
  148. c.lock.Lock()
  149. defer c.lock.Unlock()
  150. node, ok := c.activeCache.WalkEnd(pathComps)
  151. if ok && node.Value != nil {
  152. node.Value.Delete()
  153. if node.Value.state.uploading != nil {
  154. node.Value.state.uploading.isDeleted = true
  155. }
  156. }
  157. ch, err := createNewCacheFile(c, pathComps)
  158. if err != nil {
  159. logger.Warnf("create new cache file %v: %v", pathComps, err)
  160. return nil
  161. }
  162. ch.IncRef()
  163. c.activeCache.CreateWords(pathComps).Value = ch
  164. logger.Debugf("create new cache file %v", pathComps)
  165. return ch
  166. }
  167. // 尝试加载缓存文件,如果文件不存在,则使用obj的信息创建一个新缓存文件,而如果obj为nil,那么会返回nil。
  168. //
  169. // 记得使用Release减少引用计数
  170. func (c *Cache) LoadFile(pathComps []string, obj *clitypes.Object) *CacheFile {
  171. c.lock.Lock()
  172. defer c.lock.Unlock()
  173. node, ok := c.activeCache.WalkEnd(pathComps)
  174. if ok && node.Value != nil {
  175. node.Value.IncRef()
  176. return node.Value
  177. }
  178. ch, err := loadCacheFile(c, pathComps)
  179. if err == nil {
  180. ch.remoteObj = obj
  181. ch.IncRef()
  182. c.activeCache.CreateWords(pathComps).Value = ch
  183. logger.Debugf("load cache %v", pathComps)
  184. return ch
  185. }
  186. if !os.IsNotExist(err) {
  187. // TODO 日志记录
  188. logger.Warnf("load cache %v: %v", pathComps, err)
  189. return nil
  190. }
  191. if obj == nil {
  192. return nil
  193. }
  194. ch, err = newCacheFileFromObject(c, pathComps, obj)
  195. if err != nil {
  196. logger.Warnf("create cache %v from object: %v", pathComps, err)
  197. return nil
  198. }
  199. ch.IncRef()
  200. c.activeCache.CreateWords(pathComps).Value = ch
  201. logger.Debugf("create cache %v from object %v", pathComps, obj.ObjectID)
  202. return ch
  203. }
  204. // 仅加载文件的元数据,如果文件不存在,则返回nil
  205. //
  206. // 记得使用Release减少引用计数
  207. func (c *Cache) LoadReadOnlyFile(pathComps []string) *CacheFile {
  208. c.lock.Lock()
  209. defer c.lock.Unlock()
  210. node, ok := c.activeCache.WalkEnd(pathComps)
  211. if ok && node.Value != nil {
  212. node.Value.IncRef()
  213. return node.Value
  214. }
  215. ch, err := loadReadOnlyCacheFile(c, pathComps)
  216. if err == nil {
  217. ch.IncRef()
  218. c.activeCache.CreateWords(pathComps).Value = ch
  219. logger.Debugf("load cache %v", pathComps)
  220. return ch
  221. }
  222. if !os.IsNotExist(err) {
  223. // TODO 日志记录
  224. logger.Warnf("load cache %v: %v", pathComps, err)
  225. return nil
  226. }
  227. return nil
  228. }
  229. // 创建一个缓存目录。如果目录已经存在,则会重置目录属性。如果加载过程中发生了错误,或者目标位置是一个文件,则会返回nil
  230. func (c *Cache) CreateDir(pathComps []string) *CacheDir {
  231. c.lock.Lock()
  232. defer c.lock.Unlock()
  233. ch, err := createNewCacheDir(c, pathComps)
  234. if err != nil {
  235. logger.Warnf("create cache dir: %v", err)
  236. return nil
  237. }
  238. return ch
  239. }
  240. type CreateDirOption struct {
  241. ModTime time.Time
  242. }
  243. // 加载指定缓存目录,如果目录不存在,则使用createOpt选项创建目录,而如果createOpt为nil,那么会返回nil。
  244. func (c *Cache) LoadDir(pathComps []string, createOpt *CreateDirOption) *CacheDir {
  245. c.lock.Lock()
  246. defer c.lock.Unlock()
  247. ch, err := loadCacheDir(c, pathComps)
  248. if err == nil {
  249. return ch
  250. }
  251. if !os.IsNotExist(err) {
  252. // TODO 日志记录
  253. return nil
  254. }
  255. if createOpt == nil {
  256. return nil
  257. }
  258. // 创建目录
  259. ch, err = makeCacheDirFromOption(c, pathComps, *createOpt)
  260. if err != nil {
  261. // TODO 日志记录
  262. return nil
  263. }
  264. return ch
  265. }
  266. // 加载指定路径下的所有缓存条目信息
  267. func (c *Cache) StatMany(pathComps []string) []CacheEntryInfo {
  268. c.lock.RLock()
  269. defer c.lock.RUnlock()
  270. var infos []CacheEntryInfo
  271. exists := make(map[string]bool)
  272. node, ok := c.activeCache.WalkEnd(pathComps)
  273. if ok {
  274. for name, child := range node.WordNexts {
  275. if child.Value != nil {
  276. infos = append(infos, child.Value.Info())
  277. exists[name] = true
  278. }
  279. }
  280. }
  281. osEns, err := os.ReadDir(c.GetCacheDataPath(pathComps...))
  282. if err != nil {
  283. return nil
  284. }
  285. for _, e := range osEns {
  286. if exists[e.Name()] {
  287. continue
  288. }
  289. info, err := e.Info()
  290. if err != nil {
  291. continue
  292. }
  293. if e.IsDir() {
  294. info, err := loadCacheDirInfo(c, append(lo2.ArrayClone(pathComps), e.Name()), info)
  295. if err != nil {
  296. continue
  297. }
  298. infos = append(infos, *info)
  299. } else {
  300. info, err := loadCacheFileInfo(c, append(lo2.ArrayClone(pathComps), e.Name()), info)
  301. if err != nil {
  302. continue
  303. }
  304. infos = append(infos, *info)
  305. }
  306. }
  307. return infos
  308. }
  309. // 删除指定路径的缓存文件或目录。删除目录时如果目录不为空,则会报错。
  310. func (c *Cache) Remove(pathComps []string) error {
  311. c.lock.Lock()
  312. defer c.lock.Unlock()
  313. node, ok := c.activeCache.WalkEnd(pathComps)
  314. if ok {
  315. if len(node.WordNexts) > 0 {
  316. return fuse.ErrNotEmpty
  317. }
  318. if node.Value != nil {
  319. node.Value.Delete()
  320. if node.Value.state.uploading != nil {
  321. node.Value.state.uploading.isDeleted = true
  322. }
  323. }
  324. node.RemoveSelf(true)
  325. logger.Debugf("active cache %v removed", pathComps)
  326. return nil
  327. }
  328. metaPath := c.GetCacheMetaPath(pathComps...)
  329. dataPath := c.GetCacheDataPath(pathComps...)
  330. os.Remove(metaPath)
  331. err := os.Remove(dataPath)
  332. if err == nil || os.IsNotExist(err) {
  333. logger.Debugf("local cache %v removed", pathComps)
  334. return nil
  335. }
  336. if errors.Is(err, syscall.ENOTEMPTY) {
  337. return fuse.ErrNotEmpty
  338. }
  339. return err
  340. }
  341. // 移动指定路径的缓存文件或目录到新的路径。如果目标路径已经存在,则会报错。
  342. //
  343. // 如果移动成功,则返回移动后的缓存文件或目录。如果文件或目录不存在,则返回nil。
  344. func (c *Cache) Move(pathComps []string, newPathComps []string) error {
  345. c.lock.Lock()
  346. defer c.lock.Unlock()
  347. _, ok := c.activeCache.WalkEnd(newPathComps)
  348. if ok {
  349. return fuse.ErrExists
  350. }
  351. newMetaPath := c.GetCacheMetaPath(newPathComps...)
  352. newDataPath := c.GetCacheDataPath(newPathComps...)
  353. _, err := os.Stat(newDataPath)
  354. if err == nil {
  355. return fuse.ErrExists
  356. }
  357. if !os.IsNotExist(err) {
  358. return err
  359. }
  360. oldMetaPath := c.GetCacheMetaPath(pathComps...)
  361. oldDataPath := c.GetCacheDataPath(pathComps...)
  362. // 确定源文件存在,再进行后面的操作
  363. _, err = os.Stat(oldDataPath)
  364. if err != nil {
  365. if os.IsNotExist(err) {
  366. return fuse.ErrNotExists
  367. }
  368. return err
  369. }
  370. // 创建父目录是为了解决被移动的文件不在本地的问题。
  371. // 但同时也导致了如果目的路径的父目录确实不存在,这里会意外的创建了这个目录
  372. newMetaDir := filepath.Dir(newMetaPath)
  373. err = os.MkdirAll(newMetaDir, 0755)
  374. if err != nil {
  375. return err
  376. }
  377. newDataDir := filepath.Dir(newDataPath)
  378. err = os.MkdirAll(newDataDir, 0755)
  379. if err != nil {
  380. return err
  381. }
  382. // 每个缓存文件持有meta文件和data文件的句柄,所以这里移动文件,不影响句柄的使用。
  383. // 只能忽略这里的错误
  384. os.Rename(oldMetaPath, newMetaPath)
  385. os.Rename(oldDataPath, newDataPath)
  386. // 更新缓存
  387. oldNode, ok := c.activeCache.WalkEnd(pathComps)
  388. if ok {
  389. newNode := c.activeCache.CreateWords(newPathComps)
  390. newNode.Value = oldNode.Value
  391. newNode.WordNexts = oldNode.WordNexts
  392. oldNode.RemoveSelf(false)
  393. if newNode.Value != nil {
  394. newNode.Value.Move(newPathComps)
  395. }
  396. newNode.Iterate(func(path []string, node *trie.Node[*CacheFile], isWordNode bool) trie.VisitCtrl {
  397. if node.Value != nil {
  398. node.Value.Move(lo2.AppendNew(newPathComps, path...))
  399. }
  400. return trie.VisitContinue
  401. })
  402. }
  403. logger.Debugf("cache moved: %v -> %v", pathComps, newPathComps)
  404. return nil
  405. }
  406. type syncPackage struct {
  407. bktName string
  408. pkgName string
  409. pkg clitypes.Package
  410. upObjs []*uploadingObject
  411. }
  412. type uploadingObject struct {
  413. pathComps []string
  414. cache *CacheFile
  415. reader *CacheFileHandle
  416. modTime time.Time
  417. metaRevision int
  418. isDeleted bool
  419. isSuccess bool
  420. }
  421. type packageFullName struct {
  422. bktName string
  423. pkgName string
  424. }
  425. func (c *Cache) scanningCache() {
  426. ticker := time.NewTicker(time.Second * 5)
  427. defer ticker.Stop()
  428. lastScanPath := []string{}
  429. nextFullScan := false
  430. for {
  431. select {
  432. case _, ok := <-c.cacheDone:
  433. if !ok {
  434. return
  435. }
  436. case <-ticker.C:
  437. case <-c.doFullScan:
  438. nextFullScan = true
  439. }
  440. // 每完成一轮快速的渐进全量扫描,就进行一次即时全量扫描
  441. if nextFullScan {
  442. c.fullScan()
  443. nextFullScan = false
  444. continue
  445. }
  446. lastScanPath = c.fastScan(lastScanPath)
  447. if len(lastScanPath) == 0 {
  448. nextFullScan = true
  449. }
  450. }
  451. }
  452. // 全量扫描,主要是检查总缓存大小是否超标
  453. func (c *Cache) fullScan() {
  454. log := logger.WithField("Mod", "Mount")
  455. startTime := time.Now()
  456. log.Debug("begin full scan")
  457. defer func() {
  458. log.Debugf("full scan done, time: %v", time.Since(startTime))
  459. }()
  460. c.lock.Lock()
  461. defer c.lock.Unlock()
  462. totalCacheSize := int64(0)
  463. type readOnlyCache struct {
  464. Info CacheEntryInfo
  465. Node *trie.Node[*CacheFile]
  466. }
  467. var readOnlyCaches []readOnlyCache
  468. c.activeCache.Iterate(func(path []string, node *trie.Node[*CacheFile], isWordNode bool) trie.VisitCtrl {
  469. ch := node.Value
  470. if ch == nil {
  471. return trie.VisitContinue
  472. }
  473. info := ch.Info()
  474. if info.Level > LevelReadOnly {
  475. return trie.VisitContinue
  476. }
  477. if info.DataRevision > 0 || info.MetaRevision > 0 {
  478. return trie.VisitContinue
  479. }
  480. readOnlyCaches = append(readOnlyCaches, readOnlyCache{
  481. Info: info,
  482. Node: node,
  483. })
  484. totalCacheSize += info.Size
  485. return trie.VisitContinue
  486. })
  487. // 如果总缓存文件大小超过限制,那么就从最早被使用的开始删除
  488. if c.cfg.MaxCacheSize > 0 {
  489. needReclaim := totalCacheSize - c.cfg.MaxCacheSize
  490. if needReclaim > 0 {
  491. readOnlyCaches = sort2.Sort(readOnlyCaches, func(left, right readOnlyCache) int {
  492. return left.Info.FreeTime.Compare(right.Info.FreeTime)
  493. })
  494. reclaimed := int64(0)
  495. rmCnt := 0
  496. for _, rc := range readOnlyCaches {
  497. rc.Node.Value.Delete()
  498. rc.Node.RemoveSelf(true)
  499. needReclaim -= rc.Info.Size
  500. reclaimed += rc.Info.Size
  501. rmCnt += 1
  502. if needReclaim <= 0 {
  503. break
  504. }
  505. }
  506. log.Infof("%v cache file removed, reclaimed %v bytes, total cache size: %v", rmCnt, reclaimed, totalCacheSize-reclaimed)
  507. }
  508. }
  509. // TODO 还可以做点其他的检查,比如文件句柄数
  510. }
  511. // 快速扫描,每次只扫描一部分节点,做的事情会繁重一点
  512. func (c *Cache) fastScan(lastScanPath []string) []string {
  513. c.lock.Lock()
  514. uploadingPkgs := make(map[packageFullName]*syncPackage)
  515. visitCnt := 0
  516. visitBreak := false
  517. node, _ := c.activeCache.WalkEnd(lastScanPath)
  518. node.Iterate(func(path []string, node *trie.Node[*CacheFile], isWordNode bool) trie.VisitCtrl {
  519. ch := node.Value
  520. if ch == nil {
  521. return trie.VisitContinue
  522. }
  523. info := ch.Info()
  524. if info.RefCount > 0 {
  525. // logger.Debugf("skip cache %v, refCount: %v", path, info.RefCount)
  526. return trie.VisitContinue
  527. }
  528. visitCnt++
  529. c.visitNode(path, node, ch, info, uploadingPkgs)
  530. // 每次最多遍历500个节点,防止占用锁太久
  531. if visitCnt > 500 {
  532. lastScanPath = lo2.ArrayClone(path)
  533. visitBreak = true
  534. return trie.VisitBreak
  535. }
  536. return trie.VisitContinue
  537. })
  538. if !visitBreak {
  539. lastScanPath = []string{}
  540. }
  541. c.lock.Unlock()
  542. if len(uploadingPkgs) > 0 {
  543. go c.doSync(lo.Values(uploadingPkgs))
  544. }
  545. return lastScanPath
  546. }
  547. func (c *Cache) visitNode(path []string, node *trie.Node[*CacheFile], ch *CacheFile, info CacheEntryInfo, uploadingPkgs map[packageFullName]*syncPackage) {
  548. shouldUpload := true
  549. // 不存放在Package里的文件,不需要上传
  550. if len(ch.pathComps) <= 2 {
  551. shouldUpload = false
  552. }
  553. // 1. 本地缓存被修改了,如果一段时间内没有被使用,则进行上传
  554. if shouldUpload && (info.DataRevision > 0 || info.MetaRevision > 0) {
  555. if time.Since(info.FreeTime) < c.cfg.UploadPendingTime {
  556. return
  557. }
  558. if ch.state.uploading != nil {
  559. return
  560. }
  561. // 上传文件需要完全加载级别的缓存等级
  562. if !ch.LevelUp(LevelComplete) {
  563. return
  564. }
  565. fullName := packageFullName{ch.pathComps[0], ch.pathComps[1]}
  566. pkg, ok := uploadingPkgs[fullName]
  567. if !ok {
  568. pkg = &syncPackage{
  569. bktName: ch.pathComps[0],
  570. pkgName: ch.pathComps[1],
  571. }
  572. uploadingPkgs[fullName] = pkg
  573. }
  574. up := &uploadingObject{
  575. pathComps: lo2.ArrayClone(ch.pathComps),
  576. cache: ch,
  577. }
  578. pkg.upObjs = append(pkg.upObjs, up)
  579. ch.state.uploading = up
  580. if info.DataRevision > 0 {
  581. up.reader = ch.OpenReadWhenScanning()
  582. }
  583. if info.MetaRevision > 0 {
  584. up.modTime = info.ModTime
  585. up.metaRevision = info.MetaRevision
  586. }
  587. return
  588. }
  589. // 2. 本地缓存没有被修改,如果一段时间内没有被使用,则进行卸载
  590. if info.Level > LevelReadOnly {
  591. if time.Since(info.FreeTime) > c.cfg.CacheActiveTime {
  592. ch.LevelDown(LevelReadOnly)
  593. }
  594. return
  595. }
  596. // 3. 卸载后的缓存,如果一段时间内没有被使用,则进行删除。
  597. if info.Level <= LevelReadOnly {
  598. // 需要同时满足距上次使用时间和距上次卸载时间超过配置的时间,才可以删除
  599. if time.Since(info.FreeTime) > c.cfg.CacheExpireTime && time.Since(info.ChangeLevelTime) > c.cfg.CacheExpireTime {
  600. // 如果文件已经同步到远端,则可以直接删除本地缓存
  601. if info.MetaRevision == 0 && info.DataRevision == 0 {
  602. ch.Delete()
  603. }
  604. // 文件数据或者元数据有修改,但缓存等级是ReadOnly以下,意味着在之前检查是否需要上传时被判定为不需要上传
  605. // 这种文件删除缓存记录即可(但会在扫描数据目录时再次被加载进来)
  606. node.RemoveSelf(true)
  607. }
  608. return
  609. }
  610. }
  611. // 扫描文件数据目录
  612. func (c *Cache) scanningData() {
  613. ticker := time.NewTicker(c.cfg.ScanDataDirInterval)
  614. defer ticker.Stop()
  615. var walkTrace []*os.File
  616. var walkTraceComps []string
  617. for {
  618. select {
  619. case <-ticker.C:
  620. case <-c.cacheDone:
  621. return
  622. }
  623. startTime := time.Now()
  624. logger.Infof("begin scanning data dir")
  625. if len(walkTrace) == 0 {
  626. dir, err := os.Open(c.cfg.DataDir)
  627. if err != nil {
  628. logger.Warnf("open data dir: %v", err)
  629. continue
  630. }
  631. walkTrace = []*os.File{dir}
  632. walkTraceComps = []string{c.cfg.MetaDir}
  633. }
  634. const maxVisitCnt = 5000
  635. const maxUntrackedFiles = 500
  636. var untrackedFiles [][]string
  637. visitCnt := 0
  638. // 一次最多遍历5000个文件(包括路径上的文件夹),一次最多添加500个未跟踪文件
  639. for len(walkTrace) > 0 && visitCnt < maxVisitCnt && len(untrackedFiles) < maxUntrackedFiles {
  640. lastNode := walkTrace[len(walkTrace)-1]
  641. visitCnt++
  642. e, err := lastNode.Readdir(1)
  643. if err == io.EOF {
  644. lastNode.Close()
  645. walkTrace = walkTrace[:len(walkTrace)-1]
  646. walkTraceComps = walkTraceComps[:len(walkTraceComps)-1]
  647. continue
  648. }
  649. if err != nil {
  650. logger.Warnf("read dir %v: %v", lastNode.Name(), err)
  651. lastNode.Close()
  652. walkTrace = walkTrace[:len(walkTrace)-1]
  653. walkTraceComps = walkTraceComps[:len(walkTraceComps)-1]
  654. continue
  655. }
  656. // 按Readdir函数的说法,不会存在len(e) == 0且err == nil的情况,但实际发生了
  657. if len(e) == 0 {
  658. continue
  659. }
  660. if e[0].IsDir() {
  661. child, err := os.Open(filepath.Join(lastNode.Name(), e[0].Name()))
  662. if err != nil {
  663. logger.Warnf("open dir %v: %v", e[0].Name(), err)
  664. continue
  665. }
  666. walkTrace = append(walkTrace, child)
  667. walkTraceComps = append(walkTraceComps, e[0].Name())
  668. continue
  669. }
  670. // 对于不在Package层级的文件,不跟踪
  671. if len(walkTrace) <= 2 {
  672. continue
  673. }
  674. // 无条件加载缓存,可能会导致一些不需要被同步到云端的文件在缓存等级降到最低取消跟踪后,又重新被加载进来
  675. // 不过由于扫描频率不高,所以问题不大
  676. walkTraceComps = append(walkTraceComps, e[0].Name())
  677. untrackedFiles = append(untrackedFiles, lo2.ArrayClone(walkTraceComps[1:]))
  678. walkTraceComps = walkTraceComps[:len(walkTraceComps)-1]
  679. }
  680. if len(untrackedFiles) > 0 {
  681. for _, comps := range untrackedFiles {
  682. ch := c.LoadReadOnlyFile(comps)
  683. if ch != nil {
  684. ch.Release()
  685. }
  686. }
  687. }
  688. logger.Infof("%v file visited, %v untracked files found, time: %v", visitCnt, len(untrackedFiles), time.Since(startTime))
  689. }
  690. }
  691. func (c *Cache) doSync(pkgs []*syncPackage) {
  692. var uploadPkgs []*syncPackage
  693. var updateOnlyPkgs []*syncPackage
  694. for _, p := range pkgs {
  695. var updateOnly *syncPackage
  696. var upload *syncPackage
  697. for _, o := range p.upObjs {
  698. if o.reader != nil {
  699. if upload == nil {
  700. upload = &syncPackage{
  701. bktName: p.bktName,
  702. pkgName: p.pkgName,
  703. }
  704. }
  705. upload.upObjs = append(upload.upObjs, o)
  706. } else {
  707. if updateOnly == nil {
  708. updateOnly = &syncPackage{
  709. bktName: p.bktName,
  710. pkgName: p.pkgName,
  711. }
  712. }
  713. updateOnly.upObjs = append(updateOnly.upObjs, o)
  714. }
  715. }
  716. if upload != nil {
  717. uploadPkgs = append(uploadPkgs, upload)
  718. }
  719. if updateOnly != nil {
  720. updateOnlyPkgs = append(updateOnlyPkgs, updateOnly)
  721. }
  722. }
  723. // 先上传文件,再更新文件元数据。上传文件时会创建Package,这样后续更新元数据时就能查到Package。
  724. if len(uploadPkgs) > 0 {
  725. c.doUploading(uploadPkgs)
  726. }
  727. if len(updateOnlyPkgs) > 0 {
  728. c.doUpdatingOnly(updateOnlyPkgs)
  729. }
  730. }
  731. func (c *Cache) doUpdatingOnly(pkgs []*syncPackage) {
  732. /// 1. 只是更新元数据,那么就只尝试查询Package
  733. var sucPkgs []*syncPackage
  734. var failedPkgs []*syncPackage
  735. for _, pkg := range pkgs {
  736. p, err := c.db.Package().GetByFullName(c.db.DefCtx(), pkg.bktName, pkg.pkgName)
  737. if err != nil {
  738. logger.Warnf("get package %v/%v: %v", pkg.bktName, pkg.pkgName, err)
  739. failedPkgs = append(failedPkgs, pkg)
  740. continue
  741. }
  742. pkg.pkg = p
  743. sucPkgs = append(sucPkgs, pkg)
  744. }
  745. /// 2. 对于创建失败的Package, 在锁的保护下取消上传状态
  746. c.lock.Lock()
  747. for _, pkg := range failedPkgs {
  748. for _, obj := range pkg.upObjs {
  749. obj.cache.state.uploading = nil
  750. }
  751. }
  752. c.lock.Unlock()
  753. /// 3. 开始更新每个Package
  754. for _, p := range sucPkgs {
  755. pathes := make([]string, 0, len(p.upObjs))
  756. modTimes := make([]time.Time, 0, len(p.upObjs))
  757. for _, obj := range p.upObjs {
  758. pathes = append(pathes, clitypes.JoinObjectPath(obj.pathComps[2:]...))
  759. modTimes = append(modTimes, obj.modTime)
  760. }
  761. err := c.db.Object().BatchUpdateUpdateTimeByPath(c.db.DefCtx(), p.pkg.PackageID, pathes, modTimes)
  762. if err != nil {
  763. logger.Warnf("batch update package %v/%v: %v", p.bktName, p.pkgName, err)
  764. c.lock.Lock()
  765. for _, obj := range p.upObjs {
  766. obj.cache.state.uploading = nil
  767. }
  768. c.lock.Unlock()
  769. continue
  770. }
  771. logger.Infof("update %v object in package %v/%v", len(p.upObjs), p.bktName, p.pkgName)
  772. // 登记上传结果
  773. c.lock.Lock()
  774. for _, obj := range p.upObjs {
  775. obj.cache.state.uploading = nil
  776. obj.cache.RevisionUploaded(0, obj.metaRevision)
  777. }
  778. c.lock.Unlock()
  779. }
  780. }
  781. func (c *Cache) doUploading(pkgs []*syncPackage) {
  782. /// 1. 先尝试创建Package
  783. var sucPkgs []*syncPackage
  784. var failedPkgs []*syncPackage
  785. for _, pkg := range pkgs {
  786. p, err := db.DoTx21(c.db, c.db.Package().TryCreateAll, pkg.bktName, pkg.pkgName)
  787. if err != nil {
  788. logger.Warnf("try create package %v/%v: %v", pkg.bktName, pkg.pkgName, err)
  789. failedPkgs = append(failedPkgs, pkg)
  790. continue
  791. }
  792. pkg.pkg = p
  793. sucPkgs = append(sucPkgs, pkg)
  794. }
  795. /// 2. 对于创建失败的Package,直接关闭文件,不进行上传
  796. // 在锁的保护下取消上传状态
  797. c.lock.Lock()
  798. for _, pkg := range failedPkgs {
  799. for _, obj := range pkg.upObjs {
  800. obj.cache.state.uploading = nil
  801. }
  802. }
  803. c.lock.Unlock()
  804. for _, pkg := range failedPkgs {
  805. for _, obj := range pkg.upObjs {
  806. obj.reader.Close()
  807. }
  808. }
  809. /// 3. 开始上传每个Package
  810. for _, p := range sucPkgs {
  811. upder, err := c.uploader.BeginUpdate(p.pkg.PackageID, 0, nil, nil)
  812. if err != nil {
  813. logger.Warnf("begin upload package %v/%v: %v", p.bktName, p.pkgName, err)
  814. // 取消上传状态
  815. c.lock.Lock()
  816. for _, obj := range p.upObjs {
  817. obj.cache.state.uploading = nil
  818. }
  819. c.lock.Unlock()
  820. for _, obj := range p.upObjs {
  821. obj.reader.Close()
  822. }
  823. continue
  824. }
  825. upSuc := 0
  826. upSucAmt := int64(0)
  827. upFailed := 0
  828. upStartTime := time.Now()
  829. logger.Infof("begin uploading %v objects to package %v/%v", len(p.upObjs), p.bktName, p.pkgName)
  830. for _, o := range p.upObjs {
  831. rd := cacheFileReader{
  832. rw: o.reader,
  833. }
  834. counter := io2.Counter(&rd)
  835. err = upder.Upload(clitypes.PathFromComps(o.pathComps[2:]...), counter, uploader.UploadOption{
  836. CreateTime: o.modTime,
  837. })
  838. if err != nil {
  839. logger.Warnf("upload object %v: %v", o.pathComps, err)
  840. upFailed++
  841. continue
  842. }
  843. o.isSuccess = true
  844. upSuc++
  845. upSucAmt += counter.Count()
  846. }
  847. // 在锁保护下登记上传结果
  848. c.lock.Lock()
  849. upCancel := 0
  850. upRename := 0
  851. // 检查是否有文件在上传期间发生了变化
  852. var sucObjs []*uploadingObject
  853. for _, o := range p.upObjs {
  854. o.cache.state.uploading = nil
  855. if !o.isSuccess {
  856. continue
  857. }
  858. oldPath := clitypes.JoinObjectPath(o.pathComps[2:]...)
  859. newPath := clitypes.JoinObjectPath(o.cache.pathComps[2:]...)
  860. if o.isDeleted {
  861. upder.CancelObject(oldPath)
  862. upCancel++
  863. continue
  864. }
  865. // 如果对象移动到了另一个Package,那么也要取消上传
  866. if !lo2.ArrayEquals(o.pathComps[:2], o.cache.pathComps[:2]) {
  867. upder.CancelObject(oldPath)
  868. upCancel++
  869. continue
  870. }
  871. // 只有仍在同Package内移动的对象才能直接重命名
  872. if newPath != oldPath {
  873. upder.RenameObject(oldPath, newPath)
  874. upRename++
  875. }
  876. sucObjs = append(sucObjs, o)
  877. }
  878. _, err = upder.Commit()
  879. if err != nil {
  880. logger.Warnf("commit update package %v/%v: %v", p.bktName, p.pkgName, err)
  881. } else {
  882. for _, obj := range sucObjs {
  883. obj.cache.RevisionUploaded(obj.reader.revision, obj.metaRevision)
  884. }
  885. upTime := time.Since(upStartTime)
  886. logger.Infof("upload package %v/%v in %v, upload: %v, size: %v, speed: %v/s, cancel: %v, rename: %v",
  887. p.bktName, p.pkgName, upTime, upSuc, upSucAmt, bytesize.New(float64(upSucAmt)/upTime.Seconds()), upCancel, upRename)
  888. }
  889. c.lock.Unlock()
  890. // 关闭文件会影响refCount,所以无论是上传失败还是上传成功,都会在等待一段时间后才进行下一阶段的操作
  891. for _, obj := range p.upObjs {
  892. obj.reader.Close()
  893. }
  894. }
  895. }
  896. type cacheFileReader struct {
  897. rw *CacheFileHandle
  898. pos int64
  899. }
  900. func (r *cacheFileReader) Read(p []byte) (int, error) {
  901. n, err := r.rw.ReadAt(p, r.pos)
  902. r.pos += int64(n)
  903. if err != nil {
  904. return n, err
  905. }
  906. if n != len(p) {
  907. return n, io.EOF
  908. }
  909. return n, nil
  910. }

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