fix store bug
This commit is contained in:
@@ -29,8 +29,8 @@ type ApiKeyDAO struct {
|
|||||||
db *gorm.DB
|
db *gorm.DB
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewApiKeyDAO(db *gorm.DB) *ApiKeyDAO {
|
func NewApiKeyDAO(cfg *config.Config, db *gorm.DB) *ApiKeyDAO {
|
||||||
return &ApiKeyDAO{db: db}
|
return &ApiKeyDAO{cfg: cfg, db: db}
|
||||||
}
|
}
|
||||||
|
|
||||||
// CreateApiKey 创建ApiKey
|
// CreateApiKey 创建ApiKey
|
||||||
|
|||||||
@@ -27,7 +27,7 @@ func InitTeamHandler(ctx context.Context, cfg *config.Config, db *gorm.DB) (*con
|
|||||||
userService := service.NewUserService(db, userDAO)
|
userService := service.NewUserService(db, userDAO)
|
||||||
tokenDAO := dao.NewTokenDAO(db)
|
tokenDAO := dao.NewTokenDAO(db)
|
||||||
tokenService := service.NewTokenService(tokenDAO)
|
tokenService := service.NewTokenService(tokenDAO)
|
||||||
apiKeyDAO := dao.NewApiKeyDAO(db)
|
apiKeyDAO := dao.NewApiKeyDAO(cfg, db)
|
||||||
apiKeyService := service.NewApiKeyService(db, apiKeyDAO)
|
apiKeyService := service.NewApiKeyService(db, apiKeyDAO)
|
||||||
usageDAO := dao.NewUsageDAO(cfg, db)
|
usageDAO := dao.NewUsageDAO(cfg, db)
|
||||||
dailyUsageDAO := dao.NewDailyUsageDAO(cfg, db)
|
dailyUsageDAO := dao.NewDailyUsageDAO(cfg, db)
|
||||||
@@ -41,7 +41,7 @@ func InitAPIHandler(ctx context.Context, cfg *config.Config, db *gorm.DB) (*cont
|
|||||||
userServiceImpl := service2.NewUserService(cfg, db, userDAO)
|
userServiceImpl := service2.NewUserService(cfg, db, userDAO)
|
||||||
tokenDAO := dao.NewTokenDAO(db)
|
tokenDAO := dao.NewTokenDAO(db)
|
||||||
tokenServiceImpl := service2.NewTokenService(db, tokenDAO)
|
tokenServiceImpl := service2.NewTokenService(db, tokenDAO)
|
||||||
apiKeyDAO := dao.NewApiKeyDAO(db)
|
apiKeyDAO := dao.NewApiKeyDAO(cfg, db)
|
||||||
apiKeyServiceImpl := service2.NewApiKeyService(db, apiKeyDAO)
|
apiKeyServiceImpl := service2.NewApiKeyService(db, apiKeyDAO)
|
||||||
webAuthnService, err := service2.NewWebAuthnService(cfg, db)
|
webAuthnService, err := service2.NewWebAuthnService(cfg, db)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -54,7 +54,7 @@ func InitAPIHandler(ctx context.Context, cfg *config.Config, db *gorm.DB) (*cont
|
|||||||
|
|
||||||
func InitProxyHandler(ctx context.Context, cfg *config.Config, db *gorm.DB, wg *sync.WaitGroup) (*controller3.Proxy, error) {
|
func InitProxyHandler(ctx context.Context, cfg *config.Config, db *gorm.DB, wg *sync.WaitGroup) (*controller3.Proxy, error) {
|
||||||
userDAO := dao.NewUserDAO(db)
|
userDAO := dao.NewUserDAO(db)
|
||||||
apiKeyDAO := dao.NewApiKeyDAO(db)
|
apiKeyDAO := dao.NewApiKeyDAO(cfg, db)
|
||||||
tokenDAO := dao.NewTokenDAO(db)
|
tokenDAO := dao.NewTokenDAO(db)
|
||||||
usageDAO := dao.NewUsageDAO(cfg, db)
|
usageDAO := dao.NewUsageDAO(cfg, db)
|
||||||
dailyUsageDAO := dao.NewDailyUsageDAO(cfg, db)
|
dailyUsageDAO := dao.NewDailyUsageDAO(cfg, db)
|
||||||
|
|||||||
Reference in New Issue
Block a user