io.snice.networking.examples.diameter.Hss Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of networking-examples Show documentation
There is a newer version: 0.0.13
package io.snice.networking.examples.diameter;

import io.snice.codecs.codec.diameter.DiameterMessage;
import io.snice.codecs.codec.diameter.avp.api.ExperimentalResultCode;
import io.snice.codecs.codec.diameter.avp.api.ResultCode;
import io.snice.networking.app.NetworkApplication;
import io.snice.networking.app.NetworkBootstrap;
import io.snice.networking.diameter.DiameterBundle;
import io.snice.networking.diameter.DiameterEnvironment;
import io.snice.networking.diameter.PeerConnection;
import io.snice.networking.diameter.event.DiameterEvent;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Random;

import static io.snice.networking.app.NetworkBootstrap.ACCEPT_ALL;
import static io.snice.networking.examples.diameter.HssHelpers.createULR;
import static io.snice.networking.examples.diameter.HssHelpers.sleepy;

public class Hss extends NetworkApplication, PeerConnection, DiameterEvent, HssConfig> {

    private static final Logger logger = LoggerFactory.getLogger(Hss.class);

    public Hss(final DiameterBundle bundle) {
        super(bundle);
    }

    @Override
    public void run(final HssConfig configuration, final DiameterEnvironment environment) {

        // peer.establishPeer();

        final var t = new Thread(null, () -> {
            final var c1 = HssHelpers.createPeerConf(3869);
            final var c2 = HssHelpers.createPeerConf(3870);
            final var peer = environment.addPeer(c1);
            // final var peer2 = environment.addPeer(c2);

            System.err.println("About to send ULR but will sleep first");
            // sleepy(1000);
            System.err.println("Ok, sending");
            peer.establishPeer().thenAccept(p -> {
                final var ulr = createULR();
                System.out.println("Sending via a Transaction");
                final var transaction = p.createNewTransaction(ulr)
                        .onAnswer((t2, answer) -> {
                            System.out.println("Got back an Answer in the transaction callback!");
                        })
                        .start();
            });
            // peer2.establishPeer().thenAccept(p -> System.err.println("Peer2 estabalished"));

            for (int i = 0; i < 2; ++i) {
                sleepy(100);
                // System.err.println("Ok, sending via Peer.send");
                // environment.getPeers().stream().findAny().ifPresent(peer -> peer.send(createULR()));
                // peer.send(createULR());
            }

            // System.err.println("Ok, sending to both peers now.");
            sleepy(1000);
            for (int i = 0; i < 2; ++i) {
                sleepy(100);
                // System.err.println("Ok, sending via Peer.establishPeer().thenAccept");
                // environment.getPeers().stream().findAny().ifPresent(peer -> peer.send(createULR()));
                // peer.establishPeer().thenAccept(p -> p.send(createULR()));
                // peer2.send(createULR());
            }
        }, "kicking off something");
        // t.start();

        environment.getPeers().forEach(p -> {
            logger.info("Establishing peer: " + p);
            p.establishPeer().thenAccept(established -> {
                logger.info("Peer Established: " + p);
                final var myPeer = established;
                final var th = new Thread(null, () -> {
                    logger.info("Starting new Peer Thread for peer: " + myPeer);
                    myPeer.send(createULR());
                    final var random = new Random();
                    int max = random.nextInt(5);
                    logger.info("Sleeping 5 seconds then issuing " + max + " no of ULRs for Peer " + myPeer);
                    sleepy(5000);
                    for (int i = 0; i < max; ++i) {
                        myPeer.send(createULR());
                        sleepy(5);
                    }
                    int sleepy = random.nextInt(30) + 10;
                    /*
                    max = random.nextInt(50);
                    logger.info("Sleeping " + sleepy + " seconds then issuing " + max + " no of ULRs for Peer " + myPeer);
                    sleepy(sleepy * 1000);
                    for (int i = 0; i < max; ++i) {
                        myPeer.send(createULR());
                        sleepy(5);
                    }
                     */
                    sleepy = 4;
                    max = 3;
                    logger.info("Performance: Sleeping " + sleepy + " seconds then issuing " + max + " no of ULRs for Peer " + myPeer);
                    for (int i = 0; i < max; ++i) {
                        myPeer.send(createULR());
                        sleepy(2);
                    }
                });

                // th.start();
            });
        });

    }



    @Override
    public void initialize(final NetworkBootstrap bootstrap) {
        bootstrap.onConnection(ACCEPT_ALL).accept(b -> {
            b.match(DiameterEvent::isULR).map(DiameterEvent::getRequest).consume(Hss::processULR);
            b.match(DiameterEvent::isULA).map(DiameterEvent::getAnswer).consume(Hss::processULA);
            b.matchEvent(o -> o instanceof Double).consume((c, d) -> {
                System.out.println("I got the Double " + d + " firing off a ULR out of the blue, which should confuse the shit out of seagull");
                c.send(createULR());
            });
            b.matchEvent(o -> o instanceof String).map(o -> (String) o).consume(s -> {
                System.out.println("I got the String " + s + " which is of length " + s.length());
            });
            b.matchEvent(o -> o instanceof Integer).map(o -> (Integer) o).consume(i -> {
                System.out.println("I got the Integer " + i);
            });
        });

    }

    private static final void processULR(final PeerConnection peerConnection, final DiameterMessage ulr) {
        final var ula = ulr.createAnswer(ResultCode.DiameterErrorUserUnknown5032)
                .withAvp(ExperimentalResultCode.DiameterErrorUserUnknown5001);
        peerConnection.send(ula);
    }

    private static final void processULA(final PeerConnection peerConnection, final DiameterMessage ula) {
        logger.info("yay, we got a ULA back!");
    }

    public static void main(final String... args) throws Exception {
        final DiameterBundle bundle = new DiameterBundle();
        final var hss = new Hss(bundle);
        hss.run("server", "networking-examples/src/main/resources/io/snice/networking/examples/Hss.yml");
    }
}