mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 22:06:38 +08:00
ryan
299ac30ee4
- Purify core micro-kernel by removing context hardcoded helpers and reverse dependencies - Eliminate init() side effects in infra plugins with reversible lifecycle disposal - Completely isolate plugins by removing cross-plugin imports and using core/contracts - Introduce TaskService and RiskControlService contracts for unified cross-plugin APIs - Regenerate Swagger documentation and update developer guide matrix - Achieve 0 violations in check_cordis_architecture.sh and 100% test pass
118 lines
2.3 KiB
Go
118 lines
2.3 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Package database provides the relational database infrastructure plugin for Cordis.
|
|
package database
|
|
|
|
import (
|
|
"context"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"Wavelet/core"
|
|
"Wavelet/core/contracts"
|
|
)
|
|
|
|
// Option configures the database plugin.
|
|
type Option func(*Plugin)
|
|
|
|
// WithDB configures an explicit *gorm.DB instance for the plugin.
|
|
func WithDB(d *gorm.DB) Option {
|
|
return func(p *Plugin) {
|
|
p.db = d
|
|
}
|
|
}
|
|
|
|
// WithNamedDB registers a named secondary database connection.
|
|
func WithNamedDB(name string, d *gorm.DB) Option {
|
|
return func(p *Plugin) {
|
|
if p.namedDBs == nil {
|
|
p.namedDBs = make(map[string]*gorm.DB)
|
|
}
|
|
p.namedDBs[name] = d
|
|
}
|
|
}
|
|
|
|
// Plugin implements core.Plugin to provide contracts.DBService into the Cordis micro-kernel.
|
|
type Plugin struct {
|
|
db *gorm.DB
|
|
namedDBs map[string]*gorm.DB
|
|
}
|
|
|
|
// New creates a new database infrastructure plugin.
|
|
func New(opts ...Option) *Plugin {
|
|
p := &Plugin{
|
|
namedDBs: make(map[string]*gorm.DB),
|
|
}
|
|
for _, opt := range opts {
|
|
if opt != nil {
|
|
opt(p)
|
|
}
|
|
}
|
|
return p
|
|
}
|
|
|
|
// Name returns the unique identifier of the database plugin.
|
|
func (p *Plugin) Name() string {
|
|
return "database"
|
|
}
|
|
|
|
// Apply mounts the database service into the Context.
|
|
func (p *Plugin) Apply(ctx *core.Context) error {
|
|
targetDB := p.db
|
|
if targetDB == nil {
|
|
var err error
|
|
targetDB, err = InitDB()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
svc := &dbServiceImpl{
|
|
primary: targetDB,
|
|
namedDBs: p.namedDBs,
|
|
}
|
|
|
|
if sqlDB, err := targetDB.DB(); err == nil && sqlDB != nil {
|
|
ctx.OnDispose(func() error {
|
|
return sqlDB.Close()
|
|
})
|
|
}
|
|
|
|
core.Provide[contracts.DBService](ctx, svc)
|
|
return nil
|
|
}
|
|
|
|
// NewService wraps a GORM DB instance into a contracts.DBService.
|
|
func NewService(primary *gorm.DB) contracts.DBService {
|
|
return &dbServiceImpl{primary: primary}
|
|
}
|
|
|
|
type dbServiceImpl struct {
|
|
primary *gorm.DB
|
|
namedDBs map[string]*gorm.DB
|
|
}
|
|
|
|
func (s *dbServiceImpl) GORM() *gorm.DB {
|
|
if s.primary != nil {
|
|
return s.primary
|
|
}
|
|
return DB(context.Background())
|
|
}
|
|
|
|
func (s *dbServiceImpl) DB(ctx context.Context) *gorm.DB {
|
|
if s.primary != nil {
|
|
return s.primary.WithContext(ctx)
|
|
}
|
|
return DB(ctx)
|
|
}
|
|
|
|
func (s *dbServiceImpl) Named(name string) *gorm.DB {
|
|
if s.namedDBs != nil {
|
|
if d, ok := s.namedDBs[name]; ok && d != nil {
|
|
return d
|
|
}
|
|
}
|
|
return s.GORM()
|
|
}
|