u

code/allapp/allapp-go/db/repo/space_member.sql.go

@@ -9,6 +9,19 @@ import (
 	"context"
 )
 
+const countUserSpaces = `-- name: CountUserSpaces :one
+SELECT COUNT(1)
+FROM space_member
+WHERE user_id = $1
+`
+
+func (q *Queries) CountUserSpaces(ctx context.Context, userID int64) (int64, error) {
+	row := q.db.QueryRow(ctx, countUserSpaces, userID)
+	var count int64
+	err := row.Scan(&count)
+	return count, err
+}
+
 const getUserSpaceRole = `-- name: GetUserSpaceRole :one
 SELECT role
 FROM space_member
@@ -28,3 +41,27 @@ func (q *Queries) GetUserSpaceRole(ctx context.Context, arg GetUserSpaceRolePara
 	err := row.Scan(&role)
 	return role, err
 }
+
+const insertSpaceMember = `-- name: InsertSpaceMember :exec
+INSERT INTO space_member(id,space_id,user_id,role,is_default,join_time,update_time)
+VALUES ($1,$2,$3,$4,$5,NOW(),NOW())
+`
+
+type InsertSpaceMemberParams struct {
+	ID        int64
+	SpaceID   int64
+	UserID    int64
+	Role      int16
+	IsDefault int16
+}
+
+func (q *Queries) InsertSpaceMember(ctx context.Context, arg InsertSpaceMemberParams) error {
+	_, err := q.db.Exec(ctx, insertSpaceMember,
+		arg.ID,
+		arg.SpaceID,
+		arg.UserID,
+		arg.Role,
+		arg.IsDefault,
+	)
+	return err
+}