package BasePositionService

import (
	"context"
	"dsBaseWeb/Business/BasePosition/BasePositionProto"
	"dsBaseWeb/Utils/GRpcUtil"
	"dsBaseWeb/Utils/LogUtil"
	"errors"
	"time"
)

func GetBasePosition(modelArg BasePositionProto.ModelArg) (*BasePositionProto.Reply, error) {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()
	conn := GRpcUtil.GetConnection()
	if conn == nil {
		return nil, errors.New("RPC服务未启动！")
	}

	c := BasePositionProto.NewBasePositionManageClient(conn)
	Reply, err := c.GetBasePosition(ctx, &modelArg)
	if err != nil {
		LogUtil.Error("获取职务信息错误: ", err.Error())
	}
	return Reply, err
}

func AddBasePosition(modelArg BasePositionProto.ModelArg) (*BasePositionProto.Reply, error) {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()
	conn := GRpcUtil.GetConnection()
	if conn == nil {
		return nil, errors.New("RPC服务未启动！")
	}

	c := BasePositionProto.NewBasePositionManageClient(conn)
	Reply, err := c.AddBasePosition(ctx, &modelArg)
	if err != nil {
		LogUtil.Error("增加职务信息错误: ", err.Error())
	}
	return Reply, err
}

func DeleteBasePosition(deleteIdsArg BasePositionProto.DeleteIdsArg) (*BasePositionProto.Reply, error) {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()
	conn := GRpcUtil.GetConnection()
	if conn == nil {
		return nil, errors.New("RPC服务未启动！")
	}

	c := BasePositionProto.NewBasePositionManageClient(conn)
	Reply, err := c.DeleteBasePosition(ctx, &deleteIdsArg)
	if err != nil {
		LogUtil.Error("删除职务信息错误: ", err.Error())
	}
	return Reply, err
}

func UpdateBasePosition(modelArg BasePositionProto.ModelArg) (*BasePositionProto.Reply, error) {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()
	conn := GRpcUtil.GetConnection()
	if conn == nil {
		return nil, errors.New("RPC服务未启动！")
	}

	c := BasePositionProto.NewBasePositionManageClient(conn)
	Reply, err := c.UpdateBasePosition(ctx, &modelArg)
	if err != nil {
		LogUtil.Error("修改职务信息错误: ", err.Error())
	}
	return Reply, err
}

func TreeBasePosition() (*BasePositionProto.Reply, error) {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()
	conn := GRpcUtil.GetConnection()
	if conn == nil {
		return nil, errors.New("RPC服务未启动！")
	}

	c := BasePositionProto.NewBasePositionManageClient(conn)
	var queryArg BasePositionProto.QueryArg
	Reply, err := c.TreeBasePosition(ctx, &queryArg)
	if err != nil {
		LogUtil.Error("获取职务所有的树形结构信息错误: ", err.Error())
	}
	return Reply, err
}

func ListBasePositionByBureauId(listBasePositionByBureauIdArg BasePositionProto.ListBasePositionByBureauIdArg) (*BasePositionProto.Reply, error) {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()
	conn := GRpcUtil.GetConnection()
	if conn == nil {
		return nil, errors.New("RPC服务未启动！")
	}

	c := BasePositionProto.NewBasePositionManageClient(conn)
	Reply, err := c.ListBasePositionByBureauId(ctx, &listBasePositionByBureauIdArg)
	if err != nil {
		LogUtil.Error("获取职务信息错误: ", err.Error())
	}
	return Reply, err
}