Ejemplo n.º 1
0
func (p *service_pool) get_service(name ServiceType) (interface{}, error) {
	p.RLock()
	defer p.RUnlock()
	service := p.services[string(name)]
	if service == nil {
		return nil, ERROR_SERVICE_NOT_AVAILABLE
	}

	if len(service.clients) == 0 {
		return nil, ERROR_SERVICE_NOT_AVAILABLE
	}
	idx := int(atomic.AddUint32(&service.idx, 1))

	// add wrappers here ...
	pa := idx % len(service.clients)
	switch name {
	case SERVICE_SNOWFLAKE:
		return proto.NewSnowflakeServiceClient(service.clients[pa].conn), nil
	case SERVICE_GEOIP:
		return proto.NewGeoIPServiceClient(service.clients[pa].conn), nil
	case SERVICE_WORDFILTER:
		return proto.NewWordFilterServiceClient(service.clients[pa].conn), nil
	case SERVICE_BGSAVE:
		return proto.NewBgSaveServiceClient(service.clients[pa].conn), nil
	case SERVICE_AUTH:
		return proto.NewAuthServiceClient(service.clients[pa].conn), nil
	case SERVICE_CHAT:
		return proto.NewChatServiceClient(service.clients[pa].conn), nil
	case SERVICE_GAME:
		return proto.NewGameServiceClient(service.clients[pa].conn), nil

	}
	return nil, ERROR_SERVICE_NOT_AVAILABLE
}
Ejemplo n.º 2
0
func (p *service_pool) get_all_service(name ServiceType) (map[string]interface{}, error) {
	p.RLock()
	defer p.RUnlock()
	service := p.services[string(name)]
	if service == nil {
		return nil, ERROR_SERVICE_NOT_AVAILABLE
	}

	if len(service.clients) == 0 {
		return nil, ERROR_SERVICE_NOT_AVAILABLE
	}

	// all services
	conns := make(map[string]interface{})
	for _, v := range service.clients {
		k := v.key
		switch name {
		case SERVICE_SNOWFLAKE:
			conns[k] = proto.NewSnowflakeServiceClient(v.conn)
		case SERVICE_GEOIP:
			conns[k] = proto.NewGeoIPServiceClient(v.conn)
		case SERVICE_WORDFILTER:
			conns[k] = proto.NewWordFilterServiceClient(v.conn)
		case SERVICE_BGSAVE:
			conns[k] = proto.NewBgSaveServiceClient(v.conn)
		case SERVICE_AUTH:
			conns[k] = proto.NewAuthServiceClient(v.conn)
		case SERVICE_CHAT:
			conns[k] = proto.NewChatServiceClient(v.conn)
		case SERVICE_GAME:
			conns[k] = proto.NewGameServiceClient(v.conn)
		}
	}
	return conns, nil
}
Ejemplo n.º 3
0
// provide a specific key for a service
// service must be stored like /backends/xxx_service/xxx_id
func (p *service_pool) get_service_with_id(name ServiceType, id string) (interface{}, error) {
	p.RLock()
	defer p.RUnlock()
	service := p.services[string(name)]
	if service == nil {
		return nil, ERROR_SERVICE_NOT_AVAILABLE
	}

	if len(service.clients) == 0 {
		return nil, ERROR_SERVICE_NOT_AVAILABLE
	}

	var conn *grpc.ClientConn
	fullpath := string(name) + "/" + id
	for k := range service.clients {
		if service.clients[k].key == fullpath {
			conn = service.clients[k].conn
			break
		}
	}

	if conn == nil {
		return nil, ERROR_SERVICE_NOT_AVAILABLE
	}

	// add wrappers here ...
	switch name {
	case SERVICE_SNOWFLAKE:
		return proto.NewSnowflakeServiceClient(conn), nil
	case SERVICE_GEOIP:
		return proto.NewGeoIPServiceClient(conn), nil
	case SERVICE_WORDFILTER:
		return proto.NewWordFilterServiceClient(conn), nil
	case SERVICE_BGSAVE:
		return proto.NewBgSaveServiceClient(conn), nil
	case SERVICE_AUTH:
		return proto.NewAuthServiceClient(conn), nil
	case SERVICE_CHAT:
		return proto.NewChatServiceClient(conn), nil
	case SERVICE_GAME:
		return proto.NewGameServiceClient(conn), nil
	}
	return nil, ERROR_SERVICE_NOT_AVAILABLE
}