/* * * Copyright 2018 gRPC 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. * */ #include #include #include #include #include #ifdef BAZEL_BUILD #include "examples/protos/keyvaluestore.grpc.pb.h" #else #include "keyvaluestore.grpc.pb.h" #endif using grpc::Server; using grpc::ServerBuilder; using grpc::ServerContext; using grpc::ServerReaderWriter; using grpc::Status; using keyvaluestore::Request; using keyvaluestore::Response; using keyvaluestore::KeyValueStore; struct kv_pair { const char* key; const char* value; }; static const kv_pair kvs_map[] = { {"key1", "value1"}, {"key2", "value2"}, {"key3", "value3"}, {"key4", "value4"}, {"key5", "value5"}, }; const char * get_value_from_map(const char* key) { for(size_t i = 0; i < sizeof(kvs_map) / sizeof(kv_pair); ++i) { if(strcmp(key, kvs_map[i].key) == 0) { return kvs_map[i].value; } } return ""; } // Logic and data behind the server's behavior. class KeyValueStoreServiceImpl final : public KeyValueStore::Service { Status GetValues(ServerContext* context, ServerReaderWriter *stream) override { Request request; while(stream->Read(&request)) { Response response; response.set_value(get_value_from_map(request.key().c_str())); stream->Write(response); } return Status::OK; } }; void RunServer() { std::string server_address("0.0.0.0:50051"); KeyValueStoreServiceImpl service; ServerBuilder builder; // Listen on the given address without any authentication mechanism. builder.AddListeningPort(server_address, grpc::InsecureServerCredentials()); // Register "service" as the instance through which we'll communicate with // clients. In this case it corresponds to an *synchronous* service. builder.RegisterService(&service); // Finally assemble the server. std::unique_ptr server(builder.BuildAndStart()); std::cout << "Server listening on " << server_address << std::endl; // Wait for the server to shutdown. Note that some other thread must be // responsible for shutting down the server for this call to ever return. server->Wait(); } int main(int argc, char** argv) { RunServer(); return 0; }