采购单增删改查

zhangqian

2023-08-26 5193dcb9336e853502baf8a539d3f45efebe2f86

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
package system
 
import (
    "srm/global"
    "srm/model/system/response"
)
 
var AutoCodeMysql = new(autoCodeMysql)
 
type autoCodeMysql struct{}
 
// GetDB 获取数据库的所有数据库名
// Author [piexlmax](https://github.com/piexlmax)
// Author [SliverHorn](https://github.com/SliverHorn)
func (s *autoCodeMysql) GetDB(businessDB string) (data []response.Db, err error) {
    var entities []response.Db
    sql := "SELECT SCHEMA_NAME AS `database` FROM INFORMATION_SCHEMA.SCHEMATA;"
    if businessDB == "" {
        err = global.GVA_DB.Raw(sql).Scan(&entities).Error
    } else {
        err = global.GVA_DBList[businessDB].Raw(sql).Scan(&entities).Error
    }
    return entities, err
}
 
// GetTables 获取数据库的所有表名
// Author [piexlmax](https://github.com/piexlmax)
// Author [SliverHorn](https://github.com/SliverHorn)
func (s *autoCodeMysql) GetTables(businessDB string, dbName string) (data []response.Table, err error) {
    var entities []response.Table
    sql := `select table_name as table_name from information_schema.tables where table_schema = ?`
    if businessDB == "" {
        err = global.GVA_DB.Raw(sql, dbName).Scan(&entities).Error
    } else {
        err = global.GVA_DBList[businessDB].Raw(sql, dbName).Scan(&entities).Error
    }
 
    return entities, err
}
 
// GetColumn 获取指定数据库和指定数据表的所有字段名,类型值等
// Author [piexlmax](https://github.com/piexlmax)
// Author [SliverHorn](https://github.com/SliverHorn)
func (s *autoCodeMysql) GetColumn(businessDB string, tableName string, dbName string) (data []response.Column, err error) {
    var entities []response.Column
    sql := `
    SELECT COLUMN_NAME        column_name,
       DATA_TYPE          data_type,
       CASE DATA_TYPE
           WHEN 'longtext' THEN c.CHARACTER_MAXIMUM_LENGTH
           WHEN 'varchar' THEN c.CHARACTER_MAXIMUM_LENGTH
           WHEN 'double' THEN CONCAT_WS(',', c.NUMERIC_PRECISION, c.NUMERIC_SCALE)
           WHEN 'decimal' THEN CONCAT_WS(',', c.NUMERIC_PRECISION, c.NUMERIC_SCALE)
           WHEN 'int' THEN c.NUMERIC_PRECISION
           WHEN 'bigint' THEN c.NUMERIC_PRECISION
           ELSE '' END AS data_type_long,
       COLUMN_COMMENT     column_comment
    FROM INFORMATION_SCHEMA.COLUMNS c
    WHERE table_name = ?
      AND table_schema = ?
    `
    if businessDB == "" {
        err = global.GVA_DB.Raw(sql, tableName, dbName).Scan(&entities).Error
    } else {
        err = global.GVA_DBList[businessDB].Raw(sql, tableName, dbName).Scan(&entities).Error
    }
 
    return entities, err
}