github.com/pingcap/tiflow@v0.0.0-20240520035814-5bf52d54e205/pkg/fsutil/preallocate_linux.go (about)

     1  // Copyright 2022 PingCAP, Inc.
     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  // See the License for the specific language governing permissions and
    12  // limitations under the License.
    13  //
    14  //go:build linux
    15  
    16  package fsutil
    17  
    18  import (
    19  	"os"
    20  	"syscall"
    21  
    22  	cerror "github.com/pingcap/tiflow/pkg/errors"
    23  	"golang.org/x/sys/unix"
    24  )
    25  
    26  // PreAllocate can allocate disk space beforehand in case of ENOSPC error occurs
    27  // when writing a file, which is not uncommon for a rigorous redolog/WAL design.
    28  // Besides, pre-allocating disk space can reduce the overhead of file metadata updating.
    29  func PreAllocate(f *os.File, size int64) error {
    30  	if size == 0 {
    31  		return nil
    32  	}
    33  
    34  	// fallocate is supported for xfs/ext4/btrfs, etc.
    35  	err := unix.Fallocate(int(f.Fd()), 0, 0, size)
    36  	if err == unix.ENOTSUP || err == syscall.EINTR {
    37  		// if not supported, then fallback to use truncate.
    38  		return f.Truncate(size)
    39  	}
    40  
    41  	if err == unix.ENOSPC {
    42  		return cerror.ErrDiskFull
    43  	}
    44  
    45  	return err
    46  }