/* * Copyright 2020 The Dragonfly Authors * * 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 database import ( "github.com/go-redis/redis/v8" "gorm.io/gorm" "d7y.io/dragonfly/v2/manager/config" ) type Database struct { DB *gorm.DB RDB *redis.Client } func New(cfg *config.Config) (*Database, error) { db, err := newMyqsl(cfg) if err != nil { return nil, err } rdb, err := NewRedis(cfg.Database.Redis) if err != nil { return nil, err } return &Database{ DB: db, RDB: rdb, }, nil }