diff --git a/model/line_group.go b/model/line_group.go index cc8a577..28666c4 100644 --- a/model/line_group.go +++ b/model/line_group.go @@ -1,6 +1,9 @@ package model -import "time" +import ( + "database/sql" + "time" +) // LineGroup - struct type LineGroup struct { @@ -42,6 +45,9 @@ func CheckGroupOwner(user, g string) (exists bool, err error) { func GetLineGroup(id string) (g *LineGroup, err error) { g = &LineGroup{} err = x.Get(g, `select * from "public"."line_group" where "id" = $1`, id) + if err == sql.ErrNoRows { + return nil, nil + } return } @@ -49,8 +55,8 @@ func GetLineGroup(id string) (g *LineGroup, err error) { func AddLineGroup(name, owner string, notify bool) (g *LineGroup, err error) { g = &LineGroup{} err = x.Get(g, `insert into "public"."line_group" ("name", "owner", "notify") values ($1, $2, $3)`, name, owner, notify) - if err != nil { - return nil, err + if err == sql.ErrNoRows { + return nil, nil } return } diff --git a/model/line_user.go b/model/line_user.go index 36dba05..fef4700 100644 --- a/model/line_user.go +++ b/model/line_user.go @@ -1,6 +1,9 @@ package model -import "time" +import ( + "database/sql" + "time" +) // LineUser - type LineUser struct { @@ -14,6 +17,9 @@ type LineUser struct { func GetLineUserByID(id string) (u *LineUser, err error) { u = &LineUser{} err = x.Get(u, `select * from "public"."line_user" where "id" = $1`, id) + if err == sql.ErrNoRows { + return nil, nil + } return }