mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-08-03 14:27:32 +08:00
129 lines
3.4 KiB
Go
129 lines
3.4 KiB
Go
//
|
|
// Copyright 2026 The InfiniFlow Authors. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
|
|
package config
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// DatabaseConfig database configuration
|
|
type DatabaseConfig struct {
|
|
MySQL MySQLConfig `mapstructure:"mysql"`
|
|
}
|
|
|
|
type MySQLConfig struct {
|
|
DatabaseName string `mapstructure:"name"` // database name
|
|
User string `mapstructure:"user"`
|
|
Password string `mapstructure:"password"`
|
|
Host string `mapstructure:"host"`
|
|
Port int `mapstructure:"port"`
|
|
MaxConnections int `mapstructure:"max_connections"`
|
|
StaleTimeout int `mapstructure:"stale_timeout"`
|
|
MaxAllowedPacket int `mapstructure:"max_allowed_packet"`
|
|
Charset string `mapstructure:"charset"`
|
|
}
|
|
|
|
func (c *Config) ParseDatabaseConfig(v *viper.Viper) error {
|
|
databaseType := c.general.Database
|
|
switch databaseType {
|
|
case "mysql":
|
|
c.parseMySQLConfig(v)
|
|
default:
|
|
return fmt.Errorf("database type %s is not supported", databaseType)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *Config) parseMySQLConfig(v *viper.Viper) {
|
|
|
|
// Default MySQL config
|
|
c.database.MySQL.DatabaseName = "rag_flow"
|
|
c.database.MySQL.User = "root"
|
|
c.database.MySQL.Password = "infini_rag_flow"
|
|
c.database.MySQL.Host = "localhost"
|
|
c.database.MySQL.Port = 3306
|
|
c.database.MySQL.MaxConnections = 900
|
|
c.database.MySQL.StaleTimeout = 300
|
|
c.database.MySQL.MaxAllowedPacket = 1073741824
|
|
c.database.MySQL.Charset = "utf8mb4"
|
|
|
|
if !v.IsSet("mysql") {
|
|
return
|
|
}
|
|
sub := v.Sub("mysql")
|
|
if sub == nil {
|
|
return
|
|
}
|
|
|
|
if sub.IsSet("name") {
|
|
c.database.MySQL.DatabaseName = sub.GetString("name")
|
|
}
|
|
|
|
if sub.IsSet("user") {
|
|
c.database.MySQL.User = sub.GetString("user")
|
|
}
|
|
|
|
if sub.IsSet("password") {
|
|
c.database.MySQL.Password = sub.GetString("password")
|
|
}
|
|
|
|
if sub.IsSet("host") {
|
|
c.database.MySQL.Host = sub.GetString("host")
|
|
}
|
|
|
|
if sub.IsSet("port") {
|
|
c.database.MySQL.Port = sub.GetInt("port")
|
|
}
|
|
|
|
if sub.IsSet("max_connections") {
|
|
c.database.MySQL.MaxConnections = sub.GetInt("max_connections")
|
|
}
|
|
|
|
if sub.IsSet("stale_timeout") {
|
|
c.database.MySQL.StaleTimeout = sub.GetInt("stale_timeout")
|
|
}
|
|
|
|
if sub.IsSet("max_allowed_packet") {
|
|
c.database.MySQL.MaxAllowedPacket = sub.GetInt("max_allowed_packet")
|
|
}
|
|
|
|
if sub.IsSet("charset") {
|
|
c.database.MySQL.Charset = sub.GetString("charset")
|
|
}
|
|
}
|
|
|
|
func (c *Config) GetMySQLConfig() MySQLConfig {
|
|
return c.database.MySQL
|
|
}
|
|
|
|
func (m *MySQLConfig) ExportConfigs() map[string]interface{} {
|
|
var mysqlConfigs map[string]interface{}
|
|
mysqlConfigs = make(map[string]interface{})
|
|
mysqlConfigs["host"] = m.Host
|
|
mysqlConfigs["port"] = m.Port
|
|
mysqlConfigs["database_name"] = m.DatabaseName
|
|
mysqlConfigs["username"] = m.User
|
|
mysqlConfigs["password"] = m.Password
|
|
mysqlConfigs["charset"] = m.Charset
|
|
mysqlConfigs["max_connections"] = m.MaxConnections
|
|
mysqlConfigs["stale_timeout"] = m.StaleTimeout
|
|
mysqlConfigs["max_allowed_packet"] = m.MaxAllowedPacket
|
|
return mysqlConfigs
|
|
}
|