opt code

Author: zmisgod

src/components/SqlToModel.vue

@@ -1,12 +1,13 @@
 <template>
   <div class="json-container">
     <div class="json-text">
-      <el-input type="textarea" rows=2 placeholder="请输入sql创建语句"
+      <el-input type="textarea" rows=4 placeholder="请输入sql创建语句"
       @change="inputJson" v-model="jsonText">
       </el-input>
     </div>
     <div class="show-go" v-if="goStruct !== ''">
-      <el-input type="textarea" rows=8 placeholder="显示的GoStruct" v-model="goStruct"></el-input>
+      <el-input type="textarea" autosize="true" placeholder="显示的GoStruct"
+      v-model="goStruct"></el-input>
     </div>
   </div>
 </template>
@@ -23,7 +24,7 @@ export default {
   },
   methods: {
     inputJson() {
-      sqlStrToGoTemplate(this.jsonText);
+      this.goStruct = sqlStrToGoTemplate(this.jsonText);
     },
   },
 };

src/utils/go/common.js

@@ -1,3 +1,77 @@
+function ucFirst(val) {
+  return val.charAt(0).toUpperCase() + val.slice(1);
+}
+
+function dbNameToCamelCase(name) {
+  const row = [];
+  const re = name.split('_');
+  re.map((val) => {
+    row.push(ucFirst(val));
+  });
+  return row.join('');
+}
+
+function containStr(name, val) {
+  if (name.indexOf(val) >= 0) {
+    return true;
+  }
+  return false;
+}
+
+function sqlTypeToGoType(type) {
+  let isContainUsign = false;
+  if (containStr(type, 'unsigned')) {
+    isContainUsign = true;
+  }
+  if (containStr(type, 'varchar')) {
+    return 'string';
+  }
+  if (containStr(type, 'int')) {
+    if (!isContainUsign) {
+      return 'int64';
+    }
+    return 'uint64';
+  }
+  return 'unknown';
+}
+
+function sqlToModel(data) {
+  const dbName = data.tableName;
+  const dbNameStr = dbNameToCamelCase(dbName);
+  const packageName = 'db';
+  const importPackages = ['context',
+    'github.com/jinzhu/gorm',
+    'pkg.poizon.com/golang/go-common/mysql'];
+  let importPackageStr = '';
+  importPackages.map((val) => {
+    importPackageStr += `"${val}"\n`;
+  });
+  let columsStr = '';
+  data.columns.map((val) => {
+    let defaultStr = '';
+    if (val.default !== undefined && val.default !== 'NULL') {
+      defaultStr = `;default:${val.default}`;
+    }
+    columsStr += `  ${dbNameToCamelCase(val.name)} ${sqlTypeToGoType(val.type)} \`gorm:"column:${val.name}${defaultStr}" json:"${val.name}"\`\n`;
+  });
+  const resultStr = `
+package ${packageName}
+import (
+${importPackageStr}
+)
+var New${dbNameStr} ${dbNameStr}Model
+type ${dbNameStr}Model struct {
+${columsStr}
+}
+func (*${dbNameStr}Model) TableName() string {
+  return "${dbName}"
+}
+`;
+  console.log(resultStr);
+  return resultStr;
+}
+
+// 解析sql表明等基础信息
 function sqlStrToGoTemplate(str) {
   // CREATE TABLE `(.*)`\s\(([\d\D]*)\)\sENGINE=(.*)\sAUTO_INCREMENT=(\d+)\s(\w+)\sCHARSET=(\w+)\s(COLLATE=\w+)*
   const regex = /CREATE TABLE `(.*)`\s\(([\d\D]*)\)\sENGINE=(.*)\sAUTO_INCREMENT=(\d+)\s(\w+)\sCHARSET=(\w+)\s(COLLATE=\w+)*/gm;
@@ -19,14 +93,16 @@ function sqlStrToGoTemplate(str) {
   return sqlStrColumnToEachColumn(data);
 }
 
+// 解析sql每个字段
 function sqlStrColumnToEachColumn(data) {
   const regex = /`(.*)`\s(\w+[\((\d+)\)]*[\sunsigned\s]*)([NOT\s[NULL\s]*]*)(DEFAULT\s([\'](.*?)[\']|NULL)\s)*(AUTO_INCREMENT\s)*(COMMENT\s[\'](.*)[\'])*/gm;
-  console.log(data);
-  const tableName = data[1];
+  const tableInfo = {
+    tableName: data[1],
+    engine: data[3],
+    increasement: data[4],
+    dafaultCharset: data[6],
+  };
   const str = data[2];
-  const engine = data[3];
-  const incrasement = data[4];
-  const dafaultCharset = data[6];
   const columns = [];
   let m;
   while ((m = regex.exec(str)) !== null) {
@@ -41,10 +117,20 @@ function sqlStrColumnToEachColumn(data) {
     });
     columns.push(one);
   }
-  console.log(columns);
+  const columnData = [];
+  columns.map((column) => {
+    columnData.push({
+      name: column[1], type: column[2], default: column[6], comment: column[9],
+    });
+  });
+  tableInfo.columns = columnData;
+  return sqlToModel(tableInfo);
 }
 
 export {
   sqlStrToGoTemplate,
   sqlStrColumnToEachColumn,
+  sqlToModel,
+  ucFirst,
+  sqlTypeToGoType,
 };