Function InitDB

db_controller/db_controller.go:102–137

(_cfg config.Config)

)

func InitDB(_cfg config.Config) {
	cfg = _cfg
	// Ensure the directory exists
	if cfg.DatabaseType == "sqlite" {
		dbDir := filepath.Dir(cfg.DatabasePath)
		if err := os.MkdirAll(dbDir, os.ModePerm); err != nil {
			panic("failed to create database directory")
		}
	}

	// Initialize the database connection
	var err error
	switch cfg.DatabaseType {
	case "sqlite":
		db, err = gorm.Open(sqlite.Open(cfg.DatabasePath), &gorm.Config{})
	case "mysql":
		db, err = gorm.Open(mysql.Open(cfg.DatabasePath), &gorm.Config{})
	case "postgres":
		db, err = gorm.Open(postgres.Open(cfg.DatabasePath), &gorm.Config{})
	default:
		panic("unsupported database type")
	}

	if err != nil {
		panic("failed to connect database")
	}

	// Auto-migrate the schema
	db.AutoMigrate(&Token{})
	db.AutoMigrate(&MessageContext{})
	db.AutoMigrate(&TwitterIDs{})
	db.AutoMigrate(&AnalyticData{})
	db.AutoMigrate(&ShortLink{})

	StartPeriodicAnalyticsWriter(time.Minute)
}

// StoreToken stores an encrypted access token and refresh token in the database.
// It returns the UUID of the stored token or an error if the operation fails.