more

Author: Bencheng21

pkg/mssqldb/roles.go

@@ -280,6 +280,9 @@ WHERE type = 'R' AND principal_id = @p1
 
 	var roleModel RoleModel
 	row := c.db.QueryRowxContext(ctx, query, id)
+	if err := row.Err(); err != nil {
+		return nil, err
+	}
 
 	err := row.StructScan(&roleModel)
 	if err != nil {
@@ -315,6 +318,9 @@ WHERE type = 'R' AND principal_id = @p1
 
 	var roleModel RoleModel
 	row := c.db.QueryRowxContext(ctx, query, id)
+	if err := row.Err(); err != nil {
+		return nil, err
+	}
 
 	err := row.StructScan(&roleModel)
 	if err != nil {

pkg/mssqldb/users.go

@@ -223,6 +223,9 @@ WHERE
 `
 
 	rows := c.db.QueryRowxContext(ctx, query, userId)
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
 
 	var userModel UserModel
 	err := rows.StructScan(&userModel)
@@ -307,6 +310,9 @@ AND sp.principal_id = @p1
 	query = fmt.Sprintf(query, db)
 
 	row := c.db.QueryRowxContext(ctx, query, principalId)
+	if err := row.Err(); err != nil {
+		return nil, err
+	}
 
 	var userModel UserDBModel
 	err := row.StructScan(&userModel)