mirror of
https://github.com/tailscale/tailscale.git
synced 2025-12-08 19:02:03 +01:00
ece6e27f39
Implements a new disk put function for cigocacher that does not cause locking issues on Windows when there are multiple processes reading and writing the same files concurrently. Integrates cigocacher into test.yml for Windows where we are running on larger runners that support connecting to private Azure vnet resources where cigocached is hosted. Updates tailscale/corp#10808 Change-Id: I0d0e9b670e49e0f9abf01ff3d605cd660dd85ebb Signed-off-by: Tom Proctor <tomhjp@users.noreply.github.com>
45 lines
860 B
Go
45 lines
860 B
Go
// Copyright (c) Tailscale Inc & AUTHORS
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
//go:build !windows
|
|
|
|
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
func writeActionFile(dest string, b []byte) error {
|
|
_, err := writeAtomic(dest, bytes.NewReader(b))
|
|
return err
|
|
}
|
|
|
|
func writeOutputFile(dest string, r io.Reader, _ int64, _ string) (int64, error) {
|
|
return writeAtomic(dest, r)
|
|
}
|
|
|
|
func writeAtomic(dest string, r io.Reader) (int64, error) {
|
|
tf, err := os.CreateTemp(filepath.Dir(dest), filepath.Base(dest)+".*")
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
size, err := io.Copy(tf, r)
|
|
if err != nil {
|
|
tf.Close()
|
|
os.Remove(tf.Name())
|
|
return 0, err
|
|
}
|
|
if err := tf.Close(); err != nil {
|
|
os.Remove(tf.Name())
|
|
return 0, err
|
|
}
|
|
if err := os.Rename(tf.Name(), dest); err != nil {
|
|
os.Remove(tf.Name())
|
|
return 0, err
|
|
}
|
|
return size, nil
|
|
}
|