Go語言只知其一;不知其二上手記（三）

1、前言

  在前兩篇文章《Go語言只知其一;不知其二上手記（一）》、《Go語言只知其一;不知其二上手記（二）》中，咱們實現了從讀取 xlsx 格式的數據字典開始，到生成 schema 文件，到最後生成代碼文件的完整過程。

  在文章二中，咱們在 template.go 中添加了解析 schema 的方法 analyzeSchema ，在這個方法裏，又調用了 generateModelFile、generateControllerFile 來生成最後的代碼文件。這兩個方法對數據字典格式的針對性及強，參考性不大，所以文章中未進行描述。在後面的實現過程當中，咱們又增長了 router 中使用的代碼腳本的生成功能，最終，在 analyzeSchema 中便有了三個代碼文件生成函數的調用，他們目前是順序執行的，而Go語言提供了很方便的併發機制，咱們能夠再將這三個函數改形成併發執行的過程當中初步瞭解一下Go語言的併發機制。

2、Go語言併發機制初探

一、併發前的 analyzeSchema 函數

// 分析模式文件
func analyzeSchema(schemaPath string, modelTmplPath string, controllerTmplPath string, consoleRouterTmplPath string) error {
	// 1.加載 模式文件
	schema, err := ioutil.ReadFile(schemaPath)
	if err != nil {
		return err
	}
	// 2.轉換 模式文件到 數據字典集合 中
	var dataDictSlice []DataDict // 數據字典集合
	if err := Json.Unmarshal(schema, &dataDictSlice); err != nil {
		return err
	}

	// 3.加載模板文件
	modelTmpl, err := template.ParseFiles(modelTmplPath)
	if err != nil {
		return err
	}
	controllerTmpl, err := template.ParseFiles(controllerTmplPath)
	if err != nil {
		return err
	}
	consoleRouterTmpl, err := template.ParseFiles(consoleRouterTmplPath)
	if err != nil {
		return err
	}

	// 4.遍歷數據字典集合
	wg := sync.WaitGroup{}
	for _, dataDict := range dataDictSlice {

		// 4.1.生成 collectionInfo 信息
		collectionInfo := map[string]string{
			"Model":          strings.ToUpper(string(dataDict.Collection.Name[0])) + dataDict.Collection.Name[1:],
			"CollectionName": dataDict.Collection.Name,
			"CollectionDesc": dataDict.Collection.Desc,
		}

		// 4.2.生成 所需 腳本
		modelScript, joiSchemaScript, err := getFieldsScript(dataDict)
		if err != nil {
			return nil
		}

		// 4.3.生成 model 代碼文件
		err = generateModelFile(modelTmpl, collectionInfo, modelScript)
		if err != nil {
			println(err.Error())
		}

		// 4.4.生成 consoleRouter 代碼片斷
		err = generateCodeFile(consoleRouterTmpl, "./dist/consoleRouters/"+collectionInfo["CollectionName"]+".js", collectionInfo)
		if err != nil {
			println(err.Error())
		}

		// 4.5.生成 controller 代碼文件
		collectionInfo["joiSchemaScript"] = joiSchemaScript
		err = generateCodeFile(controllerTmpl, "./dist/controllers/"+dataDictInfo["collectionInfo"]+".js", collectionInfo)
		if err != nil {
			println(err.Error())
		}

	}

	// 5.沒有出錯，返回 nil
	return nil
}
複製代碼

與上一篇文章中的 analyzeSchema 函數不一樣之處以下：

1. 「加載模板文件」 時 由一個變成了三個
2. 「遍歷數據字典集合」 時 將代碼生成函數中所須要用到的信息 提早獲取出來，分別是 collectionInfo、modelScript、joiSchemaScript
3. 生成 controller 文件 及 consoleRouter 文件 的函數合併成了一個函數

二、併發後的 analyzeSchema 函數

// 分析模式文件
func analyzeSchema(schemaPath string, modelTmplPath string, controllerTmplPath string, consoleRouterTmplPath string) error {
	// 1.加載 模式文件
	schema, err := ioutil.ReadFile(schemaPath)
	if err != nil {
		return err
	}
	// 2.轉換 模式文件到 數據字典集合 中
	var dataDictSlice []DataDict // 數據字典集合
	if err := Json.Unmarshal(schema, &dataDictSlice); err != nil {
		return err
	}

	// 3.加載模板文件
	modelTmpl, err := template.ParseFiles(modelTmplPath)
	if err != nil {
		return err
	}
	controllerTmpl, err := template.ParseFiles(controllerTmplPath)
	if err != nil {
		return err
	}
	consoleRouterTmpl, err := template.ParseFiles(consoleRouterTmplPath)
	if err != nil {
		return err
	}

	// 4.遍歷數據字典集合，採用 協程 併發處理
	wg := sync.WaitGroup{}
	for _, dataDict := range dataDictSlice {

		// 4.1.生成 collectionInfo 信息
		collectionInfo := map[string]string{
			"Model":          strings.ToUpper(string(dataDict.Collection.Name[0])) + dataDict.Collection.Name[1:],
			"CollectionName": dataDict.Collection.Name,
			"CollectionDesc": dataDict.Collection.Desc,
		}

		// 4.2.生成 fields 腳本
		modelScript, joiSchemaScript, err := getFieldsScript(dataDict)
		if err != nil {
			return nil
		}

		// 4.3.生成 model 代碼文件
		wg.Add(1)

		go func() {
			defer wg.Done()
			err := generateModelFile(modelTmpl, collectionInfo, modelScript)
			if err != nil {
				println(err.Error())
			}
		}()

		// 4.4.生成 consoleRouter 代碼片斷
		wg.Add(1)
		go func() {
			defer wg.Done()
			err := generateCodeFile(consoleRouterTmpl, "./dist/consoleRouters/"+collectionInfo["CollectionName"]+".js", collectionInfo)
			if err != nil {
				println(err.Error())
			}
		}()

		// 4.5.生成 controller 代碼文件
		collectionInfo["joiSchemaScript"] = joiSchemaScript
		wg.Add(1)
		go func() {
			defer wg.Done()
			err := generateCodeFile(controllerTmpl, "./dist/controllers/"+collectionInfo["CollectionName"]+".js", collectionInfo)
			if err != nil {
				println(err.Error())
			}
		}()

	}
	wg.Wait()

	// 5.沒有出錯，返回 nil
	return nil
}
複製代碼

具體改造以下：

1. 使用 go func() { } () 將原來的執行方法包起來，實現 goroutine 併發任務
2. 使用 sync.WaitGroup 來阻塞等待全部任務的完成，wg.Add(1)、wg.Done()、wg.Wait()

3、總結

  經過上面的小改造，咱們把順序執行的三個函數更改成了併發執行的任務，再經過 sync.WaitGroup 阻塞等待全部任務的完成。咱們經過兩種不一樣實現方法上的程序運行時間，也驗證了併發確實是節省了時間的。這個例子中對於併發的應用是比較簡單的，更深刻的使用，就待往後再實戰中繼續探索吧。