lhs 4 gadi atpakaļ
vecāks
revīzija
73391c959d
3 mainītis faili ar 27 papildinājumiem un 5 dzēšanām
  1. 1 0
      errors/error.go
  2. 24 3
      repository/postgres/test.go
  3. 2 2
      service/test.go

+ 1 - 0
errors/error.go

@@ -28,6 +28,7 @@ const (
 	CodeParamErr            ErrorCode = 5000
 	CodeDATABASE            ErrorCode = 9000
 	CodeREDIS               ErrorCode = 10000
+	CodeUserErr             ErrorCode = 9001
 	CodeParam                         = ErrorCode(errors.CodeParam)
 )
 

+ 24 - 3
repository/postgres/test.go

@@ -3,6 +3,7 @@ package postgres
 import (
 	"encoding/json"
 	"fmt"
+	"gframe/errors"
 	"github.com/jackc/pgx/v4"
 )
 
@@ -10,17 +11,37 @@ type Test struct {
 	base
 }
 
-func (t Test) UserAdd(name string) {
+func (t Test) UserAdd2(name string) (err error) {
 	stCode := -999
 	sql := `select fn_user_add2($1,$2);`
-	_, err := t.writeDB().QueryFunc(t.ctx(), sql, []interface{}{name, stCode}, []interface{}{&stCode}, func(pgx.QueryFuncRow) error {
-		fmt.Printf("%v", stCode)
+	_, err = t.writeDB().QueryFunc(t.ctx(), sql, []interface{}{name, stCode}, []interface{}{&stCode}, func(pgx.QueryFuncRow) error {
+		//fmt.Printf("%v", stCode)
 		return nil
 	})
+
+	if stCode < 0 {
+		err = errors.NewServiceErr(errors.CodeUserErr, "用户添加错误:")
+	}
 	if err != nil {
 		fmt.Printf("QueryFunc error: %v", err)
 	}
+	return
+}
+func (t Test) UserAdd1(paramStr string) (err error) {
+	stCode := -999
+	sql := `select fn_user_add($1,$2);`
+	_, err = t.writeDB().QueryFunc(t.ctx(), sql, []interface{}{paramStr, stCode}, []interface{}{&stCode}, func(pgx.QueryFuncRow) error {
+		//fmt.Printf("%v", stCode)
+		return nil
+	})
 
+	if stCode < 0 {
+		err = errors.NewServiceErr(errors.CodeUserErr, "用户添加错误:")
+	}
+	if err != nil {
+		fmt.Printf("QueryFunc error: %v", err)
+	}
+	return
 }
 
 type UserListInfo struct {

+ 2 - 2
service/test.go

@@ -9,8 +9,8 @@ type Test struct {
 }
 
 func (t Test) UserAdd(name string) (err error) {
-
-	db.Test{}.UserAdd(name)
+	paramStr := "{\"name\":\"" + name + "\"}"
+	err = db.Test{}.UserAdd1(paramStr)
 	return
 }
 func (t Test) UserListQuery() (rs []interface{}, err error) {