Golang实现一个批量自动化执行树莓派指令的软件(6)简易批量指令处理

news/2024/10/19 7:29:01/

简介

基于上篇 Golang实现一个批量自动化执行树莓派指令的软件(5)模块整合，这里我们实现简单的从配置文件设置指令集，然后程序自动运行指令集的操作。

环境描述

运行环境: Windows，基于Golang，暂时没有使用什么不可跨平台接口，理论上支持Linux/MacOS
目标终端：树莓派DebianOS(主要做用它测试)

实现

package controllerimport ("encoding/json""fmt""os""ssh_remote_access/app""ssh_remote_access/models""ssh_remote_access/sshutil""ssh_remote_access/utils""sync""time"
)type JobController struct {settings models.Settingssshutil  *sshutil.SSHUtil
}func NewJobController(decryptionKey string, cfgPath string) (*JobController, error) {var (controller   = &JobController{}err          errorencrypt      *utils.EncryptionrealPassword string)/* 加载文件 */if err = controller.loadConfig(cfgPath); nil != err {return nil, err}/* 组合配置信息 */encrypt = utils.NewEncryption(decryptionKey)if realPassword, err = encrypt.Decrypt(controller.settings.User.Password); nil != err {return nil, err}controller.sshutil, err = sshutil.NewSSHUtil(sshutil.SSHConfig{IP:       controller.settings.Network.IP,Port:     controller.settings.Network.Port,Timeout:  controller.settings.Network.Timeout,User:     controller.settings.User.Name,Password: realPassword,})return controller, err
}func RunJobList() error {var (controller *JobControllererr        error)/* 初始化任务列表 */controller, err = NewJobController(app.ASE_KEY, app.CONFIG_PATH)if nil != err {fmt.Println("run job list fail, ", err.Error())return err}defer controller.Destroy()/* 执行任务 */err = controller.RunJobs(func(currFinishedJob models.SSHJob, jobNum, fnishedNum uint) error {fmt.Printf("job_type: %d, total job: %d, finished job: %d.\n", currFinishedJob.Type, jobNum, fnishedNum)return nil})return err
}func (j *JobController) Destroy() error {return j.sshutil.Destroy()
}func (j *JobController) loadConfig(filepath string) error {var bs, err = os.ReadFile(filepath)if nil != err {return err}err = json.Unmarshal(bs, &j.settings)return err
}func (j *JobController) RunJobs(processCallback func(currFinishedJob models.SSHJob, jobNum, fnishedNum uint) error) error {var (waiter sync.WaitGrouperr    error)for i, job := range j.settings.JobList {waiter.Add(1)time.Sleep(time.Duration(job.BeforeSleep) * time.Millisecond)switch models.SSH_TYPE(job.Type) {case models.SSH_COMMAND:_, err = j.sshutil.Commander().Command(job.Command)if nil != err {return err}case models.SSH_UPLOAD:err = j.sshutil.SFTFClient().UploadWithCallback(job.From, job.To, func(from, to string, num, uploaded uint) {fmt.Println("upload from ", from, " to ", to, ", total_num ", num, " upload_num ", uploaded)}, func(err error) {fmt.Println("upload finished")}, false)if nil != err {return err}case models.SSH_DOWNLOAD:err = j.sshutil.SFTFClient().DownloadWithCallback(job.From, job.To, func(from, to string, num, downloaded uint) {fmt.Println("download from ", from, " to ", to, ", total_num ", num, " download_num ", downloaded)}, func(err error) {fmt.Println("download finished")}, false)if nil != err {return err}}time.Sleep(time.Duration(job.AfterSleep) * time.Millisecond)if err = processCallback(job, uint(len(j.settings.JobList)), uint(i+1)); nil != err {}}return err
}