overwrite default connection for model

contracts/database/orm/orm.go

@@ -11,6 +11,7 @@ type Orm interface {
 	DB() (*sql.DB, error)
 	Query() Query
 	Factory() Factory
+	Model(value any) Orm
 	Observe(model any, observer Observer)
 	Transaction(txFunc func(tx Transaction) error) error
 	WithContext(ctx context.Context) Orm
@@ -84,6 +85,10 @@ type Association interface {
 	Count() int64
 }
 
+type Model interface {
+	Connection() string
+}
+
 //go:generate mockery --name=Cursor
 type Cursor interface {
 	Scan(value any) error

database/orm.go

@@ -76,6 +76,22 @@ func (r *OrmImpl) Factory() ormcontract.Factory {
 	return NewFactoryImpl(r.query)
 }
 
+func (r *OrmImpl) Model(value any) ormcontract.Orm {
+	model, ok := value.(ormcontract.Model)
+	if !ok {
+		return r
+	}
+
+	// Check if the model has a connection specified
+	if conn := model.Connection(); conn != "" {
+		r.query = r.query.Model(value)
+		return r.Connection(conn)
+	}
+
+	// If the model doesn't have a connection specified, return the current OrmImpl instance
+	return r
+}
+
 func (r *OrmImpl) Observe(model any, observer ormcontract.Observer) {
 	orm.Observers = append(orm.Observers, orm.Observer{
 		Model:    model,

database/orm_test.go

@@ -29,6 +29,10 @@ type User struct {
 	Avatar string
 }
 
+func (u *User) Connection() string {
+	return "postgresql"
+}
+
 type OrmSuite struct {
 	suite.Suite
 	orm *OrmImpl
@@ -109,6 +113,10 @@ func (s *OrmSuite) TestConnection() {
 	}
 }
 
+func (s *OrmSuite) TestModel() {
+	s.NotNil(s.orm.Model(&User{}))
+}
+
 func (s *OrmSuite) TestDB() {
 	db, err := s.orm.DB()
 	s.NotNil(db)