spider_creater/backend/config.go

package backend

import (
	"os"

	"gopkg.in/yaml.v3"
)

// 配置
type Config struct {
	DisableLoadResource             string `yaml:"disableLoadResource"`
	IsOnly4MainSite                 bool   `yaml:"isOnly4MainSite"`            //仅仅使用重点网站数据
	BrowserLoadResourceTimeout      int64  `yaml:"browserLoadResourceTimeout"` //浏览器加载资源超时设定、
	ServerCodeTimeOut               int64  `yaml:"timeout"`
	ServerCodeAddress               string `yaml:"address"`
	ServerCodeFreeAddressOcr        string `yaml:"freeaddressocr"`
	ServerCodeFreeAddressArithmetic string `yaml:"freeaddressarithmetic"`
	Username                        string `yaml:"username"`
	Password                        string `yaml:"password"`
}

var (
	Cfg *Config = new(Config)
)

// 加载配置文件
func LoadConfig(cf string) error {
	fi, err := os.Open(cf)
	if err != nil {
		return err
	}
	defer fi.Close()

	err = yaml.NewDecoder(fi).Decode(Cfg)
	if err != nil {
		return err
	}
	return nil
}