arrays - Golang 从 postgresql 函数获取值，该函数返回一个 int 但也接收一个字符串数组

Question

我有一个 postgresql 函数，它基本上返回一个数字，而且正如您所看到的，该函数接收一个字符串数组。

Create Or Replace Function fnRegisterUserRoleArray(idUserFather int, rolesArray Text[]) returns int language plpgsql
as
$body$
declare ids INT;
declare roleID INT;
declare sanitazedRole TEXT;
declare counter int = 0;
begin
if(empty2null(idUserFather::text) is null) then
    ids := 0;
elsif exists( select 1 from win_users where id_user = idUserFather limit 1) then
    for counter in 1 .. array_upper(rolesArray, 1)
    loop
        select id_role from win_roles where rolename = rolesArray[counter] into roleID;
        insert into win_user_role(id_user, id_role) values (idUserFather, roleID);
        ids := ids + 1;
    end loop;
else
    ids := 0;
end if;
return ids;
end $body$;

在我的 Go 函数中，我有接收与 postgresql 数据库连接的变量数据库，

database, err := getConnection()

但是，当我调用函数 fnRegisterUserViewsPermission 并发送值时，我收到错误消息。

这是我设置值的方式:

var resultRole int 
err = database.QueryRow("Select fnRegisterUserRoleArray($1, $2);", resultUser, pq.Array(roleArray)).Scan(&resultRole, &int)
    fmt.Println(resultRole)
    if err != nil {
        fmt.Println(resultRole)
        return nil, err
    }
    if resultRole != 0 && resultRole != 1 {
        response = response + "their roles has been assigned correctly "
    } else {
            response = response + "however there was an error during the assignation of the role."
    }

我收到的输出是这样的:

"message": "sql: Scan error on column index 0: converting driver.Value type (\"\") to a int: invalid syntax",

但是这些值存储在我的数据库中，所以该函数以一种很好的方式接收这些值，但是返回是它去的地方 bam :(

这仅在我发送数组时发生，但如果我发送任何内容而不是数组，则扫描器返回从 postgresql 函数获得的 Id。

有没有什么办法既能获取结果id又能以更优雅的方式发送数组？

这是包含角色数组的值:

[Carrier Brand]

而包含resultUser的值只是一个数字，表示已在数据库中注册的用户的ID，在这种情况下，注册用户的ID为...

result user: 63

谢谢! :)

Answer 1

如错误所述:

"message": "sql: Scan error on column index 0: converting driver.Value type (\"\") to a int: invalid syntax",

错误是在从函数返回值时扫描来自函数的结果。

删除指向 int 基本类型的指针，因为 int 类型的函数只返回单个值。

将值扫描成int类型变量为

var resultRole int 
err = database.QueryRow("Select fnRegisterUserRoleArray($1, $2);", resultUser, pq.Array(roleArray)).Scan(&resultRole)

为了处理这些类型的情况，如果您将扫描结果与查询分开处理会更好:

sqlStatement := `Select fnRegisterUserRoleArray($1, $2);`
var resultRole int 
// Replace 3 with an ID from your database or another random
// value to test the no rows use case.
row := db.QueryRow(sqlStatement, resultUser, pq.Array(roleArray))
switch err := row.Scan(&resultRole); err {
case sql.ErrNoRows:
  fmt.Println("No rows were returned!")
case nil:
  fmt.Println(resultRole)
default:
  panic(err)
}

使用上述将 queryRow 与 Scan 分开的方法可以帮助您更多地分析结果和返回的错误。