rpcplatform_newserver.go

/*
 * Copyright 2022 RPCPlatform Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package rpcplatform

import (
	"net"

	"github.com/nexcode/rpcplatform/internal/gears"
	"github.com/nexcode/rpcplatform/internal/grpcinject"
	"google.golang.org/grpc"
)

// NewServer creates a new server. You need to provide the server name, address and attributes.
// If no additional settings are needed, attributes can be nil.
func (p *RPCPlatform) NewServer(name, addr string, attributes *ServerAttributes) (*Server, error) {
	if attributes == nil {
		attributes = Attributes().Server()
	}

	options := p.config.GRPCOptions.Server

	listener, err := net.Listen("tcp", addr)
	if err != nil {
		return nil, err
	}

	if err = grpcinject.Injections(&options, p.config, listener.Addr().String()); err != nil {
		return nil, err
	}

	return &Server{
		name:       p.config.EtcdPrefix + gears.FixPath(name),
		etcd:       p.config.EtcdClient,
		server:     grpc.NewServer(options...),
		listener:   listener,
		attributes: attributes,
	}, nil
}