CalvinKirs
3 years ago
33 changed files with 301 additions and 274 deletions
@ -1,12 +0,0 @@ |
|||||||
package org.apache.dolphinscheduler.remote.rpc; |
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.remote.rpc.common.RpcResponse; |
|
||||||
import org.apache.dolphinscheduler.remote.rpc.common.RpcRequest; |
|
||||||
|
|
||||||
/** |
|
||||||
* Invoker |
|
||||||
*/ |
|
||||||
public interface Invoker { |
|
||||||
|
|
||||||
RpcResponse invoke(RpcRequest req) throws Throwable; |
|
||||||
} |
|
@ -1,15 +0,0 @@ |
|||||||
package org.apache.dolphinscheduler.remote.rpc; |
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.remote.config.NettyServerConfig; |
|
||||||
import org.apache.dolphinscheduler.remote.rpc.remote.NettyServer; |
|
||||||
|
|
||||||
/** |
|
||||||
* @author jiangli |
|
||||||
* @date 2021-01-20 14:54 |
|
||||||
*/ |
|
||||||
public class MainServerTest { |
|
||||||
|
|
||||||
public static void main(String[] args) { |
|
||||||
NettyServer nettyServer = new NettyServer(new NettyServerConfig()); |
|
||||||
} |
|
||||||
} |
|
@ -1,39 +0,0 @@ |
|||||||
package org.apache.dolphinscheduler.remote.rpc; |
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.remote.config.NettyServerConfig; |
|
||||||
import org.apache.dolphinscheduler.remote.rpc.client.IRpcClient; |
|
||||||
import org.apache.dolphinscheduler.remote.rpc.client.RpcClient; |
|
||||||
import org.apache.dolphinscheduler.remote.rpc.remote.NettyServer; |
|
||||||
import org.apache.dolphinscheduler.remote.utils.Host; |
|
||||||
|
|
||||||
/** |
|
||||||
* @author jiangli |
|
||||||
* @date 2021-01-11 21:06 |
|
||||||
*/ |
|
||||||
public class MainTest { |
|
||||||
|
|
||||||
public static void main(String[] args) throws Exception { |
|
||||||
|
|
||||||
|
|
||||||
// NettyServer nettyServer = new NettyServer(new NettyServerConfig());
|
|
||||||
|
|
||||||
// NettyClient nettyClient=new NettyClient(new NettyClientConfig());
|
|
||||||
|
|
||||||
Host host = new Host("127.0.0.1", 12636); |
|
||||||
|
|
||||||
IRpcClient rpcClient = new RpcClient(); |
|
||||||
IUserService userService = rpcClient.create(IUserService.class, host); |
|
||||||
boolean result = userService.say("calvin"); |
|
||||||
System.out.println("异步回掉成功" + result); |
|
||||||
|
|
||||||
System.out.println(userService.hi(10)); |
|
||||||
System.out.println(userService.hi(188888888)); |
|
||||||
|
|
||||||
IUserService user = rpcClient.create(IUserService.class, host); |
|
||||||
System.out.println(user.hi(99999)); |
|
||||||
System.out.println(user.hi(998888888)); |
|
||||||
|
|
||||||
System.out.println(IUserService.class.getSimpleName()); |
|
||||||
System.out.println(UserService.class.getSimpleName()); |
|
||||||
} |
|
||||||
} |
|
@ -1,15 +0,0 @@ |
|||||||
package org.apache.dolphinscheduler.remote.rpc; |
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.remote.rpc.common.AbstractRpcCallBack; |
|
||||||
|
|
||||||
/** |
|
||||||
* @author jiangli |
|
||||||
* @date 2021-01-15 07:32 |
|
||||||
*/ |
|
||||||
public class UserCallback extends AbstractRpcCallBack { |
|
||||||
@Override |
|
||||||
public void run(Object object) { |
|
||||||
Boolean msg= (Boolean) object; |
|
||||||
System.out.println("我是异步回调handle Kris"+msg); |
|
||||||
} |
|
||||||
} |
|
@ -1,23 +0,0 @@ |
|||||||
package org.apache.dolphinscheduler.remote.rpc; |
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.remote.rpc.base.Rpc; |
|
||||||
import org.apache.dolphinscheduler.remote.rpc.base.RpcService; |
|
||||||
|
|
||||||
/** |
|
||||||
* @author jiangli |
|
||||||
* @date 2021-01-11 21:05 |
|
||||||
*/ |
|
||||||
@RpcService("IUserService") |
|
||||||
public class UserService implements IUserService{ |
|
||||||
|
|
||||||
@Rpc(async = true, serviceCallback = UserCallback.class, retries = 9999) |
|
||||||
@Override |
|
||||||
public Boolean say(String s) { |
|
||||||
return true; |
|
||||||
} |
|
||||||
|
|
||||||
@Override |
|
||||||
public String hi(int num) { |
|
||||||
return "this world has " + num + "sun"; |
|
||||||
} |
|
||||||
} |
|
@ -0,0 +1,45 @@ |
|||||||
|
/* |
||||||
|
* 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 org.apache.dolphinscheduler.rpc.common; |
||||||
|
|
||||||
|
import java.util.concurrent.ArrayBlockingQueue; |
||||||
|
import java.util.concurrent.ExecutorService; |
||||||
|
import java.util.concurrent.ThreadPoolExecutor; |
||||||
|
import java.util.concurrent.ThreadPoolExecutor.DiscardPolicy; |
||||||
|
import java.util.concurrent.TimeUnit; |
||||||
|
|
||||||
|
public enum ThreadPoolManager { |
||||||
|
|
||||||
|
INSTANCE; |
||||||
|
|
||||||
|
ExecutorService executorService; |
||||||
|
|
||||||
|
ThreadPoolManager() { |
||||||
|
int SIZE_WORK_QUEUE = 200; |
||||||
|
long KEEP_ALIVE_TIME = 60; |
||||||
|
int CORE_POOL_SIZE = Runtime.getRuntime().availableProcessors() * 2; |
||||||
|
int MAXI_MUM_POOL_SIZE = CORE_POOL_SIZE * 4; |
||||||
|
executorService = new ThreadPoolExecutor(CORE_POOL_SIZE, MAXI_MUM_POOL_SIZE, KEEP_ALIVE_TIME, TimeUnit.SECONDS, |
||||||
|
new ArrayBlockingQueue<>(SIZE_WORK_QUEUE), |
||||||
|
new DiscardPolicy()); |
||||||
|
} |
||||||
|
|
||||||
|
public void addExecuteTask(Runnable task) { |
||||||
|
executorService.submit(task); |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,63 @@ |
|||||||
|
/* |
||||||
|
* 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 org.apache.dolphinscheduler.rpc; |
||||||
|
|
||||||
|
import org.apache.dolphinscheduler.remote.config.NettyServerConfig; |
||||||
|
import org.apache.dolphinscheduler.remote.utils.Host; |
||||||
|
import org.apache.dolphinscheduler.rpc.client.IRpcClient; |
||||||
|
import org.apache.dolphinscheduler.rpc.client.RpcClient; |
||||||
|
import org.apache.dolphinscheduler.rpc.remote.NettyClient; |
||||||
|
import org.apache.dolphinscheduler.rpc.remote.NettyServer; |
||||||
|
|
||||||
|
import org.junit.After; |
||||||
|
import org.junit.Assert; |
||||||
|
import org.junit.Before; |
||||||
|
import org.junit.Test; |
||||||
|
|
||||||
|
public class RpcTest { |
||||||
|
private NettyServer nettyServer; |
||||||
|
|
||||||
|
private IUserService userService; |
||||||
|
|
||||||
|
private Host host; |
||||||
|
|
||||||
|
@Before |
||||||
|
public void before() throws Exception { |
||||||
|
nettyServer = new NettyServer(new NettyServerConfig()); |
||||||
|
IRpcClient rpcClient = new RpcClient(); |
||||||
|
host = new Host("127.0.0.1", 12346); |
||||||
|
userService = rpcClient.create(IUserService.class, host); |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void sendTest() { |
||||||
|
Integer result = userService.hi(3); |
||||||
|
Assert.assertSame(4, result); |
||||||
|
result = userService.hi(4); |
||||||
|
Assert.assertSame(5, result); |
||||||
|
userService.say("sync"); |
||||||
|
} |
||||||
|
|
||||||
|
@After |
||||||
|
public void after() { |
||||||
|
NettyClient.getInstance().close(); |
||||||
|
nettyServer.close(); |
||||||
|
} |
||||||
|
|
||||||
|
} |
@ -0,0 +1,30 @@ |
|||||||
|
/* |
||||||
|
* 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 org.apache.dolphinscheduler.rpc; |
||||||
|
|
||||||
|
import org.apache.dolphinscheduler.rpc.common.AbstractRpcCallBack; |
||||||
|
|
||||||
|
/** |
||||||
|
* UserCallback |
||||||
|
*/ |
||||||
|
public class UserCallback extends AbstractRpcCallBack { |
||||||
|
@Override |
||||||
|
public void run(Object object) { |
||||||
|
|
||||||
|
} |
||||||
|
} |
@ -0,0 +1,20 @@ |
|||||||
|
package org.apache.dolphinscheduler.rpc; |
||||||
|
|
||||||
|
import org.apache.dolphinscheduler.rpc.base.RpcService; |
||||||
|
|
||||||
|
/** |
||||||
|
* UserService |
||||||
|
*/ |
||||||
|
@RpcService("IUserService") |
||||||
|
public class UserService implements IUserService{ |
||||||
|
|
||||||
|
@Override |
||||||
|
public Boolean say(String s) { |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public Integer hi(int num) { |
||||||
|
return ++num; |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue