security-c4po/security-c4po-api/src/main/kotlin/com/securityc4po/api/project/ProjectEntity.kt

27 lines
660 B
Kotlin

package com.securityc4po.api.project
import com.securityc4po.api.BaseEntity
import org.springframework.data.mongodb.core.mapping.Document
@Document(collection = "projects")
open class ProjectEntity(
data: Project
) : BaseEntity<Project>(data)
fun ProjectEntity.toProject() : Project {
return Project(
this.data.id,
this.data.client,
this.data.title,
this.data.createdAt,
this.data.tester,
this.data.createdBy
)
}
fun List<ProjectEntity>.toProjects(): List<Project> {
return this.map {
it.toProject()
}
}