Skip to content
GitLab
Projects
Groups
Snippets
Help
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
S
schbrain-parent
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Service Desk
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Operations
Operations
Incidents
Environments
Packages & Registries
Packages & Registries
Package Registry
Analytics
Analytics
CI / CD
Repository
Value Stream
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
panwangnan
schbrain-parent
Commits
71348641
Commit
71348641
authored
May 31, 2023
by
liaozan
🏀
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Refine bizId column
parent
f202e5d2
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
15 additions
and
7 deletions
+15
-7
starters/mybatis-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/mybatis/biz/BizIdHelper.java
...rain/framework/autoconfigure/mybatis/biz/BizIdHelper.java
+15
-7
No files found.
starters/mybatis-spring-boot-starter/src/main/java/com/schbrain/framework/autoconfigure/mybatis/biz/BizIdHelper.java
View file @
71348641
package
com.schbrain.framework.autoconfigure.mybatis.biz
;
import
com.baomidou.mybatisplus.core.metadata.TableInfo
;
import
com.baomidou.mybatisplus.core.metadata.TableInfoHelper
;
import
com.baomidou.mybatisplus.core.metadata.*
;
import
com.baomidou.mybatisplus.core.toolkit.StringUtils
;
import
com.schbrain.common.exception.BaseException
;
import
com.schbrain.framework.autoconfigure.mybatis.annotation.BizId
;
import
org.apache.
ibatis.session.Configuration
;
import
org.apache.
commons.collections4.CollectionUtils
;
import
java.lang.reflect.Field
;
import
java.util.List
;
import
java.util.Map
;
import
java.util.concurrent.ConcurrentHashMap
;
import
java.util.stream.Collectors
;
/**
* @author liaozan
...
...
@@ -37,11 +39,17 @@ public class BizIdHelper {
return
annotation
.
value
();
}
TableInfo
tableInfo
=
TableInfoHelper
.
getTableInfo
(
entityClass
);
Configuration
configuration
=
tableInfo
.
getConfiguration
();
if
(
configuration
.
isMapUnderscoreToCamelCase
())
{
return
StringUtils
.
camelToUnderline
(
bizIdField
.
getName
());
return
getBizIdFieldInfo
(
tableInfo
,
bizIdField
).
getColumn
();
}
private
static
TableFieldInfo
getBizIdFieldInfo
(
TableInfo
tableInfo
,
Field
bizIdField
)
{
List
<
TableFieldInfo
>
fieldInfoList
=
tableInfo
.
getFieldList
().
stream
()
.
filter
(
fieldInfo
->
fieldInfo
.
getField
()
==
bizIdField
)
.
collect
(
Collectors
.
toList
());
if
(
CollectionUtils
.
isEmpty
(
fieldInfoList
))
{
throw
new
BaseException
(
String
.
format
(
"%s can not be found in fieldList"
,
bizIdField
.
getName
()));
}
return
bizIdField
.
getName
(
);
return
fieldInfoList
.
get
(
0
);
}
}
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment