测试通过

config/config.go

@@ -11,23 +11,67 @@ import (
 
 // Config 应用配置
 type Config struct {
-	Database DBConfig    `yaml:"database"`
-	Redis    RedisConfig `yaml:"redis"`
-	Doris    DorisConfig `yaml:"doris"`
-	Auth     Auth        `yaml:"auth"`
-	Service  Service     `yaml:"service"`
-	HTTP     HTTPConfig  `yaml:"http"` // 新增 HTTP 配置
+	Database DBConfig       `yaml:"database"`
+	Redis    RedisConfig    `yaml:"redis"`
+	Doris    DorisConfig    `yaml:"doris"`
+	RabbitMQ RabbitMQConfig `yaml:"rabbitmq"` // 新增 RabbitMQ 配置
+	Auth     Auth           `yaml:"auth"`
+	Service  Service        `yaml:"service"`
+	HTTP     HTTPConfig     `yaml:"http"`
+}
+
+// RabbitMQConfig RabbitMQ配置
+type RabbitMQConfig struct {
+	// 连接配置
+	Host     string `yaml:"host"`
+	Port     int    `yaml:"port"`
+	Username string `yaml:"username"`
+	Password string `yaml:"password"`
+	Vhost    string `yaml:"vhost"`
+
+	// TLS配置
+	UseTLS   bool   `yaml:"use_tls"`
+	CACert   string `yaml:"ca_cert"`
+	CertFile string `yaml:"cert_file"`
+	KeyFile  string `yaml:"key_file"`
+
+	// 连接池和超时配置
+	MaxOpenChannels      int `yaml:"max_open_channels"`      // 最大通道数
+	ReconnectDelay       int `yaml:"reconnect_delay"`        // 重连延迟（毫秒）
+	MaxReconnectAttempts int `yaml:"max_reconnect_attempts"` // 最大重试次数
+
+	// 心跳和超时
+	Heartbeat   int `yaml:"heartbeat"`    // 心跳间隔（秒）
+	ChannelSize int `yaml:"channel_size"` // 通道缓冲区大小
+
+	// 队列和交换机默认配置
+	DefaultExchange string `yaml:"default_exchange"`
+	DefaultQueue    string `yaml:"default_queue"`
+
+	// 消息确认模式
+	AutoAck   bool `yaml:"auto_ack"`
+	Mandatory bool `yaml:"mandatory"`
+	Immediate bool `yaml:"immediate"`
+
+	// QoS配置
+	PrefetchCount int  `yaml:"prefetch_count"` // 预取数量
+	PrefetchSize  int  `yaml:"prefetch_size"`  // 预取大小（字节）
+	Global        bool `yaml:"global"`         // 是否是全局QoS
+
+	// 发布确认
+	PublisherConfirms bool `yaml:"publisher_confirms"` // 是否启用发布者确认
+	ConfirmTimeout    int  `yaml:"confirm_timeout"`    // 确认超时（秒）
 }
 
 // HTTPConfig HTTP客户端配置
 type HTTPConfig struct {
-	Timeout             int  `yaml:"timeout"`                 // 请求超时时间（秒）
-	MaxIdleConns        int  `yaml:"max_idle_conns"`          // 最大空闲连接数
-	MaxIdleConnsPerHost int  `yaml:"max_idle_conns_per_host"` // 每个主机的最大空闲连接数
-	IdleConnTimeout     int  `yaml:"idle_conn_timeout"`       // 空闲连接超时时间（秒）
-	MaxConnsPerHost     int  `yaml:"max_conns_per_host"`      // 每个主机的最大连接数
-	DisableCompression  bool `yaml:"disable_compression"`     // 是否禁用压缩
-	DisableKeepAlives   bool `yaml:"disable_keep_alives"`     // 是否禁用长连接
+	Timeout             int  `yaml:"timeout"`
+	MaxIdleConns        int  `yaml:"max_idle_conns"`
+	MaxIdleConnsPerHost int  `yaml:"max_idle_conns_per_host"`
+	IdleConnTimeout     int  `yaml:"idle_conn_timeout"`
+	MaxConnsPerHost     int  `yaml:"max_conns_per_host"`
+	DisableCompression  bool `yaml:"disable_compression"`
+	DisableKeepAlives   bool `yaml:"disable_keep_alives"`
 }
 
 // DBConfig 数据库配置
@@ -40,45 +84,36 @@ type DBConfig struct {
 	Database        string `yaml:"database"`
 	MaxOpenConns    int    `yaml:"max_open_conns"`
 	MaxIdleConns    int    `yaml:"max_idle_conns"`
-	ConnMaxLifetime int    `yaml:"conn_max_lifetime"` // 单位：秒
+	ConnMaxLifetime int    `yaml:"conn_max_lifetime"`
 }
 
 // RedisConfig Redis配置
 type RedisConfig struct {
-	Host     string `yaml:"host"`
-	Port     int    `yaml:"port"`
-	Password string `yaml:"password"`
-	DB       int    `yaml:"db"`
-	PoolSize int    `yaml:"pool_size"`
-	// 连接超时和读写超时（秒）
-	DialTimeout  int `yaml:"dial_timeout"`
-	ReadTimeout  int `yaml:"read_timeout"`
-	WriteTimeout int `yaml:"write_timeout"`
-	// 连接池配置
-	IdleTimeout int `yaml:"idle_timeout"` // 空闲连接超时（秒）
-	MaxConnAge  int `yaml:"max_conn_age"` // 连接最大存活时间（秒）
+	Host         string `yaml:"host"`
+	Port         int    `yaml:"port"`
+	Password     string `yaml:"password"`
+	DB           int    `yaml:"db"`
+	PoolSize     int    `yaml:"pool_size"`
+	DialTimeout  int    `yaml:"dial_timeout"`
+	ReadTimeout  int    `yaml:"read_timeout"`
+	WriteTimeout int    `yaml:"write_timeout"`
+	IdleTimeout  int    `yaml:"idle_timeout"`
+	MaxConnAge   int    `yaml:"max_conn_age"`
 }
 
 // DorisConfig Doris配置
 type DorisConfig struct {
-	// FE节点配置
-	FEHost     string `yaml:"fe_host"`
-	FEPort     int    `yaml:"fe_port"`
-	FEUsername string `yaml:"fe_username"`
-	FEPassword string `yaml:"fe_password"`
-
-	// MySQL协议配置
-	MySQLHost string `yaml:"mysql_host"`
-	MySQLPort int    `yaml:"mysql_port"`
-
-	// 连接池配置
-	MaxOpenConns    int `yaml:"max_open_conns"`
-	MaxIdleConns    int `yaml:"max_idle_conns"`
-	ConnMaxLifetime int `yaml:"conn_max_lifetime"` // 单位：秒
-
-	// Stream Load配置
-	StreamLoadTimeout int `yaml:"stream_load_timeout"` // 秒
-	BatchSize         int `yaml:"batch_size"`          // 批量大小
+	FEHost            string `yaml:"fe_host"`
+	FEPort            int    `yaml:"fe_port"`
+	FEUsername        string `yaml:"fe_username"`
+	FEPassword        string `yaml:"fe_password"`
+	MySQLHost         string `yaml:"mysql_host"`
+	MySQLPort         int    `yaml:"mysql_port"`
+	MaxOpenConns      int    `yaml:"max_open_conns"`
+	MaxIdleConns      int    `yaml:"max_idle_conns"`
+	ConnMaxLifetime   int    `yaml:"conn_max_lifetime"`
+	StreamLoadTimeout int    `yaml:"stream_load_timeout"`
+	BatchSize         int    `yaml:"batch_size"`
 }
 
 // Auth 认证配置
@@ -101,13 +136,16 @@ type IConfig interface {
 	GetDatabase() DBConfig
 	GetRedis() RedisConfig
 	GetDoris() DorisConfig
+	GetRabbitMQ() RabbitMQConfig // 新增获取RabbitMQ配置方法
 	GetAuth() Auth
 	GetService() Service
-	GetHTTP() HTTPConfig // 新增获取HTTP配置方法
+	GetHTTP() HTTPConfig
 	IsDatabaseConfigured() bool
 	IsRedisConfigured() bool
 	IsDorisConfigured() bool
-	IsHTTPConfigured() bool // 新增检查HTTP配置方法
+	IsRabbitMQConfigured() bool // 新增检查RabbitMQ配置方法
+	IsHTTPConfigured() bool
+	IsAuthConfigured() bool
 }
 
 // 实现接口的具体类型
@@ -127,6 +165,10 @@ func (cw *configWrapper) GetDoris() DorisConfig {
 	return cw.config.Doris
 }
 
+func (cw *configWrapper) GetRabbitMQ() RabbitMQConfig {
+	return cw.config.RabbitMQ
+}
+
 func (cw *configWrapper) GetAuth() Auth {
 	return cw.config.Auth
 }
@@ -155,10 +197,15 @@ func (cw *configWrapper) IsRedisConfigured() bool {
 
 func (cw *configWrapper) IsDorisConfigured() bool {
 	doris := cw.config.Doris
-	// 检查FE配置（HTTP API方式）
 	return doris.FEHost != "" && doris.FEPort > 0
 }
 
+func (cw *configWrapper) IsRabbitMQConfigured() bool {
+	rabbit := cw.config.RabbitMQ
+	// 基础连接配置检查
+	return rabbit.Host != "" && rabbit.Port > 0
+}
+
 func (cw *configWrapper) IsHTTPConfigured() bool {
 	httpCfg := cw.config.HTTP
 	return httpCfg.Timeout > 0
@@ -180,7 +227,6 @@ func GetConfig() IConfig {
 		config, err := loadConfig()
 		if err != nil {
 			initErr = err
-			// 创建一个空的配置实例，避免nil指针
 			instance = &configWrapper{config: &Config{}}
 			return
 		}
@@ -196,7 +242,6 @@ func GetInitError() error {
 
 // loadConfig 加载配置文件
 func loadConfig() (*Config, error) {
-
 	configFile, err := findConfigFile()
 	if err != nil {
 		return nil, err
@@ -222,7 +267,7 @@ func loadConfig() (*Config, error) {
 			ReadTimeout:  3,
 			WriteTimeout: 3,
 			IdleTimeout:  300,
-			MaxConnAge:   0, // 0表示不限制
+			MaxConnAge:   0,
 		},
 		Doris: DorisConfig{
 			FEPort:            8030,
@@ -238,11 +283,34 @@ func loadConfig() (*Config, error) {
 			MaxIdleConns:        100,
 			MaxIdleConnsPerHost: 10,
 			IdleConnTimeout:     90,
-			MaxConnsPerHost:     0, // 0表示不限制
+			MaxConnsPerHost:     0,
 			DisableCompression:  false,
 			DisableKeepAlives:   false,
 		},
+		RabbitMQ: RabbitMQConfig{
+			Port:                 5672,
+			Username:             "guest",
+			Password:             "guest",
+			Vhost:                "/",
+			UseTLS:               false,
+			MaxOpenChannels:      10,
+			ReconnectDelay:       5000, // 5秒
+			MaxReconnectAttempts: 10,   // 最多重试10次
+			Heartbeat:            30,   // 30秒心跳
+			ChannelSize:          100,
+			DefaultExchange:      "amq.direct",
+			DefaultQueue:         "",
+			AutoAck:              false, // 默认手动确认
+			Mandatory:            false,
+			Immediate:            false,
+			PrefetchCount:        1, // 一次预取1条消息
+			PrefetchSize:         0,
+			Global:               false,
+			PublisherConfirms:    false, // 默认不启用发布确认
+			ConfirmTimeout:       30,    // 5秒确认超时
+		},
 	}
+
 	// 读取配置文件
 	data, err := os.ReadFile(configFile)
 	if err != nil {

types/types.go

@@ -9,16 +9,24 @@ type QueryRequest struct {
 	SQL              string                 `json:"sql" binding:"required"`
 	Params           map[string]interface{} `json:"params,omitempty"`           //名称参数
 	PositionalParams []interface{}          `json:"positionalParams,omitempty"` // 位置参数
-	WriterHeader     bool                   //如果查询返回cvs格式时候，包含表头还是不包含。true包含。第一行是表头
+	WriterHeader     bool                   `json:"writerHeader,omitempty"`     //如果查询返回cvs格式时候，包含表头还是不包含。true包含。第一行是表头
+	AgentUrl         string                 `json:"agentUrl,omitempty"`         //远程服务器地址
+	AgentToken       string                 `json:"agentToken,omitempty"`       //远程代理服务器token
+	DorisDatabase    string                 `json:"dorisDatabase,omitempty"`    //doris数据库名称
+	DorisTable       string                 `json:"dorisTable,omitempty"`       //doris数据库里的表名称
+
 }
 
 // QueryResult 查询结果
 type QueryResult struct {
-	Success bool        `json:"success"`
-	Data    interface{} `json:"data,omitempty"`
-	Error   string      `json:"error,omitempty"`
-	Count   int         `json:"count,omitempty"`
-	Time    string      `json:"time,omitempty"`
+	Success    bool          `json:"success"`
+	Data       interface{}   `json:"data,omitempty"`
+	Error      string        `json:"error,omitempty"`
+	Count      int           `json:"count,omitempty"`
+	Time       string        `json:"time,omitempty"`
+	QueryTime  time.Duration `json:"queryTime,omitempty""`
+	SaveTime   time.Duration `json:"saveTime,omitempty""`
+	TotalCount int           `json:"totalCount,omitempty""`
 }
 
 // PageResult 分页结果