eta_data_analysis/utils/config.go

package utils

import (
	"fmt"
	"strconv"
	"time"

	beeLogger "github.com/beego/bee/v2/logger"
	"github.com/beego/beego/v2/server/web"
	"github.com/patrickmn/go-cache"
)

var (
	RunMode string //运行模式
)

// 公共api内部服务调用
var (
	// EDB_LIB_URL 公共指标库
	EDB_LIB_URL         string
	APP_EDB_LIB_NAME_EN string
	EDB_LIB_Md5_KEY     string
	OpUserId            int    // 操作人id
	OpUserRealName      string // 操作人真实名称
)

// 弘则
const (
	APPID  = "SJCrpOPvagscPxEv"
	SECRET = "gLLjT72uFHQZEFtaFCuoZegD1z2ezfyX"
)

// 日志配置
var (
	LogPath    string //调用过程中的日志存放地址
	LogFile    string
	BinLogPath string //数据库相关的日志存放地址
	BinLogFile string
	LogMaxDays int //日志最大保留天数
)

// 涌益生猪
var (
	YongyiFilePath     string //excel文件地址
	YongyiReadFilePath string //已读的excel文件地址
	YongyiOpen         string //是否配置涌益生猪数据源，1已配置
	YongyiDownloadHost string
)

// 中国煤炭网
var (
	CoalMineFilePath                string //excel文件地址
	CoalMineOpen                    string //是否配置中国煤炭网数据源，1已配置
	CoalMailAttachmentScriptPath    string // 获取邮件附件的脚本目录
	CoalMailAttachmentOpen          string // 获取邮件附件功能，1已配置
	CoalMailAttachmentTime          string // 获取邮件附件功能时间
	CoalMailAttachmentPythonVersion string // 获取邮件附件功能python版本
)

// 汾渭煤炭
var (
	FenweiFileDir    string // excel文件目录
	FenweiOldFileDir string // 已读取过的excel文件目录
	FenweiOpen       string // 是否配置汾渭数据源

	FenweiNetOpen     string // 是否配置汾渭网页数据源
	FenweiNetUseName  string // 汾渭登录账号
	FenweiNetPassword string // 汾渭登录密码
	FenweiNetJsonPath string // 汾渭json文件地址
)

// 煤炭江湖
var (
	MtjhFilePath string //excel文件地址
	MtjhOpen     string //是否配置煤炭江湖数据源，1已配置
)

// 粮油商务网
var (
	LY_USERNAME  string
	LY_PASSWORD  string
	LY_JSON_PATH string
	LY_OPEN      string
)

// 睿姿得数据
var (
	RZD_TASK_TIME     string
	RZD_USERNAME      string
	RZD_PASSWORD      string
	RZD_EXCEL_PATH    string
	RZD_OPEN          string
	RZD_DOWNLOAD_PATH string
	RZD_LOGIN_PATH    string
)

// CCF化纤信息
var (
	CCFOpen                  string // 是否配置CCF
	CCFCookieFile            string // CCF登录Cookie
	CCFDataRuleFile          string // CCF数据爬取规则
	CCFDailyTaskTime         string // CCF数据日度任务时间
	CCFWeeklyTaskTime        string // CCF数据周度任务时间
	CCFStockTaskTime         string // CCF数据装置任务时间
	CCFUseName               string // CCF登录账号
	CCFPassword              string // CCF登录密码
	CCFDailyFetchNum         int    // CCF数据日度每次获取报告数量
	CCFWeeklyFetchNum        int    // CCF数据周度每次获取报告数量
	CCFStockFetchNum         int    // CCF数据装置每次获取报告数量
	CCFChartRuleFile         string // CCF图表爬取规则
	CCFChartAdditionRuleFile string // CCF图表爬取附加规则
	CCFChartTaskTime         string
)

var (
	OilchemAccount    string
	OilchemPassword   string
	OilchemCookieFile string
	OilchemOpen       string
	OilchemDataInit   string
)

var (
	HisugarAccount  string
	HisugarPassword string
	HisugarOpen     string
)

var TerminalCode string

// 克拉克森
var (
	ClarkSonsFilePath string //excel文件地址
	ClarkSonsOpen     string //是否配置克拉克森数据源，1已配置
)

// minio配置
var (
	MinIoAccessKeyId     string
	MinIoAccessKeySecret string
	MinIoEndpoint        string
	MinIoUseSSL          string
	MinIoBucketName      string
	MinIoUploadDir       string
	MinIoViewHost        string
	ChromePath           string
	CommandPython        string
	OssType              string
)

// S3配置
var (
	S3Endpoint        string
	S3BackEndpoint    string
	S3BucketName      string
	S3UploadDir       string
	S3AccessKeyId     string
	S3AccessKeySecret string
	S3Host            string
	S3Region          string
	S3ForceStyle      string
	S3EndpointPort    string
	S3Protocol        string
	S3DisableSSL      string
	S3OpenAcl         string
)

// 阿里云配置
var (
	Bucketname       string
	Endpoint         string
	Imghost          string
	UploadDir        string
	Upload_Audio_Dir string
	AccessKeyId      string
	AccessKeySecret  string
)

// Kpler
var (
	KplerExcelFilePath string //excel文件地址
	KplerExcelOpen     string //是否配置Kpler数据源，1已配置
	KplerRefreshUrl    string //Kpler刷新url
	KplerApiAccount    string //Kpler API调用方式的账号
	KplerApiPassword   string //Kpler API调用方式的密码
)

var (
	CacheClient *cache.Cache
)

func init() {
	tmpRunMode, err := web.AppConfig.String("run_mode")
	if err != nil {
		panic("配置文件读取run_mode错误 " + err.Error())
	}
	RunMode = tmpRunMode
	fmt.Println("RunMode:", RunMode)
	config, err := web.AppConfig.GetSection(RunMode)
	if err != nil {
		panic("配置文件读取错误 " + err.Error())
	}
	beeLogger.Log.Info(RunMode + " 模式")

	// 公共api内部服务调用
	{
		// 公共指标库相关
		EDB_LIB_URL = config["edb_lib_url"]
		APP_EDB_LIB_NAME_EN = config["app_edb_lib_name_en"]
		EDB_LIB_Md5_KEY = config["edb_lib_md5_key"]
		opUserIdStr := config["op_user_id"]
		if opUserIdStr != `` {
			OpUserId, err = strconv.Atoi(opUserIdStr)
			if err != nil {
				panic("配置文件中操作人配置错误 " + err.Error())
			}
		}
		OpUserRealName = config["op_user_real_name"]
	}
	//日志配置
	{
		LogPath = config["log_path"]
		LogFile = config["log_file"]
		BinLogPath = config["binlog_path"]
		BinLogFile = config["binlog_file"]
		logMaxDaysStr := config["log_max_day"]
		LogMaxDays, _ = strconv.Atoi(logMaxDaysStr)
	}
	//涌益咨询文件夹配置
	{
		YongyiFilePath = config["yongyi_file_path"]
		YongyiReadFilePath = config["yongyi_read_file_path"]
		YongyiOpen = config["yongyi_open"]
		YongyiDownloadHost = config["yongyi_download_host"]
		if YongyiDownloadHost == "" {
			YongyiDownloadHost = "http://127.0.0.1:7010/"
		}
	}

	//中国煤炭网文件夹配置
	{
		CoalMineFilePath = config["coal_mine_file_path"]
		CoalMineOpen = config["coal_mine_open"]
		CoalMailAttachmentScriptPath = config["coal_mail_attachment_script_path"]
		CoalMailAttachmentOpen = config["coal_mail_attachment_open"]
		CoalMailAttachmentTime = config["coal_mail_attachment_time"]
		CoalMailAttachmentPythonVersion = config["coal_mail_attachment_python_version"]
	}

	// 汾渭配置
	{
		FenweiOpen = config["fenwei_open"]
		FenweiFileDir = config["fenwei_file_dir"]
		FenweiOldFileDir = config["fenwei_old_file_dir"]

		FenweiNetOpen = config["fenwei_net_open"]
		FenweiNetUseName = config["fenwei_net_username"]
		FenweiNetPassword = config["fenwei_net_password"]
		FenweiNetJsonPath = config["fenwei_net_json_path"]
	}

	//煤炭江湖文件夹配置
	{
		MtjhFilePath = config["mtjh_file_path"]
		MtjhOpen = config["mtjh_open"]
	}

	TerminalCode = config["terminal_code"]

	// CCF化纤信息
	{
		CCFOpen = config["ccf_open"]
		CCFCookieFile = config["ccf_cookie_file"]
		CCFDataRuleFile = config["ccf_data_rule_file"]
		CCFDailyTaskTime = config["ccf_daily_task_time"]
		CCFWeeklyTaskTime = config["ccf_weekly_task_time"]
		CCFStockTaskTime = config["ccf_stock_task_time"]
		CCFUseName = config["ccf_username"]
		CCFPassword = config["ccf_password"]
		CCFDailyFetchNum, _ = strconv.Atoi(config["ccf_daily_fetch_num"])
		if CCFDailyFetchNum <= 0 {
			CCFDailyFetchNum = 7
		}
		CCFWeeklyFetchNum, _ = strconv.Atoi(config["ccf_weekly_fetch_num"])
		if CCFWeeklyFetchNum <= 0 {
			CCFWeeklyFetchNum = 3
		}
		CCFStockFetchNum, _ = strconv.Atoi(config["ccf_stock_fetch_num"])
		if CCFStockFetchNum <= 0 {
			CCFStockFetchNum = 3
		}
		CCFChartRuleFile = config["ccf_chart_rule_file"]
		if CCFChartRuleFile == "" {
			CCFChartRuleFile = "static/ccf_chart_rule.json"
		}
		if CCFChartAdditionRuleFile == "" {
			CCFChartAdditionRuleFile = "static/ccf_chart_addition_rule.json"
		}
		CCFChartTaskTime = config["ccf_chart_task_time"]
	}

	{
		LY_USERNAME = config["ly_username"]
		LY_PASSWORD = config["ly_password"]
		LY_JSON_PATH = config["ly_json_path"]
		LY_OPEN = config["ly_open"]
	}

	{
		RZD_TASK_TIME = config["rzd_task_time"]
		RZD_USERNAME = config["rzd_username"]
		RZD_PASSWORD = config["rzd_password"]
		RZD_EXCEL_PATH = config["rzd_excel_path"]
		RZD_OPEN = config["rzd_open"]
		RZD_DOWNLOAD_PATH = config["rzd_download_path"]
		RZD_LOGIN_PATH = config["rzd_login_path"]
	}

	// 隆众数据
	{
		OilchemAccount = config["oilchem_account"]
		OilchemPassword = config["oilchem_password"]
		OilchemCookieFile = config["oilchem_cookie_file"]
		OilchemOpen = config["oilchem_open"]
		OilchemDataInit = config["oilchem_data_init"]
	}

	// 泛糖科技
	{
		HisugarAccount = config["hisugar_account"]
		HisugarPassword = config["hisugar_password"]
		HisugarOpen = config["hisugar_open"]
	}

	//克拉克森文件夹配置
	{
		ClarkSonsFilePath = config["clarksons_file_path"]
		ClarkSonsOpen = config["clarksons_open"]
	}

	//PDF生成配置
	{
		MinIoAccessKeyId = config["minio_access_key_id"]
		MinIoAccessKeySecret = config["minio_access_key_secret"]
		MinIoEndpoint = config["minio_endpoint"]
		MinIoUseSSL = config["minio_use_ssl"]
		MinIoBucketName = config["minio_bucket_name"]
		MinIoUploadDir = config["minio_upload_dir"]
		MinIoViewHost = config["minio_view_host"]
		ChromePath = config["chrome_path"]
		CommandPython = config["command_python"]
		if CommandPython == "" {
			CommandPython = "python3"
		}
		OssType = config["oss_type"]
		if OssType == "" {
			OssType = MINIO
		}
		// S3-OSS相关
		S3Endpoint = config["s3_endpoint"]
		S3BackEndpoint = config["s3_back_endpoint"]
		S3BucketName = config["s3_bucket_name"]
		S3Host = config["s3_host"]
		S3AccessKeyId = config["s3_access_key_id"]
		S3AccessKeySecret = config["s3_access_key_secret"]
		S3UploadDir = config["s3_upload_dir"]
		S3Region = config["s3_region"]
		S3ForceStyle = config["s3_force_style"]
		S3EndpointPort = config["s3_endpoint_port"]
		S3Protocol = config["s3_protocol"]
		S3DisableSSL = config["s3_disable_ssl"]
		S3OpenAcl = config["s3_open_acl"]

		//aliyun oss
		Endpoint = config["endpoint"]
		Bucketname = config["bucket_name"]
		Imghost = config["img_host"]
		UploadDir = config["upload_dir"]
		Upload_Audio_Dir = config["upload_audio_dir"]
		AccessKeyId = config["access_key_id"]
		AccessKeySecret = config["access_key_secret"]
	}
	KplerExcelOpen = config["kpler_excel_open"]
	KplerExcelFilePath = config["kpler_excel_file_path"]
	KplerRefreshUrl = config["kpler_refresh_url"]
	KplerApiAccount = config["kpler_api_account"]
	KplerApiPassword = config["kpler_api_password"]
	CacheClient = cache.New(365*24*time.Hour, 365*24*time.Hour)
}

//修改接口文档
//http://8.136.199.33:8300/swagger/