preallocate_unix.go 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. // Copyright 2016 The etcd Authors
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. // +build linux
  15. package fileutil
  16. import (
  17. "os"
  18. "syscall"
  19. )
  20. func preallocExtend(f *os.File, sizeInBytes int64) error {
  21. // use mode = 0 to change size
  22. err := syscall.Fallocate(int(f.Fd()), 0, 0, sizeInBytes)
  23. if err != nil {
  24. errno, ok := err.(syscall.Errno)
  25. // not supported; fallback
  26. // fallocate EINTRs frequently in some environments; fallback
  27. if ok && (errno == syscall.ENOTSUP || errno == syscall.EINTR) {
  28. return preallocExtendTrunc(f, sizeInBytes)
  29. }
  30. }
  31. return err
  32. }
  33. func preallocFixed(f *os.File, sizeInBytes int64) error {
  34. // use mode = 1 to keep size; see FALLOC_FL_KEEP_SIZE
  35. err := syscall.Fallocate(int(f.Fd()), 1, 0, sizeInBytes)
  36. if err != nil {
  37. errno, ok := err.(syscall.Errno)
  38. // treat not supported as nil error
  39. if ok && errno == syscall.ENOTSUP {
  40. return nil
  41. }
  42. }
  43. return err
  44. }