一个备份任务分发的Golang实现原型

       前段在玩协程时跟着libtask的作者Russ Cox转到Golang了,一堆重量级大牛们想出来的东西就是不同,思路非常特别,和常用的差别有点大,

但非常注重实用,减少了一堆的语法糖后,让人重新思考,开发语言本来应当是怎么样?如果用来做服务端开发的话,真的是很不错的东西。

       在一个练习中,我设计了个发送Oracle数据库备份任务的简单场景。通过 发送备份任务到各个客户端,来练习Golang中的sync.WaitGroup.

       sync.WaitGroup是Golang提供的一种简单的同步方法集合。它有三个方法.
             Add() 添加计数,数目可以为一个,也可以为多个。
             Done() 减掉一个计数,如果计数不为0,则Wait()会阻塞在那,直到全部为0
             Wait() 等待计数为0.

 

         弄个例子直观的见识下:

package main

import (
	"fmt"
	"sync"
)

var waitGrp sync.WaitGroup

func main() {

	bakJobs := []string{"Full Database", "Archive Log", "Control File"}

	for i := 0; i < len(bakJobs); i++ {
		waitGrp.Add(1) //add job

		go func(jobName string) {
			fmt.Println(jobName)
			waitGrp.Done() //sub job
		}(bakJobs[i])
	}

	waitGrp.Wait()
}

/*
E:\GitHub\Golang\XCLTools\src\test>wg
Full Database
Archive Log
Control File
*/
可以看到会依次打印bakJobs中的数据。但如果把waitGrp.Wait()注释掉。啥都不会输出。即,在for后,wait会等待所有的job执行完后,

才会执行后面的。  这个例子说明,除了熟悉了下waitGroup的使用外,还说明waitGroup能保证作业的执行顺序,及会等待所有子作业全部完成再继续的特性。

       就Oracle数据库备份本身而言,在这使用go func()...其实意义不大,所以我想了个更复杂点的场景,给多个服务器上的数据库批量发送备份指令,并依数据库

的重要性顺序来执行。

      下面是实现的代码(纯演示,忽略掉其它通信和逻辑处理方面的东西):       

package main

import (
	"fmt"
	"sync"
)

var waitGrp sync.WaitGroup

func main() {
	waitGrp.Add(2)
	for _, ip := range []string{"ip1", "ip2"} {
		go doJobs(ip)
		fmt.Println("IP = ", ip)
	}
	waitGrp.Wait()
	fmt.Println("main() end.")
}

func doJobs(ip string) {

	bakJobs := []string{"Full Database", "Archive Log", "Control File"}
	for i := 0; i < len(bakJobs); i++ {
		doBak(ip, i, bakJobs[i])
	}
	fmt.Println("The backup is complete!")
	fmt.Println("......")
	defer waitGrp.Done()
}

func doBak(ip string, id int, jobName string) (bool, error) {
	fmt.Println("doBak() :", ip, " - ", id, " - ", jobName)
	return true, nil
}

/*
IP =  ip1
IP =  ip2
doBak() : ip1  -  0  -  Full Database
doBak() : ip1  -  1  -  Archive Log
doBak() : ip1  -  2  -  Control File
The backup is complete!
......
doBak() : ip2  -  0  -  Full Database
doBak() : ip2  -  1  -  Archive Log
doBak() : ip2  -  2  -  Control File
The backup is complete!
......
main() end.

*/

    代码中,会首先向各个IP发送执行jobs的指令。各个指令执行的时间有长有短,用go func()...能保证是在同一时间向所有相关IP发出命令让各IP依次执行相关备份。

主goroutine则等待这个总作业全部结束后再执行其它。同时由于waitGroup的顺序性,能保证重要点的ip,可以优先执行。

    我设计场景的目的是想在例子中加入更多的Golang新鲜元素来增强感觉,所以针对上一个例子,我又设想了一个情况,因为数据库备份是很耗时的,且要等待

所有IP上的作业都执行完毕要花费很长时间,如果这时,用户要求取消这次作业怎么办?

    于是这就有了下面的例子:

package main

import (
	"bufio"
	"fmt"
	"os"
	"sync"
	"time"
)

var waitGrp sync.WaitGroup

func main() {

	ch := make(chan bool)

	go schedule(ch)

	r := bufio.NewReader(os.Stdin)
	for {
		time.Sleep(time.Second)

		fmt.Print("Command:> ")
		ln, _, _ := r.ReadLine()
		cmd := string(ln)

		if "q" == cmd || "quit" == cmd {
			close(ch)
			break
		} else {
			fmt.Println(" = cmd = ", cmd, "\n")
		}
	}

	waitGrp.Wait()
	fmt.Println("main() end.")
}

func schedule(ch chan bool) {

	for _, ip := range []string{"ip1", "ip2"} {
		waitGrp.Add(1)

		go doJobs(ip, ch)
		fmt.Println("schedule() IP = ", ip)
	}
	fmt.Println("schedule() end.")
	return
}

func doJobs(ip string, ch chan bool) {

	defer waitGrp.Done()

	for i := 0; i < 10; i++ {

		select {
		case <-ch:
			fmt.Println("doJobs() ", ip, "=>Job Cancel......")
			return
		default:
		}

		fmt.Println("doJobs()...... ", ip, " for:", i)
		time.Sleep(time.Second)
	}
}

/*

E:\GitHub\Golang\XCLTools\src\test>wg5
schedule() IP =  ip1
schedule() IP =  ip2
schedule() end.
doJobs()......  ip1  for: 0
doJobs()......  ip2  for: 0
Command:> doJobs()......  ip1  for: 1
doJobs()......  ip2  for: 1
doJobs()......  ip1  for: 2
doJobs()......  ip2  for: 2
doJobs()......  ip1  for: 3
doJobs()......  ip2  for: 3
doJobs()......  ip1  for: 4
doJobs()......  ip2  for: 4
doJobs()......  ip1  for: 5
doJobs()......  ip2  for: 5
doJobs()......  ip1  for: 6
doJobs()......  ip2  for: 6
doJobs()......  ip1  for: 7
doJobs()......  ip2  for: 7
doJobs()......  ip1  for: 8
doJobs()......  ip2  for: 8
doJobs()......  ip1  for: 9
doJobs()......  ip2  for: 9
q
main() end.

E:\GitHub\Golang\XCLTools\src\test>wg5
schedule() IP =  ip1
schedule() IP =  ip2
schedule() end.
doJobs()......  ip1  for: 0
doJobs()......  ip2  for: 0
Command:> doJobs()......  ip2  for: 1
doJobs()......  ip1  for: 1
doJobs()......  ip2  for: 2
doJobs()......  ip1  for: 2
doJobs()......  ip2  for: 3
doJobs()......  ip1  for: 3
q
doJobs()  ip2 =>Job Cancel......
doJobs()  ip1 =>Job Cancel......
main() end.
*/
        我用了个for{},让用户在作业执行过程中,可以输入指令退出执行。并且,在退出过程中,各个IP也会作相关的取消处理。以保证不会因

强制中断而出现一些不必要的麻烦。

   这个例子到这,已经引入了足够多的东西,我觉得已达到练习的目的,场景就先到此了。

   在此整理下,以备后查。


   MAIL:  [email protected]

  BLOG: http://blog.csdn.net/xcl168


你可能感兴趣的:(golang,go语言,goroutine,任务分发,sync.WaitGroup)