Scala编程的蛋糕模式和依赖注入

jopen 9年前

 

如果你是一个Java开发者,熟悉 依赖注入 模式, 深度依赖Spring框架的话,在使用Scala做开发时,会遇到一个问题,在Scala世界里,如何实现类似Spring框架的依赖注入呢?

尽管函数式编程的信徒认为他们不需要DI框架,高阶(high-order)函数足够了。但是对于同时支持面向对象的编程和函数式编程的Scala来说,依赖注入是很好的实现应用的一种设计模式。

蛋糕模式(Cake pattern)是Scala实现依赖注入的方式之一。

蛋糕模式是 Scala 之父 Martin Odersky 在其论文 Scalable Component Abstractions 中首先提到。

什么是蛋糕模式呢? 一个非正式的但是很形象的解释是:

  1. 蛋糕有很多风味, 你可以根据你的需要增加相应的风味。依赖注入就是增加调料。
  2. 蛋糕有很多层 (layer)。如果你想要一个更大的蛋糕你就可以增加更多的层。

我们先以Spring常用的User数据库读取的实现为例,看看Scala 风格的依赖注入(蛋糕模式)是如何实现的。

Java/Spring风格的依赖注入实现

一个传统的Spring实现是将程序划分为"Repository"层(DAO layer) 和Service层。

case class User (name:String)

trait UserRepository {

def create(user: User)

def find(name: String)

def update(user: User)

def delete(user: User)

}

class MockUserRepository extends UserRepository {

def create(user: User) = println( "creating user: " + user)

def find(name: String) = println( "finding user: " + name)

def update(user: User) = println( "udating user: " + user)

def delete(user: User) = println( "deleting user: " + user)

}

class UserService {

@Resource

var userRepository: UserRepository = _

def create(user: User) = userRepository.create(user)

def find(name: String) = userRepository.find(name)

def update(user: User) = userRepository.update(user)

def delete(user: User) = userRepository.delete(user)

}

我们可能有多个UserRepository的实现, 比如JPA, JDBC, Hibernate, iBATIS 等,然后将具体的实现通过Spring注入到 UserService中。这里我们用一个Mock来简单这些这个Trait, 然后模拟Spring注入,(Spring会根据配置和Reflect自动实现实例化和注入,这里我们只是模拟其原理,并没有使用Spring框架):

object Main extends App {

val service = new UserService()

val userRepository = new MockUserRepository()

service.userRepository = userRepository // inject userRepository into userService

service.create(User( "user" ))

}

Scala/Cake pattern实现依赖注入

和上面的代码类似,我们有三个class/trait:UserRepository,MockUserRepository和UserService, 其中MockUserRepository是UserRepository的具体实现,

现在我们想把MockUserRepository注入到UserService。注意UserService和UserRepository目前没有任何依赖关系。

Scala的蛋糕模式中我们需要声明几个Component:

trait UserRepositoryComponent {

val userRepository:UserRepository

}

trait UserServiceComponent {

this : UserRepositoryComponent =>

val userService: UserService

}

这里使用 self-type annotation 声明UserServiceComponent需要UserRepositoryComponent(this: UserRepositoryComponent =>)。 如果需要多个依赖,可以使用下面的格式:

this : Foo with Bar with Baz =>

剩下的就是注入了,生成一个ComponentRegistry对象:

object ComponentRegistry extends UserServiceComponent with UserRepositoryComponent {

override val userRepository: UserRepository = new MockUserRepository

override val userService: UserService = new UserService(userRepository)

}

挺漂亮的实现, 如果相应的依赖没有提供,或者拼写错误等,编译时能立刻提示我们。

这还有一个好处就是所有的对象都是val类型的。

这样你就可以通过ComponentRegistry.userService来使用顶层的组件了。

这样我们可以实现一种"干净"的方式实现不同的组件注入,比如我们想单元测试userService,其中它的依赖userRepository通过mock的方式提供:

import org.scalatest.mock.MockitoSugar

import org.mockito.Matchers._

import org.mockito.Mockito._

trait TestEnvironment extends UserServiceComponent with UserRepositoryComponent {

val userRepository = MockitoSugar.mock[UserRepository]

val userService = new UserService(userRepository)

}

object Test extends App with TestEnvironment {

val user = User( "user" )

userService.create(user)

verify(userRepository, times( 1 )).create(any[User])

}

可以很方便的为测试实现新的依赖注入,其中userRepository由mock实现。

这就是蛋糕模式的实现。通过 component trait 抽象接口和依赖,最后在一个ComponentRegistry注入各个具体的实现。