Esempio n. 1
0
// 编译模板,并输出 p 的内容。
func handleGroup(w http.ResponseWriter, r *http.Request, tplDir string, p *htmlPage) {
	t, err := compileHTMLTemplate(tplDir)
	if err != nil {
		app.Errorln(err)
		w.WriteHeader(http.StatusInternalServerError)
		return
	}

	tplName := "group"
	if p.GroupName == app.DefaultGroupName {
		tplName = "index"
	}

	if err = t.ExecuteTemplate(w, tplName, p); err != nil {
		app.Errorln(err)
		w.WriteHeader(http.StatusInternalServerError)
		return
	}
}
Esempio n. 2
0
// 真正的程序入口,main 主要是作参数的处理。
func run() {
	start := time.Now()

	path, err := getConfigFile()
	if err != nil {
		app.Errorln(err)
		return
	}

	cfg, err := loadConfig(path)
	if err != nil {
		app.Errorln(err)
		return
	}

	// 比较版本号兼容问题
	compatible, err := version.SemVerCompatible(app.Version, cfg.Version)
	if err != nil {
		app.Errorln(err)
		return
	}
	if !compatible {
		app.Errorln("当前程序与配置文件中指定的版本号不兼容")
		return
	}

	// 分析文档内容
	docs := doc.New()
	wg := &sync.WaitGroup{}
	for _, opt := range cfg.Inputs {
		wg.Add(1)
		go func(o *input.Options) {
			if err := input.Parse(docs, o); err != nil {
				app.Errorln(err)
			}
			wg.Done()
		}(opt)
	}
	wg.Wait()

	if len(docs.Title) == 0 {
		docs.Title = app.DefaultTitle
	}

	// 输出内容
	cfg.Output.Elapsed = time.Now().Sub(start)
	if err := output.Render(docs, cfg.Output); err != nil {
		app.Errorln(err)
		return
	}

	app.Infoln("完成!文档保存在", cfg.Output.Dir, "总用时", time.Now().Sub(start))
}
Esempio n. 3
0
func main() {
	h := flag.Bool("h", false, "显示帮助信息")
	v := flag.Bool("v", false, "显示版本信息")
	l := flag.Bool("l", false, "显示所有支持的语言")
	g := flag.Bool("g", false, "在当前目录下创建一个默认的配置文件")
	pprofType := flag.String("pprof", "", "指定一种调试输出类型,可以为 cpu 或是 mem")
	flag.Usage = usage
	flag.Parse()

	switch {
	case *h:
		flag.Usage()
		return
	case *v:
		fmt.Fprintln(os.Stdout, app.Name, app.Version, "build with", runtime.Version())
		return
	case *l:
		fmt.Fprintln(os.Stdout, "目前支持以下语言", input.Langs())
		return
	case *g:
		path, err := getConfigFile()
		if err != nil {
			app.Errorln(err)
			return
		}
		if err = genConfigFile(path); err != nil {
			app.Errorln(err)
			return
		}
		app.Infoln("配置内容成功写入", path)
		return
	}

	// 指定了 pprof 参数
	if len(*pprofType) > 0 {
		profile := filepath.Join("./", app.Profile)
		f, err := os.Create(profile)
		if err != nil {
			app.Errorln(err)
			return
		}
		defer func() {
			if err := f.Close(); err != nil {
				app.Errorln(err)
				return
			}
			app.Infoln("pprof 的相关参数已经写入到", profile)
		}()

		switch strings.ToLower(*pprofType) {
		case "mem":
			defer func() {
				if err = pprof.Lookup("heap").WriteTo(f, 1); err != nil {
					app.Errorln(err)
				}
			}()
		case "cpu":
			if err := pprof.StartCPUProfile(f); err != nil {
				app.Errorln(err)
			}
			defer pprof.StopCPUProfile()
		default:
			app.Errorln("无效的 pprof 参数")
			return
		}
	}

	run()
}