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

84 lines
1.7 KiB

3 years ago
3 years ago
3 years ago
  1. package mount
  2. import (
  3. "github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
  4. "sync"
  5. )
  6. type FileHandleToInode struct {
  7. sync.RWMutex
  8. nextFh FileHandleId
  9. inode2fh map[uint64]*FileHandle
  10. fh2inode map[FileHandleId]uint64
  11. }
  12. func NewFileHandleToInode() *FileHandleToInode {
  13. return &FileHandleToInode{
  14. inode2fh: make(map[uint64]*FileHandle),
  15. fh2inode: make(map[FileHandleId]uint64),
  16. nextFh: 0,
  17. }
  18. }
  19. func (i *FileHandleToInode) GetFileHandle(fh FileHandleId) *FileHandle {
  20. i.RLock()
  21. defer i.RUnlock()
  22. inode, found := i.fh2inode[fh]
  23. if found {
  24. return i.inode2fh[inode]
  25. }
  26. return nil
  27. }
  28. func (i *FileHandleToInode) FindFileHandle(inode uint64) (fh *FileHandle, found bool) {
  29. i.RLock()
  30. defer i.RUnlock()
  31. fh, found = i.inode2fh[inode]
  32. return
  33. }
  34. func (i *FileHandleToInode) AcquireFileHandle(wfs *WFS, inode uint64, entry *filer_pb.Entry) *FileHandle {
  35. i.Lock()
  36. defer i.Unlock()
  37. fh, found := i.inode2fh[inode]
  38. if !found {
  39. fh = newFileHandle(wfs, i.nextFh, inode, entry)
  40. i.nextFh++
  41. i.inode2fh[inode] = fh
  42. i.fh2inode[fh.fh] = inode
  43. } else {
  44. fh.counter++
  45. }
  46. return fh
  47. }
  48. func (i *FileHandleToInode) ReleaseByInode(inode uint64) {
  49. i.Lock()
  50. defer i.Unlock()
  51. fh, found := i.inode2fh[inode]
  52. if found {
  53. fh.counter--
  54. if fh.counter <= 0 {
  55. delete(i.inode2fh, inode)
  56. delete(i.fh2inode, fh.fh)
  57. }
  58. }
  59. }
  60. func (i *FileHandleToInode) ReleaseByHandle(fh FileHandleId) {
  61. i.Lock()
  62. defer i.Unlock()
  63. inode, found := i.fh2inode[fh]
  64. if found {
  65. fhHandle, fhFound := i.inode2fh[inode]
  66. if !fhFound {
  67. delete(i.fh2inode, fh)
  68. } else {
  69. fhHandle.counter--
  70. if fhHandle.counter <= 0 {
  71. delete(i.inode2fh, inode)
  72. delete(i.fh2inode, fhHandle.fh)
  73. }
  74. }
  75. }
  76. }