github.com/iaas-resource-provision/iaas-rpc@v1.0.7-0.20211021023331-ed21f798c408/internal/command/testdata/statelocker.go (about)

     1  // statelocker use used for testing command with a locked state.
     2  // This will lock the state file at a given path, then wait for a sigal. On
     3  // SIGINT and SIGTERM the state will be Unlocked before exit.
     4  package main
     5  
     6  import (
     7  	"io"
     8  	"log"
     9  	"os"
    10  	"os/signal"
    11  	"syscall"
    12  	"time"
    13  
    14  	"github.com/iaas-resource-provision/iaas-rpc/internal/command/clistate"
    15  	"github.com/iaas-resource-provision/iaas-rpc/internal/states/statemgr"
    16  )
    17  
    18  func main() {
    19  	if len(os.Args) != 2 {
    20  		log.Fatal(os.Args[0], "statefile")
    21  	}
    22  
    23  	s := &clistate.LocalState{
    24  		Path: os.Args[1],
    25  	}
    26  
    27  	info := statemgr.NewLockInfo()
    28  	info.Operation = "test"
    29  	info.Info = "state locker"
    30  
    31  	lockID, err := s.Lock(info)
    32  	if err != nil {
    33  		io.WriteString(os.Stderr, err.Error())
    34  		return
    35  	}
    36  
    37  	// signal to the caller that we're locked
    38  	io.WriteString(os.Stdout, "LOCKID "+lockID)
    39  
    40  	defer func() {
    41  		if err := s.Unlock(lockID); err != nil {
    42  			io.WriteString(os.Stderr, err.Error())
    43  		}
    44  	}()
    45  
    46  	c := make(chan os.Signal, 1)
    47  	signal.Notify(c, syscall.SIGINT, syscall.SIGTERM, syscall.SIGHUP)
    48  
    49  	// timeout after 10 second in case we don't get cleaned up by the test
    50  	select {
    51  	case <-time.After(10 * time.Second):
    52  	case <-c:
    53  	}
    54  }