写几个简单点的shell脚本，配合IDEA，将oracle的建表语句轻松转换为java bean和mybatis的mapper配置

青旅半醒 2022-12-03 05:17 9阅读 0赞

# 背景 #

前段时间，遇到一个需求。需求很普通，就是oracle表的一些CRUD。

就是工作量有点大，给了几个sql脚本，里面表有六十多张，每个表一般二三十个字段或者四五十个字段。

用的是mybatis，这样的话，每张表要建一个对应的java bean类，一个对应的mapper文件，需要提供相关的查询和插入语句。

我想了想，硬敲的话，要疯。

mybatis有个逆向工具，我是听过的，只是没用过。不过我一看这些个建表语句都长一个样，就写了几个简单的脚本来转换为对应的bean或mapper。

最近学的东西有点深，不适合写博客，就分享一下我写的这个脚本，有兴趣也能用的上，因为这些sql脚本用pl/sql导出来的时候，基本也长这个样子。

# 脚本 #

打个比方，这是一个建表语句，实际表字段很多，我只写了几个字段作为示例，字段名就是这样命名，大小写字母以下划线连接

create table example_table  (
       user_name            VARCHAR2(200)                   not null,
       str1                 VARCHAR2(200)                   not null,
       str2                 VARCHAR2(200)                   not null,
       str3                 VARCHAR2(200),
       ONE_FLAG             VARCHAR2(1)                    default 'N' not null,
       time5                TIMESTAMP,
       str6                 VARCHAR2(1)                    default 'N' not null,
       reserved1            VARCHAR2(200),
       reserved2            VARCHAR2(200)
    );
    
    comment on table example_table is
    '样例表';
    
    comment on column example_table.user_name is
    '名';
    
    comment on column example_table.str1 is
    '这是一个字符串1';
    
    comment on column example_table.str2 is
    '这是一个字符串2';
    
    comment on column example_table.str3 is
    '这是一个字符串3';
    
    comment on column example_table.ONE_FLAG is
    '这是一个标记字符串';
    
    comment on column example_table.time5 is
    '时间';
    
    comment on column example_table.str6 is
    '又来一个字符串';
    
    comment on column example_table.reserved1 is
    '预留字段1';
    
    comment on column example_table.reserved2 is
    '预留字段2';

在下面要写的脚本所在目录下创建一个名字为txt的文件，把上面的建表语句复制进去。

## 生成java bean ##

1. 写一个awk脚本（ generate\_bean.awk）：把表字段转换为变量定义语法，并且加上注释

#! /usr/bin/awk -f
    
    BEGIN {
            i = 1;
    }
    {
            comment="";
            if ( NR != 1 && i > 0 && $1 != "(" ) {
                    if ( $1  == ");" ) {
                            i = -1;
                    } else {
                            if ( NF == 1 ) {
                                    print "// "$1;
    
                            }
                            else {
                                    "grep -A 1 comment "FILENAME" | grep -i  -A 1  "$1" | grep -i -v "$1 | getline comment;
                                    print "// "comment;
                                    ". to_lower_word.sh " $1 | getline lower_word;
                                    print "private String "lower_word";";
                            }
                    }
    
                    i++;
            }
    
    }

可以执行下看下效果：./generate\_bean.awk txt

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3g3NjM3OTUxNTE_size_16_color_FFFFFF_t_70][]

其中to\_lower\_word.sh，是将str1\_str2这种转换为小驼峰命名的变量，如：

#! /bin/bash
    
    echo $1 | tr [:upper:] [:lower:] | sed 's/_[a-z]\{1\}/\U&/g' | sed 's/_//g'

![20200901202415734.png][]

2. 再写一个脚本（gen\_ben.sh）调用generate\_bean.awk，并去掉注释上的杂质

#! /bin/bash
    
    sed -i 's/^$//g' txt;
    
    if [ ! -e tmp_txt ]; then
            touch tmp_txt;
    fi
    
    generate_bean.awk txt > tmp_txt;
    sed "s/';//g" tmp_txt | sed "s/'//g"
    
    rm tmp_txt

看一下执行效果：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3g3NjM3OTUxNTE_size_16_color_FFFFFF_t_70 1][]

3.打开IDEA，新建一个类，复制进去。

alt+insert，快速生成get和set还有toString等方法：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3g3NjM3OTUxNTE_size_16_color_FFFFFF_t_70 2][]

这里有一个问题，写脚本的时候为了赶时间，写的比较low，有些地方需要调整，比如这里都默认转换为String类型了，对于其它类型需要手动改一下。

## 生成mytatis的ResultMap ##

1.写一个awk脚本（generate\_result.awk）：将字段转换为小驼峰，然后拼接处理下

#! /usr/bin/awk -f
    
    
    BEGIN {
            i = 1;
            print "<resultMap type=\"\" id=\"\">";
    }
    {
            file = FILENAME;
            if ( NR != 1 && i > 0 && $1 != "(" ) {
                    if ( $1  == ");" ) {
                            i = -1;
                    } else {
                            if ( NF > 1 ) {
    
                                    ". to_lower_word.sh " $1 | getline lower_word;
                                    print "\t<result column=\""$1"\"\tproperty=\""lower_word"\"/>";
                            }
                    }
    
                    i++;
            }
    
    }
    END {
            print "</resultMap>"
    }

2.再写一个脚本（gen\_result.sh）调用这个脚本：

#! /bin/bash
    generate_result.awk txt

3.执行看下效果

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3g3NjM3OTUxNTE_size_16_color_FFFFFF_t_70 3][]

复制到mapper的xml文件里

## 生成select的sql ##

1.写一个awk脚本（generate\_fields.awk）将数据库表字段转换为小驼峰格式两列数据，第一行是表名：

#! /usr/bin/awk -f
    
    BEGIN {
            i = 1;
    }
    {
            if ( NR == 1 ) {
                    print $3;
            }
            if ( NR != 1 && i > 0 && $1 != "(" ) {
                    if ( $1  == ");" ) {
                            i = -1;
                    } else {
                            ". to_lower_word.sh " $1 | getline lower_word;
                            print $1,lower_word;
                    }
    
                    i++;
            }
    
    }

看一下效果：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3g3NjM3OTUxNTE_size_16_color_FFFFFF_t_70 4][]

2. 再写一个脚本（gen\_select.sh）将这拼接为一个select的sql

#! /bin/bash
    
    generate_fields.awk txt > fields_tmp.txt
    
    table_name=`head -1 fields_tmp.txt`
    lines=`wc -l fields_tmp.txt | awk '{print $1}'`
    awk -v total=$lines 'BEGIN {print "select";}{ if (NR > 1 && NR < total && NF > 1) {print "\t"$1",";}} END {print "\t"$1;}' fields_tmp.txt;
    
    echo "from $table_name"
    echo "<where>"
    
    awk '{if ( NR > 1 ) { print "\t""<if test=\""$2" != null and "$2" != ""\x27\x27""\">" "\n\t\t AND " $1" = #{" $2 "}" "\n\t</if>" }}' fields_tmp.txt;
    
    echo "</where>"
    
    rm fields_tmp.txt

3.执行下，复制到mapper.xml中,看下效果：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3g3NjM3OTUxNTE_size_16_color_FFFFFF_t_70 5][]

里面有部分字段，根据需要该删除删除，该调整调整

## 生成insert ##

1. 脚本（gen\_insert.sh）

#! /bin/bash
    
    generate_fields.awk txt > fields_tmp.txt
    
    table_name=`head -1 fields_tmp.txt`
    lines=`wc -l fields_tmp.txt | awk '{print $1}'`
    awk -v total=$lines -v table_name=$table_name 'BEGIN {print "insert into "table_name"(";}{ if (NR > 1 && NR < total && NF > 1) {print "\t"$1",";}} END {print "\t"$1")";}' fields_tmp.txt;
    
    echo "values("
    
    awk -v total=$lines '{if ( NR > 1 && NR < total ) { print "\t#{" $2 ", jdbcType=VARCHAR},"  }} END {print "\t#{"$1 ", jdbcType=VARCHAR}"}' fields_tmp.txt;
    
    echo ")"
    
    rm fields_tmp.txt

2. 执行，copy到mapper.xml中，效果如下：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3g3NjM3OTUxNTE_size_16_color_FFFFFF_t_70 6][]

# 后记 #

犹记得那天，我中午没睡用了2个小时才把这个脚本鼓捣出来 。

次日上午，以我疯狂的手速，用了2个小时，不到半天的时间，提交了上万行代码。

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3g3NjM3OTUxNTE_size_16_color_FFFFFF_t_70]: /images/20221123/896db5ad4f78467f837caca2d796fd3c.png
[20200901202415734.png]: /images/20221123/3d02eae7ab2648bfa5f7d7ca791b6298.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3g3NjM3OTUxNTE_size_16_color_FFFFFF_t_70 1]: /images/20221123/5fafb578110a45debd25e427a02cf3be.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3g3NjM3OTUxNTE_size_16_color_FFFFFF_t_70 2]: /images/20221123/c41e49e3250941a6af3d27ab5eaf7e9d.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3g3NjM3OTUxNTE_size_16_color_FFFFFF_t_70 3]: /images/20221123/a3a4aa23b0304950ac168932639a92f4.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3g3NjM3OTUxNTE_size_16_color_FFFFFF_t_70 4]: /images/20221123/c1e4786a995b41c9ac628260dc7d61f4.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3g3NjM3OTUxNTE_size_16_color_FFFFFF_t_70 5]: /images/20221123/1e11dc78c24b48d09b012b3db9f0b258.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3g3NjM3OTUxNTE_size_16_color_FFFFFF_t_70 6]: /images/20221123/265c5948ee844b5b835db6cbaa3e580c.png