refact:openteam

2025-01-01 23:10:01 +08:00
parent 1f5e1c221c
commit 65d6d12972
18 changed files with 1126 additions and 106 deletions
--- a/team/model/apikey.go
+++ b/team/model/apikey.go
@@ -0,0 +1,49 @@
+package model
+
+import (
+	"time"
+
+	"github.com/lib/pq"
+)
+
+type ApiKey_PG struct {
+	ID             int64          `gorm:"column:id;primaryKey;autoIncrement"`
+	Name           string         `gorm:"column:name;not null;unique;index:idx_apikey_name"`
+	ApiType        string         `gorm:"column:apitype;not null;unique;index:idx_apikey_apitype"`
+	ApiKey         string         `gorm:"column:apikey;not null;unique;index:idx_apikey_apikey"`
+	Status         int            `gorm:"type:int;default:0"` // enabled 0, disabled 1
+	Endpoint       string         `gorm:"column:endpoint"`
+	ResourceNmae   string         `gorm:"column:resource_name"`
+	DeploymentName string         `gorm:"column:deployment_name"`
+	ApiSecret      string         `gorm:"column:api_secret"`
+	ModelPrefix    string         `gorm:"column:model_prefix"`
+	ModelAlias     string         `gorm:"column:model_alias"`
+	SupportModels  pq.StringArray `gorm:"type:text[]"`
+	UpdatedAt      time.Time      `json:"updatedAt,omitempty"`
+	CreatedAt      time.Time      `json:"createdAt,omitempty"`
+}
+
+func (ApiKey_PG) TableName() string {
+	return "apikeys"
+}
+
+type ApiKey struct {
+	ID             int64     `gorm:"column:id;primaryKey;autoIncrement"`
+	Name           string    `gorm:"column:name;not null;unique;index:idx_apikey_name"`
+	ApiType        string    `gorm:"column:apitype;not null;unique;index:idx_apikey_apitype"`
+	ApiKey         string    `gorm:"column:apikey;not null;unique;index:idx_apikey_apikey"`
+	Status         int       `json:"status" gorm:"type:int;default:0"` // enabled 0, disabled 1
+	Endpoint       string    `gorm:"column:endpoint"`
+	ResourceNmae   string    `gorm:"column:resource_name"`
+	DeploymentName string    `gorm:"column:deployment_name"`
+	ApiSecret      string    `gorm:"column:api_secret"`
+	ModelPrefix    string    `gorm:"column:model_prefix"`
+	ModelAlias     string    `gorm:"column:model_alias"`
+	SupportModels  []string  `gorm:"type:json"`
+	CreatedAt      time.Time `json:"created_at,omitempty" gorm:"autoUpdateTime"`
+	UpdatedAt      time.Time `json:"updated_at,omitempty" gorm:"autoCreateTime"`
+}
+
+func (ApiKey) TableName() string {
+	return "apikeys"
+}