Files
paralus/server/user.go
nirav-rafay c66bdc25cd restructure rcloud-base as a single base controller (#37)
* restructure rcloud-base as a single base controller
* updated master.rest
* moved sentry from internal to pkg as it is used by relay
* removing unused rpc and it's dependencies
* Fix usermgmt tests
* Don't redefine variables in rest file
Co-authored-by: Abin Simon <abin.simon@rafay.co>
2022-03-03 17:59:06 +05:30

56 lines
1.7 KiB
Go

package server
import (
"context"
"github.com/RafaySystems/rcloud-base/pkg/service"
rpcv3 "github.com/RafaySystems/rcloud-base/proto/rpc/user"
v3 "github.com/RafaySystems/rcloud-base/proto/types/commonpb/v3"
userpbv3 "github.com/RafaySystems/rcloud-base/proto/types/userpb/v3"
"google.golang.org/protobuf/types/known/timestamppb"
)
type userServer struct {
service.UserService
}
// NewUserServer returns new user server implementation
func NewUserServer(ps service.UserService) rpcv3.UserServer {
return &userServer{ps}
}
func updateUserStatus(req *userpbv3.User, resp *userpbv3.User, err error) *userpbv3.User {
if err != nil {
req.Status = &v3.Status{
ConditionStatus: v3.ConditionStatus_StatusFailed,
LastUpdated: timestamppb.Now(),
Reason: err.Error(),
}
return req
}
resp.Status = &v3.Status{ConditionStatus: v3.ConditionStatus_StatusOK}
return resp
}
func (s *userServer) CreateUser(ctx context.Context, req *userpbv3.User) (*userpbv3.User, error) {
resp, err := s.Create(ctx, req)
return updateUserStatus(req, resp, err), err
}
func (s *userServer) GetUsers(ctx context.Context, req *userpbv3.User) (*userpbv3.UserList, error) {
return s.List(ctx, req)
}
func (s *userServer) GetUser(ctx context.Context, req *userpbv3.User) (*userpbv3.User, error) {
resp, err := s.GetByName(ctx, req)
return updateUserStatus(req, resp, err), err
}
func (s *userServer) DeleteUser(ctx context.Context, req *userpbv3.User) (*rpcv3.DeleteUserResponse, error) {
return s.Delete(ctx, req)
}
func (s *userServer) UpdateUser(ctx context.Context, req *userpbv3.User) (*userpbv3.User, error) {
resp, err := s.Update(ctx, req)
return updateUserStatus(req, resp, err), err
}