io.activej.launchers.crdt.rpc.CrdtRpcClientModule Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2020 ActiveJ LLC.
 *
 * 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 io.activej.launchers.crdt.rpc;

import io.activej.config.Config;
import io.activej.eventloop.Eventloop;
import io.activej.inject.annotation.Provides;
import io.activej.inject.binding.OptionalDependency;
import io.activej.inject.module.AbstractModule;
import io.activej.rpc.client.RpcClient;
import io.activej.rpc.client.sender.RpcStrategies;
import io.activej.rpc.client.sender.RpcStrategy;
import io.activej.rpc.client.sender.RpcStrategyList;
import io.activej.rpc.hash.ShardingFunction;

import java.net.InetSocketAddress;
import java.util.Collections;
import java.util.List;

import static io.activej.config.Config.ofClassPathProperties;
import static io.activej.config.Config.ofSystemProperties;
import static io.activej.config.converter.ConfigConverters.ofInetSocketAddress;
import static io.activej.config.converter.ConfigConverters.ofList;
import static io.activej.launchers.crdt.rpc.CrdtRpcServerModule.DEFAULT_PORT;

public class CrdtRpcClientModule extends AbstractModule {
	public static final String PROPERTIES_FILE = "crdt-rpc-client.properties";

	@Provides
	Eventloop eventloop() {
		return Eventloop.create();
	}

	@Provides
	Config config() {
		return Config.create()
				.with("address", Config.ofValue(ofInetSocketAddress(), new InetSocketAddress(DEFAULT_PORT)))
				.overrideWith(ofClassPathProperties(PROPERTIES_FILE, true))
				.overrideWith(ofSystemProperties("config"));
	}

	@Provides
	RpcClient client(Eventloop eventloop, RpcStrategy strategy, List> messageTypes) {
		return RpcClient.create(eventloop)
				.withMessageTypes(messageTypes)
				.withStrategy(strategy);
	}

	@Provides
	RpcStrategy strategy(Config config, OptionalDependency> shardingFn) {
		List addresses = config.get(ofList(ofInetSocketAddress()), "addresses", Collections.emptyList());

		if (addresses.isEmpty()) {
			InetSocketAddress address = config.get(ofInetSocketAddress(), "address");
			return RpcStrategies.server(address);
		}

		return RpcStrategies.sharding(shardingFn.get(), RpcStrategyList.ofAddresses(addresses));
	}
}