file_sftp_handler.go

package main

import (
	"time"
	"log"
	"sync"

	"github.com/pkg/sftp"
)

var (
	wg sync.WaitGroup
	err           error
	sftpClient    *sftp.Client
)

func fileSftpHandler() {
	// Connect
	sftpClient, err = connect(config.SshUserName, config.SshPassword, config.SshHost, config.SshPort)

	if err != nil {
		log.Fatal("SSH connect error: ", err)
	}

	defer sftpClient.Close()

	// Check if the remote dir exist
	if _, err := sftpClient.Stat(remoteDir); err != nil {
		panic("Remote dir dose not exist: " + remoteDir)
	}

	wg.Add(5)

	// Create Event handler
	go func() {
		for {
			select {

			case <-time.After(fileHandleTimeOut):
				//timeout for 3 seconds

			case createFileName := <-fileCreateEvent:
				log.Print("createFile:" + createFileName)

				if isDir(createFileName) == 1 {
					uploadDirectory(createFileName, remoteDir)
				} else {
					uploadFile(createFileName, remoteDir)
				}
			}
		}

		defer wg.Done()
	}()

	// Write event handler
	go func() {
		for {
			select {

			case <-time.After(fileHandleTimeOut):
				//timeout for 3 seconds

			case writeFileName := <-fileWriteEvent:
				log.Print("write file: " + writeFileName)

				isDir := isDir(writeFileName)

				if isDir == 1 {
					uploadDirectory(writeFileName, remoteDir)
				} else if isDir == 0 {
					uploadFile(writeFileName, remoteDir)
				}
			}
		}

		defer wg.Done()
	}()

	// Remove event handler
	go func() {
		for {
			select {

			case <-time.After(fileHandleTimeOut):
				//timeout for 3 seconds

			case removeFileName := <-fileRemoveEvent:
				log.Print("remove file: " + removeFileName)

				remove(removeFileName, remoteDir)
			}
		}

		defer wg.Done()
	}()

	// Rename event handler
	go func() {
		for {
			select {

			case <-time.After(fileHandleTimeOut):
				//timeout for 3 seconds

			case renameFileName := <-fileRenameEvent:
				log.Print("rename file: " + renameFileName)

				// Rename same as remove
				remove(renameFileName, remoteDir)
			}
		}

		defer wg.Done()
	}()

	// Chmod event handler
	go func() {
		for {
			select {

			case <-time.After(fileHandleTimeOut):
				//timeout for 3 seconds

			case chmodFileName := <-fileChmodEvent:
				log.Print("chmod file" + chmodFileName)
			}
		}

		defer wg.Done()
	}()

	wg.Wait()
}