Impl roles service server

Author: gedorinku

app/di/store_component.go

@@ -15,6 +15,7 @@ import (
 	feedstore "github.com/ProgrammingLab/prolab-accounts/infra/store/feed"
 	heartbeatstore "github.com/ProgrammingLab/prolab-accounts/infra/store/heartbeat"
 	profilestore "github.com/ProgrammingLab/prolab-accounts/infra/store/profile"
+	rolestore "github.com/ProgrammingLab/prolab-accounts/infra/store/role"
 	sessionstore "github.com/ProgrammingLab/prolab-accounts/infra/store/session"
 	userstore "github.com/ProgrammingLab/prolab-accounts/infra/store/user"
 	userblogstore "github.com/ProgrammingLab/prolab-accounts/infra/store/user_blog"
@@ -28,6 +29,7 @@ type StoreComponent interface {
 	UserBlogStore(ctx context.Context) store.UserBlogStore
 	FeedStore(ctx context.Context) store.FeedStore
 	EntryStore(ctx context.Context) store.EntryStore
+	RoleStore(ctx context.Context) store.RoleStore
 	HeartbeatStore(ctx context.Context) store.HeartbeatStore
 }
 
@@ -154,6 +156,10 @@ func (s *storeComponentImpl) EntryStore(ctx context.Context) store.EntryStore {
 	return entrystore.NewEntryStore(ctx, s.db)
 }
 
+func (s *storeComponentImpl) RoleStore(ctx context.Context) store.RoleStore {
+	return rolestore.NewRoleStore(ctx, s.db)
+}
+
 func (s *storeComponentImpl) HeartbeatStore(ctx context.Context) store.HeartbeatStore {
 	return heartbeatstore.NewHeartbeatStore(ctx, s.client, s.cfg)
 }

app/run.go

@@ -57,6 +57,7 @@ func Run() error {
 			server.NewUserBlogServiceServer(store),
 			server.NewEntryServiceServer(store, cfg),
 			server.NewPingServiceServer(store),
+			server.NewRoleServiceServer(store),
 		),
 	)
 

app/server/roles_server.go

@@ -2,13 +2,15 @@ package server
 
 import (
 	"context"
+	"database/sql"
 
-	"github.com/golang/protobuf/ptypes/empty"
+	"github.com/ProgrammingLab/prolab-accounts/app/util"
 	"github.com/izumin5210/grapi/pkg/grapiserver"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/status"
+	"github.com/pkg/errors"
 
 	api_pb "github.com/ProgrammingLab/prolab-accounts/api"
+	"github.com/ProgrammingLab/prolab-accounts/app/di"
+	"github.com/ProgrammingLab/prolab-accounts/infra/record"
 )
 
 // RoleServiceServer is a composite interface of api_pb.RoleServiceServer and grapiserver.Server.
@@ -18,34 +20,60 @@ type RoleServiceServer interface {
 }
 
 // NewRoleServiceServer creates a new RoleServiceServer instance.
-func NewRoleServiceServer() RoleServiceServer {
-	return &roleServiceServerImpl{}
+func NewRoleServiceServer(store di.StoreComponent) RoleServiceServer {
+	return &roleServiceServerImpl{
+		StoreComponent: store,
+	}
 }
 
 type roleServiceServerImpl struct {
+	di.StoreComponent
 }
 
 func (s *roleServiceServerImpl) ListRoles(ctx context.Context, req *api_pb.ListRolesRequest) (*api_pb.ListRolesResponse, error) {
-	// TODO: Not yet implemented.
-	return nil, status.Error(codes.Unimplemented, "TODO: You should implement it!")
+	rs := s.RoleStore(ctx)
+	roles, err := rs.ListRoles()
+	if err != nil {
+		if errors.Cause(err) == sql.ErrNoRows {
+			return &api_pb.ListRolesResponse{
+				Roles: rolesToResponse(nil),
+			}, nil
+		}
+
+		return nil, err
+	}
+
+	return &api_pb.ListRolesResponse{
+		Roles: rolesToResponse(roles),
+	}, nil
 }
 
 func (s *roleServiceServerImpl) GetRole(ctx context.Context, req *api_pb.GetRoleRequest) (*api_pb.Role, error) {
-	// TODO: Not yet implemented.
-	return nil, status.Error(codes.Unimplemented, "TODO: You should implement it!")
-}
+	rs := s.RoleStore(ctx)
+	r, err := rs.GetRole(int64(req.GetRoleId()))
+	if err != nil {
+		if errors.Cause(err) == sql.ErrNoRows {
+			return nil, util.ErrNotFound
+		}
 
-func (s *roleServiceServerImpl) CreateRole(ctx context.Context, req *api_pb.CreateRoleRequest) (*api_pb.Role, error) {
-	// TODO: Not yet implemented.
-	return nil, status.Error(codes.Unimplemented, "TODO: You should implement it!")
+		return nil, err
+	}
+
+	return roleToResponse(r), nil
 }
 
-func (s *roleServiceServerImpl) UpdateRole(ctx context.Context, req *api_pb.UpdateRoleRequest) (*api_pb.Role, error) {
-	// TODO: Not yet implemented.
-	return nil, status.Error(codes.Unimplemented, "TODO: You should implement it!")
+func rolesToResponse(roles []*record.Role) []*api_pb.Role {
+	resp := make([]*api_pb.Role, 0, len(roles))
+	for _, r := range roles {
+		resp = append(resp, roleToResponse(r))
+	}
+
+	return resp
 }
 
-func (s *roleServiceServerImpl) DeleteRole(ctx context.Context, req *api_pb.DeleteRoleRequest) (*empty.Empty, error) {
-	// TODO: Not yet implemented.
-	return nil, status.Error(codes.Unimplemented, "TODO: You should implement it!")
+func roleToResponse(role *record.Role) *api_pb.Role {
+	return &api_pb.Role{
+		RoleId: uint32(role.ID),
+		Name:   role.Name.String,
+	}
 }

app/server/users_server.go

@@ -220,7 +220,7 @@ func userToResponse(user *record.User, includePrivate bool, cfg *config.Config)
 
 		if r := p.R; p.R != nil {
 			if role := r.Role; role != nil {
-				u.Role = role.Name.String
+				u.Role = roleToResponse(role)
 			}
 			if dep := r.Department; dep != nil {
 				u.Department = &type_pb.Department{