opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/programs/appendentries/Client.java

   1 /*
   2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.controller.programs.appendentries;
  10
  11 import akka.actor.ActorSelection;
  12 import akka.actor.ActorSystem;
  13 import akka.actor.Props;
  14 import akka.actor.UntypedActor;
  15 import com.typesafe.config.ConfigFactory;
  16 import org.opendaylight.controller.cluster.datastore.CompositeModificationPayload;
  17 import org.opendaylight.controller.cluster.datastore.modification.MutableCompositeModification;
  18 import org.opendaylight.controller.cluster.datastore.modification.WriteModification;
  19 import org.opendaylight.controller.cluster.example.messages.KeyValue;
  20 import org.opendaylight.controller.cluster.raft.ReplicatedLogEntry;
  21 import org.opendaylight.controller.cluster.raft.messages.AppendEntries;
  22 import org.opendaylight.controller.cluster.raft.protobuff.client.messages.Payload;
  23 import org.opendaylight.controller.md.cluster.datastore.model.TestModel;
  24 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
  25
  26 import java.util.ArrayList;
  27 import java.util.List;
  28
  29 public class Client {
  30
  31     private static ActorSystem actorSystem;
  32
  33     public static class ClientActor extends UntypedActor {
  34
  35         @Override public void onReceive(Object message) throws Exception {
  36
  37         }
  38     }
  39
  40     public static void main(String[] args){
  41         actorSystem = ActorSystem.create("appendentries", ConfigFactory
  42             .load().getConfig("ODLCluster"));
  43
  44         ActorSelection actorSelection = actorSystem.actorSelection(
  45             "akka.tcp://appendentries@127.0.0.1:2550/user/server");
  46
  47         AppendEntries appendEntries = modificationAppendEntries();
  48
  49         Payload data = appendEntries.getEntries().get(0).getData();
  50         if(data instanceof CompositeModificationPayload) {
  51             System.out.println(
  52                 "Sending : " + ((CompositeModificationPayload) data)
  53                     .getModification());
  54         } else {
  55             System.out.println(
  56                 "Sending : " + ((KeyValue) data)
  57                     .getKey());
  58
  59         }
  60
  61         actorSelection.tell(appendEntries.toSerializable(), null);
  62
  63
  64
  65
  66         actorSystem.actorOf(Props.create(ClientActor.class), "client");
  67     }
  68
  69     public static AppendEntries modificationAppendEntries() {
  70         List<ReplicatedLogEntry> modification = new ArrayList<>();
  71
  72         modification.add(0, new ReplicatedLogEntry() {
  73             @Override public Payload getData() {
  74                 WriteModification writeModification =
  75                     new WriteModification(TestModel.TEST_PATH, ImmutableNodes
  76                         .containerNode(TestModel.TEST_QNAME),
  77                         TestModel.createTestContext()
  78                     );
  79
  80                 MutableCompositeModification compositeModification =
  81                     new MutableCompositeModification();
  82
  83                 compositeModification.addModification(writeModification);
  84
  85                 return new CompositeModificationPayload(
  86                     compositeModification.toSerializable());
  87             }
  88
  89             @Override public long getTerm() {
  90                 return 1;
  91             }
  92
  93             @Override public long getIndex() {
  94                 return 1;
  95             }
  96         });
  97
  98         return new AppendEntries(1, "member-1", 0, 100, modification, 1);
  99     }
 100
 101     public static AppendEntries keyValueAppendEntries() {
 102         List<ReplicatedLogEntry> modification = new ArrayList<>();
 103
 104         modification.add(0, new ReplicatedLogEntry() {
 105             @Override public Payload getData() {
 106                 return new KeyValue("moiz", "test");
 107             }
 108
 109             @Override public long getTerm() {
 110                 return 1;
 111             }
 112
 113             @Override public long getIndex() {
 114                 return 1;
 115             }
 116         });
 117
 118         return new AppendEntries(1, "member-1", 0, 100, modification, 1);
 119     }
 120 }