package main

import (
	"app.yhyue.com/BP/servicerd/proto"
	"context"
	"log"
)

/**
服务治理
*/
type Service struct {
}

//
func (s *Service) Registe(ctx context.Context, in *proto.ServiceMeta) (*proto.StringRepData, error) {
	AddServiceMeta(in.Name, in.Ip, in.Port, in.Workers, in.Balance)
	return &proto.StringRepData{Data: "ok"}, nil
}

//
func (s *Service) Destory(ctx context.Context, meta *proto.ServiceMeta) (*proto.StringRepData, error) {
	DestoryServiceMeta(meta.Ip, meta.Port)
	return &proto.StringRepData{Data: "ok"}, nil
}

//
func (s *Service) Apply(ctx context.Context, in *proto.ApplyReqData) (*proto.ApplyRepData, error) {
	var addr, id string
	var err error
	if in.Balance == LOAD {
		addr, id, err = ApplyWithLoad(in.Name)
	} else if in.Balance == SEQ {
		addr, id, err = ApplyWithNotUse(in.Name)
	} else {
		addr, id, err = ApplyWithRandom(in.Name)
	}
	if err != nil {
		log.Println(err.Error())
		return &proto.ApplyRepData{Addr: "", ResourceId: ""}, err
	} else {
		return &proto.ApplyRepData{Addr: addr, ResourceId: id}, nil
	}
}

func (s *Service) Release(ctx context.Context, in *proto.StringReqData) (*proto.StringRepData, error) {
	Release(in.Data)
	return &proto.StringRepData{Data: "ok"}, nil
}