mirror of
https://github.com/kubevirt/containerized-data-importer.git
synced 2025-06-03 06:30:22 +00:00
aaacbae797
The io/ioutil package has been deprecated as of Go 1.16 [1]. This commit replaces the existing io/ioutil functions with their new definitions in io and os packages. [1]: https://golang.org/doc/go1.16#ioutil Signed-off-by: Eng Zer Jun <engzerjun@gmail.com>
83 lines
2.0 KiB
Go
Executable File
83 lines
2.0 KiB
Go
Executable File
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
func printFiles(dir string) error {
|
|
return filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
|
|
fmt.Println(path)
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func getImageFilename(dir string) (string, error) {
|
|
entries, err := os.ReadDir(dir)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if len(entries) != 1 || entries[0].IsDir() {
|
|
return "", errors.Errorf("Invalid container image")
|
|
}
|
|
return entries[0].Name(), nil
|
|
}
|
|
|
|
func main() {
|
|
port := flag.Int("p", 8100, "server port")
|
|
directory := flag.String("image-dir", ".", "directory to serve")
|
|
readyFile := flag.String("ready-file", "/shared/ready", "file to create when ready for connections")
|
|
doneFile := flag.String("done-file", "/shared/done", "file created when the client is done")
|
|
flag.Parse()
|
|
|
|
if err := printFiles(*directory); err != nil {
|
|
log.Fatalf("Failed walking the directory %s: %v", *directory, err)
|
|
}
|
|
imageFilename, err := getImageFilename(*directory)
|
|
if err != nil {
|
|
log.Fatalf("Failed get image filename in %s: %v", *directory, err)
|
|
}
|
|
server := &http.Server{
|
|
Handler: http.FileServer(http.Dir(*directory)),
|
|
}
|
|
addr := fmt.Sprintf("localhost:%d", *port)
|
|
listener, err := net.Listen("tcp", addr)
|
|
if err != nil {
|
|
log.Fatalf("Failed listening on %s err: %v", addr, err)
|
|
}
|
|
|
|
if err := os.WriteFile(*readyFile, []byte(imageFilename), 0666); err != nil {
|
|
log.Fatalf("Failed creating \"ready\" file: %v", err)
|
|
}
|
|
defer os.Remove(*readyFile)
|
|
|
|
go func() {
|
|
log.Printf("Serving %s on HTTP port: %d\n", *directory, *port)
|
|
if err := server.Serve(listener); err != nil && err != http.ErrServerClosed {
|
|
log.Fatalf("Serve failed: %v", err)
|
|
}
|
|
}()
|
|
|
|
for {
|
|
if _, err := os.Stat(*doneFile); err == nil {
|
|
break
|
|
}
|
|
time.Sleep(time.Second)
|
|
}
|
|
|
|
os.Remove(*doneFile)
|
|
if err := server.Shutdown(context.TODO()); err != nil {
|
|
log.Printf("Shutdown failed: %v\n", err)
|
|
}
|
|
log.Println("Importer has completed")
|
|
}
|