-
Notifications
You must be signed in to change notification settings - Fork 188
/
SqaleRepoContext.java
99 lines (84 loc) · 4.23 KB
/
SqaleRepoContext.java
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
/*
* Copyright (C) 2010-2021 Evolveum and contributors
*
* This work is dual-licensed under the Apache License 2.0
* and European Union Public License. See LICENSE file for details.
*/
package com.evolveum.midpoint.repo.sqale;
import javax.annotation.PostConstruct;
import javax.sql.DataSource;
import javax.xml.namespace.QName;
import com.querydsl.sql.types.EnumAsObjectType;
import org.jetbrains.annotations.NotNull;
import com.evolveum.midpoint.repo.sqale.qmodel.common.MContainerType;
import com.evolveum.midpoint.repo.sqale.qmodel.object.MObjectType;
import com.evolveum.midpoint.repo.sqale.qmodel.ref.MReferenceType;
import com.evolveum.midpoint.repo.sqlbase.JdbcRepositoryConfiguration;
import com.evolveum.midpoint.repo.sqlbase.JdbcSession;
import com.evolveum.midpoint.repo.sqlbase.SqlRepoContext;
import com.evolveum.midpoint.repo.sqlbase.mapping.QueryModelMappingRegistry;
import com.evolveum.midpoint.repo.sqlbase.querydsl.QuerydslJsonbType;
import com.evolveum.midpoint.xml.ns._public.common.common_3.*;
/**
* SQL repository context adding support for QName cache.
*/
public class SqaleRepoContext extends SqlRepoContext {
private final UriCache uriCache;
public SqaleRepoContext(
JdbcRepositoryConfiguration jdbcRepositoryConfiguration,
DataSource dataSource,
QueryModelMappingRegistry mappingRegistry) {
super(jdbcRepositoryConfiguration, dataSource, mappingRegistry);
// each enum type must be registered if we want to map it as objects (to PG enum types)
querydslConfig.register(new EnumAsObjectType<>(ActivationStatusType.class));
querydslConfig.register(new EnumAsObjectType<>(AvailabilityStatusType.class));
querydslConfig.register(new EnumAsObjectType<>(MContainerType.class));
querydslConfig.register(new EnumAsObjectType<>(MObjectType.class));
querydslConfig.register(new EnumAsObjectType<>(MReferenceType.class));
querydslConfig.register(new EnumAsObjectType<>(LockoutStatusType.class));
querydslConfig.register(new EnumAsObjectType<>(OperationExecutionRecordTypeType.class));
querydslConfig.register(new EnumAsObjectType<>(OperationResultStatusType.class));
querydslConfig.register(new EnumAsObjectType<>(OrientationType.class));
querydslConfig.register(new EnumAsObjectType<>(ResourceAdministrativeStateType.class));
querydslConfig.register(new EnumAsObjectType<>(ShadowKindType.class));
querydslConfig.register(new EnumAsObjectType<>(SynchronizationSituationType.class));
querydslConfig.register(new EnumAsObjectType<>(TaskBindingType.class));
querydslConfig.register(new EnumAsObjectType<>(TaskExecutionStateType.class));
querydslConfig.register(new EnumAsObjectType<>(TaskRecurrenceType.class));
querydslConfig.register(new EnumAsObjectType<>(TaskWaitingReasonType.class));
querydslConfig.register(new EnumAsObjectType<>(ThreadStopActionType.class));
querydslConfig.register(new EnumAsObjectType<>(TimeIntervalStatusType.class));
// JSONB type support
querydslConfig.register(new QuerydslJsonbType());
uriCache = new UriCache();
}
// This has nothing to do with "repo cache" which is higher than this.
@PostConstruct
public void clearCaches() {
try (JdbcSession jdbcSession = newJdbcSession().startReadOnlyTransaction()) {
uriCache.initialize(jdbcSession);
jdbcSession.commit();
}
}
/** @see UriCache#searchId(QName) */
@NotNull
public Integer searchCachedUriId(QName qName) {
return uriCache.searchId(qName);
}
/** @see UriCache#searchId(String) */
public Integer searchCachedUriId(String uri) {
return uriCache.searchId(uri);
}
/** @see UriCache#resolveUriToId(String) */
public @NotNull Integer resolveUriToId(String uri) {
return uriCache.resolveUriToId(uri);
}
/** @see UriCache#resolveUriToId(QName) */
public Integer resolveUriToId(QName uri) {
return uriCache.resolveUriToId(uri);
}
/** Returns ID for URI creating new cache row in DB as needed. */
public Integer processCacheableUri(String uri, JdbcSession jdbcSession) {
return uriCache.processCacheableUri(uri, jdbcSession);
}
}