lijianbin
1 week ago
10 changed files with 816 additions and 134 deletions
Split View
Diff Options
-
+120 -15chain.go
-
+134 -7chain_test.go
-
+116 -7common.go
-
+143 -54conn.go
-
+70 -19db.go
-
+52 -1go.mod
-
+15 -0go.sum
-
+36 -12prepare.go
-
+19 -7transaction.go
-
+111 -12transaction_chain.go
+ 120
- 15
chain.go
View File
+ 134
- 7
chain_test.go
View File
| @ -1,17 +1,144 @@ | |||
| package dbquery | |||
| import ( | |||
| "fmt" | |||
| "testing" | |||
| ) | |||
| // 测试各数据库下各种情况 | |||
| func Test_Chain(t *testing.T) { | |||
| Connect("127.0.0.1", "root", "123456", "shop", "3306") | |||
| //测试数据库连接 | |||
| //err := Connect("127.0.0.1", "root", "root", "canyin", "3306") | |||
| err := PgConnect("192.168.233.145", "bin", "Bin123456", "canyin", "5432") | |||
| //err := DmConnect("192.168.233.141", "SHOPV2", "Bin123456", "", "5236") | |||
| if err != nil { | |||
| t.Log(err) | |||
| } | |||
| db_name := "" | |||
| table_name := "ttl_user_log" | |||
| //time := time.Now().Unix() | |||
| ret, err := new(Query).Db("shop").Table("ttl_order_product").Alias("op"). | |||
| Join([]string{"ttl_product as p", "op.product_id=p.id"}). | |||
| Title("op.id,op.order_price,p.thumb").WhereOr("op.id =?").WhereOr("op.id = ?").Value(63).Value(64).Debug(true).List() | |||
| //================查询表结构=========== | |||
| ret, err := new(Query).Db(db_name).GetTableInfo(table_name) | |||
| if err != nil { | |||
| t.Log(err) | |||
| } | |||
| fmt.Println("===GetTableInfo:", ret) | |||
| t.Log(len(ret)) | |||
| t.Log(ret) | |||
| t.Log(err) | |||
| //==========获取信息================= | |||
| //info, err := new(Query).Db(db_name).Clean().Table(table_name).Where("id=?").Value("3").Title("*").Find() | |||
| //info, err := new(Query).Db(db_name).Clean().Table(table_name).Clean().Alias("user").Join([]string{"ttl_user u", "u.id = user.user_id", "inner"}).Where("user.id=?").Value("3").Title("user.id,user.user_id,u.nickname").Find() | |||
| info, err := GetDataByStmt(db_name, table_name, "*", []string{"id = ?"}, []interface{}{3}, nil) | |||
| if err != nil { | |||
| t.Log(err) | |||
| } | |||
| fmt.Println("===Find:", info) | |||
| //============获取列表================== | |||
| list, err := new(Query).Db(db_name).Clean().Table(table_name).Alias("u").Join([]string{"ttl_user user", "user.id = u.user_id", "inner"}).Where("user.id=?").Value(1).Title("user.id,user.nickname,user.status").Orderby("user.id desc").Page(1).PageSize(10).Select() | |||
| //list, err := DoQuery("select user.nickname,log.* from ttl_user user left join ttl_user_log log on user.id = log.user_id where user.id = ?", "1") | |||
| //list, err := GetListByStmt(db_name, table_name, "*", []string{"id = ?"}, []interface{}{3}, map[string]string{"id": "id asc"}) | |||
| //list, err := QueryByStmt("select * from "+table_name+" where id < ?", []interface{}{10}) | |||
| if err != nil { | |||
| t.Log(err) | |||
| } | |||
| fmt.Println("===List:", list) | |||
| //===========添加数据============ | |||
| //insert_res, err := new(Query).Db(db_name).Clean().Table(table_name).Data("user_id=?").Value("1").Data("createtime=?").Value(time).Create() | |||
| //insert_res, err := InsertByStmt(db_name, table_name, []string{"user_id=?", "createtime=?"}, []interface{}{"1", time}) | |||
| //insert_res, err := Insert(db_name, table_name, map[string]string{"user_id": "1", "createtime": helper.ToStr(time)}) | |||
| //if err != nil { | |||
| // t.Log(err) | |||
| //} | |||
| //fmt.Println("===Insert:", insert_res) | |||
| //================更新数据===================== | |||
| //update_res, err := new(Query).Db(db_name).Clean().Table(table_name).Data("user_id=?").Value("2").Data("createtime=?").Value(time).Where("id=?").Value("6").Update() | |||
| //update_res, err := UpdateByStmt(db_name, table_name, []string{"createtime=?", "user_id=?"}, []string{"id=?"}, []interface{}{time, 3, 6}) | |||
| //if err != nil { | |||
| // t.Log(err) | |||
| //} | |||
| //fmt.Println("===Update:", update_res) | |||
| //=============事务================ | |||
| /*fmt.Println("================开启事务============") | |||
| tx, err := DB.Begin() | |||
| if err != nil { | |||
| t.Log(err) | |||
| } | |||
| update_log, err := TxPreUpdate(tx, db_name, table_name, []string{"createtime= ?"}, []string{"id=?"}, []interface{}{time, 2}) | |||
| if err != nil { | |||
| tx.Rollback() | |||
| t.Log(err) | |||
| } | |||
| fmt.Println("===========事务执行:==================") | |||
| fmt.Println("===事务update：", update_log) | |||
| insert_log, err := TxPreInsert(tx, db_name, table_name, map[string]interface{}{"user_id": "1", "createtime": helper.ToStr(time)}) | |||
| if err != nil { | |||
| tx.Rollback() | |||
| t.Log(err) | |||
| } | |||
| fmt.Println("===事务insert：", insert_log) | |||
| del_log, err := TxDelete(tx, db_name, table_name, map[string]string{"id": "2"}) | |||
| if err != nil { | |||
| tx.Rollback() | |||
| t.Log(err) | |||
| } | |||
| fmt.Println("====事务delete:", del_log) | |||
| err = tx.Commit() | |||
| if err != nil { | |||
| t.Log(err) | |||
| tx.Rollback() | |||
| } | |||
| fmt.Println("=======事务执行完成==========")*/ | |||
| /*fmt.Println("====================执行事务trans============") | |||
| trans := NewTxQuery().Db(db_name) | |||
| info_trans, err := trans.Clean().Table(table_name).Where("id = ?").Value(5).Find() | |||
| if err != nil { | |||
| trans.Rollback() | |||
| t.Log(err) | |||
| } | |||
| fmt.Println("===事务Find_trans:", info_trans) | |||
| list_trans, err := trans.Clean().Table(table_name).Title("*").Select() | |||
| if err != nil { | |||
| trans.Rollback() | |||
| t.Log(err) | |||
| } | |||
| fmt.Println("=========事务List_trans:", list_trans) | |||
| data := map[string]interface{}{ | |||
| "user_id": 5, | |||
| "memo": "test", | |||
| "createtime": time, | |||
| } | |||
| add_trans, err := trans.Clean().Table(table_name).SaveData(data).CreateAll() | |||
| if err != nil { | |||
| trans.Rollback() | |||
| t.Log(err) | |||
| } | |||
| fmt.Println("======事务Add_trans:", add_trans) | |||
| data["id"] = 15 | |||
| update_trans_res, err := trans.Clean().Table(table_name).SaveData(data).UpdateAll() | |||
| if err != nil { | |||
| trans.Rollback() | |||
| t.Log(err) | |||
| } | |||
| fmt.Println("=======事务update_trans", update_trans_res) | |||
| err = trans.Commit() | |||
| if err != nil { | |||
| trans.Rollback() | |||
| t.Log(err) | |||
| } | |||
| fmt.Println("====================执行事务结束==================")*/ | |||
| } | |||
+ 116
- 7
common.go
View File
| @ -1,21 +1,130 @@ | |||
| package dbquery | |||
| import ( | |||
| "fmt" | |||
| "git.tetele.net/tgo/helper" | |||
| "log" | |||
| "reflect" | |||
| "regexp" | |||
| "strconv" | |||
| "strings" | |||
| "time" | |||
| ) | |||
| // 对执行前的sql语句进行处理--针对pgsql用 | |||
| func SqlReplace(sql_s, sql_type string) string { | |||
| sql := strings.Replace(sql_s, "`", `"`, -1) | |||
| // ===================达梦兼容=============== | |||
| // 非关键字可以不添加标识符，关键字须添加 | |||
| // 日期函数的使用TO_CHAR(TO_DATE('1970-01-01','yyyy-mm-dd') + (createtime / 86400), 'yyyy-mm-dd') | |||
| // group_concat替换成LISTAGG | |||
| // ======================================== | |||
| if sql_type == "add" { | |||
| // 关键字替换-支持达梦和高斯 | |||
| func ReplaeByOtherSql(sql, sql_type, action string) string { | |||
| sql_type_arr := []string{"DmSql", "PgsqlDb"} | |||
| if !helper.IsInStringArray(sql_type_arr, sql_type) { | |||
| log.Println("sql_type error", sql_type) | |||
| return "" | |||
| } | |||
| // PgsqlDb用 | |||
| if action == "add" { | |||
| sql = helper.StringJoin(sql, " RETURNING id") | |||
| } | |||
| // 定义需要处理的关键字列表 | |||
| keywords := []string{"user", "order", "group", "table", "view", "admin", "new"} | |||
| if sql_type == "PgsqlDb" { | |||
| keywords = []string{"user", "order", "group"} | |||
| } | |||
| // 移除所有反引号 | |||
| sql = strings.ReplaceAll(sql, "`", "") | |||
| // 使用单词边界 \b 确保只匹配完整单词 | |||
| pattern := `\b(` + strings.Join(keywords, "|") + `)\b` | |||
| re := regexp.MustCompile(pattern) | |||
| //设置保护词组 | |||
| excludePhrases := []string{ | |||
| "order by", "group by", | |||
| } | |||
| //保护排除短语 | |||
| phraseMap := make(map[string]string) | |||
| for i, phrase := range excludePhrases { | |||
| placeholder := fmt.Sprintf("__EXCLUDE_%d__", i) | |||
| phraseMap[placeholder] = phrase | |||
| sql = strings.Replace(sql, phrase, placeholder, -1) | |||
| } | |||
| sql = strings.Replace(sql, " user ", ` "user" `, -1) | |||
| sql = strings.Replace(sql, " user.", ` "user".`, -1) | |||
| sql = strings.Replace(sql, "=user.", `="user".`, -1) | |||
| // 执行替换 | |||
| sql = re.ReplaceAllStringFunc(sql, func(match string) string { | |||
| // 检查匹配是否在字符串常量中 | |||
| if isInStringLiteral(sql, match) { | |||
| return match | |||
| } | |||
| if sql_type == "DmSql" { | |||
| return "`" + match + "`" | |||
| } else { | |||
| return `"` + match + `"` | |||
| } | |||
| }) | |||
| // 恢复排除短语 | |||
| for placeholder, phrase := range phraseMap { | |||
| sql = strings.Replace(sql, placeholder, phrase, -1) | |||
| } | |||
| return sql | |||
| } | |||
| // 检查匹配是否在字符串常量中 | |||
| func isInStringLiteral(sql, match string) bool { | |||
| index := strings.Index(sql, match) | |||
| if index == -1 { | |||
| return false | |||
| } | |||
| // 检查匹配前的单引号数量 | |||
| prefix := sql[:index] | |||
| singleQuotes := strings.Count(prefix, "'") - strings.Count(prefix, "\\'") | |||
| // 奇数表示在字符串常量中 | |||
| return singleQuotes%2 != 0 | |||
| } | |||
| // 字段值类型转换--针对达梦用 | |||
| func ToString(value interface{}) string { | |||
| switch v := value.(type) { | |||
| case string: | |||
| return v | |||
| case []byte: | |||
| return string(v) | |||
| case int, int8, int16, int32, int64: | |||
| return strconv.FormatInt(reflect.ValueOf(value).Int(), 10) | |||
| case uint, uint8, uint16, uint32, uint64: | |||
| return strconv.FormatUint(reflect.ValueOf(value).Uint(), 10) | |||
| case float32, float64: | |||
| return strconv.FormatFloat(reflect.ValueOf(value).Float(), 'f', -1, 64) | |||
| case bool: | |||
| return strconv.FormatBool(v) | |||
| case time.Time: | |||
| return v.Format("2006-01-02 15:04:05") | |||
| default: | |||
| return fmt.Sprintf("%v", v) | |||
| } | |||
| } | |||
| // 字符串切片元素追加前缀 | |||
| func addPrefixInField(slice []string, prefix string) []string { | |||
| new_slice := make([]string, len(slice)) | |||
| for i, v := range slice { | |||
| new_slice[i] = prefix + v | |||
| } | |||
| return new_slice | |||
| } | |||
| func DmFieldDeal(fields string) string { | |||
| //移除所有反引号 | |||
| title := strings.Replace(fields, "`", "", -1) | |||
| return title | |||
| } | |||