Merge pull request #2 from CodeIntegrity/main

fix: 无base_url情况下添加判断

Author: DullJZ

Dockerfile

@@ -1,13 +1,33 @@
-FROM golang:1.22-alpine
+# 构建阶段
+FROM golang:1.22-alpine AS builder
+
+# 安装构建依赖
+RUN apk add --no-cache gcc musl-dev
 
 WORKDIR /app
 
+# 复制依赖文件
 COPY go.mod go.sum ./
 
+# 下载依赖
 RUN go mod download
+RUN go mod tidy
 
+# 复制源代码
 COPY . .
 
-RUN CGO_ENABLED=0 GOOS=linux go build -o main .
+# 构建应用
+RUN CGO_ENABLED=1 GOOS=linux go build -o main .
+
+# 运行阶段
+FROM alpine:latest
+
+# 安装 SQLite 运行时依赖
+RUN apk add --no-cache sqlite-libs
+
+WORKDIR /app
+
+# 从构建阶段复制二进制文件
+COPY --from=builder /app/main .
 
 CMD ["./main"]

go.mod

@@ -4,6 +4,7 @@ go 1.22.1
 
 require github.com/go-sql-driver/mysql v1.8.1
 
+
 require (
 	github.com/golang-sql/civil v0.0.0-20220223132316-b832511892a9 // indirect
 	github.com/golang-sql/sqlexp v0.1.0 // indirect

main.go

@@ -14,6 +14,7 @@ import (
 
 type Channel struct {
 	ID      int
+	Type    int
 	Name    string
 	BaseURL string
 	Key     string
@@ -26,7 +27,7 @@ var (
 )
 
 func fetchChannels() ([]Channel, error) {
-	query := "SELECT id, name, base_url, `key`, status FROM channels"
+	query := "SELECT id, type, name, base_url, `key`, status FROM channels"
 	rows, err := db.Raw(query).Rows()
 	if err != nil {
 		return nil, err
@@ -36,12 +37,20 @@ func fetchChannels() ([]Channel, error) {
 	var channels []Channel
 	for rows.Next() {
 		var c Channel
-		if err := rows.Scan(&c.ID, &c.Name, &c.BaseURL, &c.Key, &c.Status); err != nil {
+		if err := rows.Scan(&c.ID, &c.Type, &c.Name, &c.BaseURL, &c.Key, &c.Status); err != nil {
 			return nil, err
 		}
-		if c.BaseURL == "" {
-			c.BaseURL = "https://api.openai.com"
-		}
+		
+		switch c.Type {
+		case 40:
+			c.BaseURL = "https://api.siliconflow.cn"
+		case 999:
+			c.BaseURL = "https://api.siliconflow.cn"
+    case 1:
+		  if c.BaseURL == "" {
+			  c.BaseURL = "https://api.openai.com"
+		  }
+    }
 		// 检查是否在排除列表中
 		if contains(config.ExcludeChannel, c.ID) {
 			log.Printf("渠道 %s(ID:%d) 在排除列表中，跳过\n", c.Name, c.ID)
@@ -186,6 +195,7 @@ func main() {
 	}
 
 	db, err = NewDB(*config)
+  
 	if err != nil {
 		log.Fatal("数据库连接失败：", err)
 	}