spider_creater/backend/config.go

package backend

import (
	"os"

	"gopkg.in/yaml.v3"
)

// 配置
type Config struct {
	IsOnly4MainSite     bool `yaml:"isOnly4MainSite"` //仅仅使用重点网站数据
	DisableLoadResource string
}

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
}