Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commitcc1e470

Browse files
authored
Initial attempt at adding jOOQ (#59)
* Initial attempt at adding jOOQ
1 parentf533ce5 commitcc1e470

File tree

10 files changed

+516
-11
lines changed

10 files changed

+516
-11
lines changed

‎docker/mysql/setup.sh‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,3 +2,5 @@
22

33
echo'creating databases'
44
mysql -u root -e'CREATE DATABASE IF NOT EXISTS stubbornjava;'
5+
6+
mysql -u root -e'CREATE DATABASE IF NOT EXISTS sj_access;'

‎stubbornjava-common/build.gradle‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,5 +38,6 @@ dependencies {
3838
compile libs.jooqCodegen
3939

4040
testCompile libs.junit
41+
testCompile libs.hsqldb
4142
}
4243
// {{end:dependencies}}
Lines changed: 93 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,93 @@
1+
packagecom.stubbornjava.common.db;
2+
3+
importjava.util.function.Supplier;
4+
5+
importjavax.sql.DataSource;
6+
7+
importorg.jooq.DSLContext;
8+
importorg.slf4j.Logger;
9+
importorg.slf4j.LoggerFactory;
10+
11+
importcom.stubbornjava.common.Configs;
12+
importcom.stubbornjava.common.HealthChecks;
13+
importcom.stubbornjava.common.Metrics;
14+
importcom.typesafe.config.Config;
15+
importcom.zaxxer.hikari.HikariDataSource;
16+
17+
publicclassDSLs {
18+
privatestaticfinalLoggerlogger =LoggerFactory.getLogger(DSLs.class);
19+
privatestaticfinalConfigconf =Configs.properties();
20+
21+
privateDSLs() {}
22+
23+
// Letting HikariDataSource leak out on purpose here. It won't go very far.
24+
privateenumTransactional {
25+
INSTANCE(ConnectionPool.getDataSourceFromConfig(conf.getConfig("pools.transactional"),Metrics.registry(),HealthChecks.getHealthCheckRegistry()));
26+
privatefinalHikariDataSourcedataSource;
27+
privateTransactional(HikariDataSourcedatasource) {
28+
this.dataSource =datasource;
29+
}
30+
publicHikariDataSourcegetDataSource() {
31+
returndataSource;
32+
}
33+
}
34+
privatestaticHikariDataSourcegetTransactionalDataSource() {
35+
returnTransactional.INSTANCE.getDataSource();
36+
}
37+
38+
privateenumProcessing {
39+
INSTANCE(ConnectionPool.getDataSourceFromConfig(conf.getConfig("pools.processing"),Metrics.registry(),HealthChecks.getHealthCheckRegistry()));
40+
privatefinalHikariDataSourcedataSource;
41+
privateProcessing(HikariDataSourcedatasource) {
42+
this.dataSource =datasource;
43+
}
44+
publicHikariDataSourcegetDataSource() {
45+
returndataSource;
46+
}
47+
}
48+
49+
privatestaticHikariDataSourcegetProcessingDataSource() {
50+
returnProcessing.INSTANCE.getDataSource();
51+
}
52+
53+
publicstaticDSLContextany() {
54+
returnThreadLocalJooqConfig.getCurrentContext();
55+
}
56+
57+
publicstaticDSLContextWrappertransactional() {
58+
returnnewDSLContextWrapper(Transactional.INSTANCE.getDataSource());
59+
}
60+
61+
publicstaticDSLContextWrapperprocessing() {
62+
returnnewDSLContextWrapper(Processing.INSTANCE.getDataSource());
63+
}
64+
65+
publicstaticfinalclassDSLContextWrapper {
66+
privatefinalHikariDataSourceds;
67+
68+
publicDSLContextWrapper(HikariDataSourceds) {
69+
this.ds =ds;
70+
}
71+
72+
publicfinalDSLContextget() {
73+
returnThreadLocalJooqConfig.ensureNamedConfiguration(ds.getPoolName());
74+
}
75+
76+
publicfinalvoidnewTransaction(Runnablerunnable) {
77+
ThreadLocalJooqConfig.threadLocalTransaction(ds.getPoolName(),ds,runnable);
78+
}
79+
80+
publicfinal <T>TnewTransactionResult(Supplier<T>supplier) {
81+
returnThreadLocalJooqConfig.threadLocalTransactionResult(ds.getPoolName(),ds,supplier);
82+
}
83+
}
84+
85+
publicstaticvoidmain(String[]args) {
86+
logger.debug("starting");
87+
DataSourceprocessing =DSLs.getProcessingDataSource();
88+
logger.debug("processing started");
89+
DataSourcetransactional =DSLs.getTransactionalDataSource();
90+
logger.debug("transactional started");
91+
logger.debug("done");
92+
}
93+
}

‎stubbornjava-common/src/main/java/com/stubbornjava/common/db/jooq/JooqConfig.java‎renamed to ‎stubbornjava-common/src/main/java/com/stubbornjava/common/db/JooqConfig.java‎

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,11 @@
1-
packagecom.stubbornjava.common.db.jooq;
1+
packagecom.stubbornjava.common.db;
22

33
importjava.util.Arrays;
44
importjava.util.List;
55

66
importjavax.sql.DataSource;
77

8+
importorg.jooq.Configuration;
89
importorg.jooq.SQLDialect;
910
importorg.jooq.impl.DataSourceConnectionProvider;
1011
importorg.jooq.impl.DefaultConfiguration;
@@ -14,11 +15,14 @@
1415

1516
publicclassJooqConfig {
1617

17-
publicstaticDefaultConfigurationdefaultConfigFromDataSource(DataSourceds) {
18+
publicstaticConfigurationdefaultConfigFromDataSource(DataSourceds) {
1819
DataSourceConnectionProviderdcp =newDataSourceConnectionProvider(ds);
19-
DefaultConfigurationjooqConfig =newDefaultConfiguration();
20+
ConfigurationjooqConfig =newDefaultConfiguration();
2021
jooqConfig.set(SQLDialect.MYSQL);
2122
jooqConfig.set(dcp);
23+
//jooqConfig.set(new ThreadLocalTransactionProvider(dcp));
24+
jooqConfig.settings()
25+
.withExecuteWithOptimisticLockingExcludeUnversioned(true);
2226
returnjooqConfig;
2327
}
2428

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
packagecom.stubbornjava.common.db;
2+
3+
importorg.jooq.Configuration;
4+
5+
publicclassNamedConfiguration {
6+
privatefinalStringname;
7+
privatefinalConfigurationconfiguration;
8+
9+
publicNamedConfiguration(Stringname,Configurationconfiguration) {
10+
super();
11+
this.name =name;
12+
this.configuration =configuration;
13+
}
14+
15+
publicStringgetName() {
16+
returnname;
17+
}
18+
19+
publicConfigurationgetConfiguration() {
20+
returnconfiguration;
21+
}
22+
}
Lines changed: 157 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,157 @@
1+
packagecom.stubbornjava.common.db;
2+
3+
importjava.util.ArrayList;
4+
importjava.util.Arrays;
5+
importjava.util.Collection;
6+
importjava.util.Collections;
7+
importjava.util.List;
8+
importjava.util.function.Function;
9+
importjava.util.function.Supplier;
10+
11+
importorg.jooq.Condition;
12+
importorg.jooq.DSLContext;
13+
importorg.jooq.Field;
14+
importorg.jooq.Record;
15+
importorg.jooq.RecordMapper;
16+
importorg.jooq.RecordUnmapper;
17+
importorg.jooq.UniqueKey;
18+
importorg.jooq.UpdatableRecord;
19+
importorg.jooq.impl.TableImpl;
20+
21+
publicclassTableCrud<RecextendsUpdatableRecord<Rec>,T> {
22+
privatefinalTableImpl<Rec>table;
23+
privatefinalRecordMapper<Record,T>mapper;
24+
privatefinalRecordUnmapper<T,Rec>unmapper;
25+
privatefinalSupplier<DSLContext>configSupplier;
26+
publicTableCrud(TableImpl<Rec>table,
27+
RecordMapper<Rec,T>mapper,
28+
RecordUnmapper<T,Rec>unmapper,
29+
Supplier<DSLContext>configSupplier) {
30+
super();
31+
this.table =table;
32+
this.mapper = (RecordMapper<Record,T>)mapper;
33+
this.unmapper =unmapper;
34+
this.configSupplier =configSupplier;
35+
}
36+
37+
publicTinsertReturning(Tobj) {
38+
Recrec =records(Collections.singletonList(obj),false).get(0);
39+
rec.insert();
40+
returnrec.map(mapper);
41+
}
42+
43+
publicvoidinsert(Tobj) {
44+
insert(Collections.singletonList(obj));
45+
}
46+
47+
@SuppressWarnings("unchecked")
48+
publicvoidinsert(T...objects) {
49+
insert(Arrays.asList(objects));
50+
}
51+
52+
publicvoidinsert(Collection<T>objects) {
53+
// Execute a batch INSERT
54+
if (objects.size() >1) {
55+
configSupplier.get().batchInsert(records(objects,false)).execute();
56+
}
57+
58+
// Execute a regular INSERT
59+
elseif (objects.size() ==1) {
60+
records(objects,false).get(0).insert();
61+
}
62+
}
63+
64+
publicvoidupdate(Tobj) {
65+
update(Collections.singletonList(obj));
66+
}
67+
68+
@SuppressWarnings("unchecked")
69+
publicvoidupdate(T...objects) {
70+
update(Arrays.asList(objects));
71+
}
72+
73+
publicvoidupdate(Collection<T>objects) {
74+
// Execute a batch UPDATE
75+
if (objects.size() >1) {
76+
configSupplier.get().batchUpdate(records(objects,false)).execute();
77+
}
78+
79+
// Execute a regular UPDATE
80+
elseif (objects.size() ==1) {
81+
records(objects,false).get(0).update();
82+
}
83+
}
84+
85+
publicvoiddelete(Tobj) {
86+
delete(Collections.singletonList(obj));
87+
}
88+
89+
@SuppressWarnings("unchecked")
90+
publicvoiddelete(T...objects) {
91+
delete(Arrays.asList(objects));
92+
}
93+
94+
publicvoiddelete(Collection<T>objects) {
95+
// Execute a batch DELETE
96+
if (objects.size() >1) {
97+
configSupplier.get().batchDelete(records(objects,false)).execute();
98+
}
99+
100+
// Execute a regular DELETE
101+
elseif (objects.size() ==1) {
102+
records(objects,false).get(0).delete();
103+
}
104+
}
105+
106+
publicTfindOne(Function<TableImpl<Rec>,Condition>func) {
107+
returnconfigSupplier.get().fetchOne(table,func.apply(table)).map(mapper);
108+
}
109+
110+
publicList<T>find(Function<TableImpl<Rec>,Condition>func) {
111+
returnconfigSupplier.get().fetch(table,func.apply(table)).map(mapper);
112+
}
113+
114+
publicintdeleteWhere(Function<TableImpl<Rec>,Condition>func) {
115+
returnconfigSupplier.get().deleteFrom(table).where(func.apply(table)).execute();
116+
}
117+
118+
// Copy pasted from jOOQ's DAOImpl.java
119+
private/* non-final */Field<?>[]pk() {
120+
UniqueKey<?>key =table.getPrimaryKey();
121+
returnkey ==null ?null :key.getFieldsArray();
122+
}
123+
124+
// Copy pasted from jOOQ's DAOImpl.java
125+
private/* non-final */List<Rec>records(Collection<T>objects,booleanforUpdate) {
126+
List<Rec>result =newArrayList<>();
127+
Field<?>[]pk =pk();
128+
129+
for (Tobject :objects) {
130+
Recrecord =unmapper.unmap(object);
131+
record.attach(configSupplier.get().configuration());
132+
133+
if (forUpdate &&pk !=null)
134+
for (Field<?>field :pk)
135+
record.changed(field,false);
136+
137+
resetChangedOnNotNull(record);
138+
result.add(record);
139+
}
140+
141+
returnresult;
142+
}
143+
144+
// Copy pasted from jOOQ's Tools.java
145+
/**
146+
* [#2700] [#3582] If a POJO attribute is NULL, but the column is NOT NULL
147+
* then we should let the database apply DEFAULT values
148+
*/
149+
privatestaticfinalvoidresetChangedOnNotNull(Recordrecord) {
150+
intsize =record.size();
151+
152+
for (inti =0;i <size;i++)
153+
if (record.get(i) ==null)
154+
if (!record.field(i).getDataType().nullable())
155+
record.changed(i,false);
156+
}
157+
}
Lines changed: 65 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,65 @@
1+
packagecom.stubbornjava.common.db;
2+
3+
importjava.util.function.Supplier;
4+
5+
importjavax.sql.DataSource;
6+
7+
importorg.jooq.Configuration;
8+
importorg.jooq.DSLContext;
9+
importorg.jooq.impl.DSL;
10+
importorg.slf4j.Logger;
11+
importorg.slf4j.LoggerFactory;
12+
13+
importcom.google.common.base.Preconditions;
14+
15+
publicclassThreadLocalJooqConfig {
16+
privatestaticfinalLoggerlog =LoggerFactory.getLogger(ThreadLocalJooqConfig.class);
17+
privatestaticfinalThreadLocal<NamedConfiguration>configurations =newThreadLocal<>();
18+
19+
// This is used to fetch any currently existing configuration.
20+
publicstaticDSLContextgetCurrentContext() {
21+
returnDSL.using(safeGet().getConfiguration());
22+
}
23+
24+
// Make sure we have the expected named config.
25+
// This is useful if we know we are running a long running query
26+
// and want to ensure we are using a different connection pool than
27+
// the default pool.
28+
publicstaticDSLContextensureNamedConfiguration(Stringname) {
29+
NamedConfigurationconfig =safeGet();
30+
if (!name.equals(config.getName())) {
31+
log.error("Expected to find a {} configuration but found {}",name,config.getName());
32+
thrownewIllegalStateException(String.format("Expected to find a %s configuration but found %s",name,config.getName()));
33+
}
34+
returnDSL.using(config.getConfiguration());
35+
}
36+
37+
privatestaticvoidsetNamedConfiguration(Stringname,Configurationconfiguration) {
38+
configurations.set(newNamedConfiguration(name,configuration));
39+
log.debug("Set ThreadLocal configuration with name {}",name);
40+
}
41+
42+
publicstaticvoidthreadLocalTransaction(Stringname,DataSourceds,Runnablerunnable) {
43+
threadLocalTransactionResult(name,ds, () -> {runnable.run();returnnull;} );
44+
}
45+
46+
publicstatic <T>TthreadLocalTransactionResult(Stringname,DataSourceds,Supplier<T>supplier) {
47+
returnDSL.using(JooqConfig.defaultConfigFromDataSource(ds))
48+
.transactionResult(ctx -> {
49+
try {
50+
ThreadLocalJooqConfig.setNamedConfiguration(name,ctx);
51+
returnsupplier.get();
52+
}finally {
53+
configurations.remove();
54+
log.debug("Removed ThreadLocal configuration with name {}",name);
55+
}
56+
});
57+
}
58+
59+
privatestaticNamedConfigurationsafeGet() {
60+
NamedConfigurationconfig =configurations.get();
61+
Preconditions.checkNotNull(config,"No Configuration has been initialized on this thread.");
62+
log.debug("Found ThreadLocal configuration with name {}",config.getName());
63+
returnconfig;
64+
}
65+
}

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp