-
Notifications
You must be signed in to change notification settings - Fork 162
/
PersistenceMetadataImpl.kt
59 lines (45 loc) · 1.71 KB
/
PersistenceMetadataImpl.kt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
package misk.hibernate
import com.google.common.collect.ImmutableSet
import javax.inject.Inject
import org.hibernate.SessionFactory
import org.hibernate.persister.entity.AbstractEntityPersister
import com.google.common.base.Preconditions.checkState
internal class PersistenceMetadataImpl @Inject constructor(
private val sessionFactory: SessionFactory
) : PersistenceMetadata {
override fun <T: DbEntity<T>> getTableName(entityType: T): String {
return hibernateMetadataForClass(entityType).tableName
}
override fun <T: DbEntity<T>> getColumnNames(
entityType: T
): ImmutableSet<String> {
val result = ImmutableSet.builder<String>()
val classMetadata = hibernateMetadataForClass(entityType)
for (column in classMetadata.identifierColumnNames) {
result.add(column)
}
val propertyNames = classMetadata.propertyNames
for (i in propertyNames.indices) {
result.add(*classMetadata.getPropertyColumnNames(i))
}
return result.build()
}
override fun <T: DbEntity<T>> getColumnNames(
entityType: T,
propertyName: String
): Array<String> {
return hibernateMetadataForClass(entityType).getPropertyColumnNames(propertyName)
}
private fun <T: DbEntity<T>> hibernateMetadataForClass(
entityType: T
): AbstractEntityPersister {
val hibernateMetadata = sessionFactory.getClassMetadata(entityType::class.qualifiedName)
checkState(hibernateMetadata != null,
"${entityType::class.qualifiedName} does not map to a known entity type"
)
checkState(hibernateMetadata is AbstractEntityPersister,
"${entityType::class.qualifiedName} does not map to a persistent class"
)
return hibernateMetadata as AbstractEntityPersister
}
}