mirror of https://github.com/longtai-cn/hippo4j
pull/946/head
pizihao
3 years ago
parent
d9a37dbeba
commit
695a78b2c6
@ -0,0 +1,33 @@
|
||||
/*
|
||||
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||
* contributor license agreements. See the NOTICE file distributed with
|
||||
* this work for additional information regarding copyright ownership.
|
||||
* The ASF licenses this file to You 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 cn.hippo4j.rpc.discovery;
|
||||
|
||||
/**
|
||||
* Gets the top-level interface of the instance port
|
||||
*/
|
||||
@FunctionalInterface
|
||||
public interface ServerPort {
|
||||
|
||||
/**
|
||||
* Gets the listening or exposed port
|
||||
*
|
||||
* @return port
|
||||
*/
|
||||
int getPort();
|
||||
|
||||
}
|
||||
@ -0,0 +1,113 @@
|
||||
/*
|
||||
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||
* contributor license agreements. See the NOTICE file distributed with
|
||||
* this work for additional information regarding copyright ownership.
|
||||
* The ASF licenses this file to You 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 cn.hippo4j.rpc.support;
|
||||
|
||||
import cn.hippo4j.rpc.discovery.ClassRegistry;
|
||||
import cn.hippo4j.rpc.discovery.DefaultInstance;
|
||||
import cn.hippo4j.rpc.discovery.ServerPort;
|
||||
import cn.hippo4j.rpc.handler.HandlerManager;
|
||||
import cn.hippo4j.rpc.handler.NettyServerTakeHandler;
|
||||
import cn.hippo4j.rpc.server.AbstractNettyServerConnection;
|
||||
import cn.hippo4j.rpc.server.RPCServer;
|
||||
import cn.hippo4j.rpc.server.Server;
|
||||
import io.netty.channel.ChannelHandler;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.util.Arrays;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* This is a server-side build class that allows you to quickly prepare data on the server side and start the server side.<br>
|
||||
* <p>
|
||||
* The composite pattern is adopted, which means that it is itself a server-side implementation, so it is stateless.
|
||||
*/
|
||||
public class NettyServerSupport implements Server {
|
||||
|
||||
/**
|
||||
* The interface that the server side can call,
|
||||
* All the methods in the interface are brokered during initialization
|
||||
*/
|
||||
List<Class<?>> classes;
|
||||
|
||||
/**
|
||||
* Extract the port number of the web container,
|
||||
* which is the port information exposed by the server
|
||||
*/
|
||||
ServerPort serverPort;
|
||||
|
||||
/**
|
||||
* ChannelHandler
|
||||
*/
|
||||
HandlerManager<ChannelHandler> handlerManager;
|
||||
|
||||
Server server;
|
||||
|
||||
public NettyServerSupport(ServerPort serverPort, Class<?>... classes) {
|
||||
this(serverPort, new AbstractNettyServerConnection(), classes);
|
||||
}
|
||||
|
||||
public NettyServerSupport(ServerPort serverPort, List<Class<?>> classes) {
|
||||
this(serverPort, new AbstractNettyServerConnection(), classes);
|
||||
}
|
||||
|
||||
public NettyServerSupport(ServerPort serverPort, HandlerManager<ChannelHandler> handlerManager, Class<?>... classes) {
|
||||
this(serverPort, handlerManager, classes != null ? Arrays.asList(classes) : Collections.emptyList());
|
||||
}
|
||||
|
||||
public NettyServerSupport(ServerPort serverPort, HandlerManager<ChannelHandler> handlerManager, List<Class<?>> classes) {
|
||||
this.classes = classes;
|
||||
this.serverPort = serverPort;
|
||||
this.handlerManager = handlerManager;
|
||||
initServer();
|
||||
}
|
||||
|
||||
/**
|
||||
* Initializes the entire server side, which includes interface registration, processors, and ports.<br>
|
||||
* Only interfaces are registered during registration. Classes and abstract classes are not registered.
|
||||
* If no processor is available, a default processor is provided
|
||||
*/
|
||||
private void initServer() {
|
||||
// Register the interface that can be invoked
|
||||
classes.stream().filter(Class::isInterface)
|
||||
.forEach(cls -> ClassRegistry.put(cls.getName(), cls));
|
||||
AbstractNettyServerConnection connection = (handlerManager instanceof AbstractNettyServerConnection)
|
||||
? (AbstractNettyServerConnection) handlerManager
|
||||
: new AbstractNettyServerConnection();
|
||||
// Assign a default handler if no handler exists
|
||||
if (connection.isEmpty()) {
|
||||
connection.addLast(new NettyServerTakeHandler(new DefaultInstance()));
|
||||
}
|
||||
server = new RPCServer(connection, serverPort);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void bind() {
|
||||
server.bind();
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean isActive() {
|
||||
return server.isActive();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void close() throws IOException {
|
||||
server.close();
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue