Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 1 | package org.onlab.onos.store.serializers; |
| 2 | |
| 3 | import java.net.URI; |
| 4 | import java.util.ArrayList; |
Madan Jampani | 53e44e6 | 2014-10-07 12:39:51 -0700 | [diff] [blame] | 5 | import java.util.Arrays; |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 6 | import java.util.HashMap; |
| 7 | |
| 8 | import org.onlab.onos.cluster.ControllerNode; |
| 9 | import org.onlab.onos.cluster.DefaultControllerNode; |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 10 | import org.onlab.onos.cluster.NodeId; |
Yuta HIGUCHI | 80912e6 | 2014-10-12 00:15:47 -0700 | [diff] [blame] | 11 | import org.onlab.onos.mastership.MastershipTerm; |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 12 | import org.onlab.onos.net.ConnectPoint; |
| 13 | import org.onlab.onos.net.DefaultAnnotations; |
| 14 | import org.onlab.onos.net.DefaultDevice; |
| 15 | import org.onlab.onos.net.DefaultLink; |
| 16 | import org.onlab.onos.net.DefaultPort; |
| 17 | import org.onlab.onos.net.Device; |
| 18 | import org.onlab.onos.net.DeviceId; |
| 19 | import org.onlab.onos.net.Element; |
Yuta HIGUCHI | 5fa3dc0 | 2014-10-15 17:08:13 -0700 | [diff] [blame] | 20 | import org.onlab.onos.net.HostId; |
| 21 | import org.onlab.onos.net.HostLocation; |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 22 | import org.onlab.onos.net.Link; |
| 23 | import org.onlab.onos.net.LinkKey; |
Yuta HIGUCHI | d40483d | 2014-10-09 15:20:30 -0700 | [diff] [blame] | 24 | import org.onlab.onos.net.MastershipRole; |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 25 | import org.onlab.onos.net.Port; |
| 26 | import org.onlab.onos.net.PortNumber; |
Madan Jampani | 53e44e6 | 2014-10-07 12:39:51 -0700 | [diff] [blame] | 27 | import org.onlab.onos.net.device.DefaultDeviceDescription; |
| 28 | import org.onlab.onos.net.device.DefaultPortDescription; |
Madan Jampani | 38b250d | 2014-10-17 11:02:38 -0700 | [diff] [blame^] | 29 | import org.onlab.onos.net.flow.DefaultFlowRule; |
Yuta HIGUCHI | 5fa3dc0 | 2014-10-15 17:08:13 -0700 | [diff] [blame] | 30 | import org.onlab.onos.net.host.DefaultHostDescription; |
| 31 | import org.onlab.onos.net.host.HostDescription; |
Madan Jampani | 2ff0559 | 2014-10-10 15:42:47 -0700 | [diff] [blame] | 32 | import org.onlab.onos.net.link.DefaultLinkDescription; |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 33 | import org.onlab.onos.net.provider.ProviderId; |
Yuta HIGUCHI | 9ee60f6 | 2014-10-09 10:00:01 -0700 | [diff] [blame] | 34 | import org.onlab.onos.store.Timestamp; |
alshabib | 7911a05 | 2014-10-16 17:49:37 -0700 | [diff] [blame] | 35 | import org.onlab.packet.ChassisId; |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 36 | import org.onlab.packet.IpAddress; |
| 37 | import org.onlab.packet.IpPrefix; |
Yuta HIGUCHI | 5fa3dc0 | 2014-10-15 17:08:13 -0700 | [diff] [blame] | 38 | import org.onlab.packet.MacAddress; |
| 39 | import org.onlab.packet.VlanId; |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 40 | import org.onlab.util.KryoPool; |
| 41 | |
Yuta HIGUCHI | 47c4088 | 2014-10-10 18:44:37 -0700 | [diff] [blame] | 42 | import com.google.common.collect.ImmutableList; |
| 43 | import com.google.common.collect.ImmutableMap; |
Yuta HIGUCHI | 5fa3dc0 | 2014-10-15 17:08:13 -0700 | [diff] [blame] | 44 | import com.google.common.collect.ImmutableSet; |
Yuta HIGUCHI | 47c4088 | 2014-10-10 18:44:37 -0700 | [diff] [blame] | 45 | |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 46 | public final class KryoPoolUtil { |
| 47 | |
| 48 | /** |
| 49 | * KryoPool which can serialize ON.lab misc classes. |
| 50 | */ |
| 51 | public static final KryoPool MISC = KryoPool.newBuilder() |
| 52 | .register(IpPrefix.class, new IpPrefixSerializer()) |
| 53 | .register(IpAddress.class, new IpAddressSerializer()) |
Yuta HIGUCHI | 5fa3dc0 | 2014-10-15 17:08:13 -0700 | [diff] [blame] | 54 | .register(MacAddress.class, new MacAddressSerializer()) |
| 55 | .register(VlanId.class) |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 56 | .build(); |
| 57 | |
| 58 | // TODO: Populate other classes |
| 59 | /** |
| 60 | * KryoPool which can serialize API bundle classes. |
| 61 | */ |
| 62 | public static final KryoPool API = KryoPool.newBuilder() |
| 63 | .register(MISC) |
Yuta HIGUCHI | 47c4088 | 2014-10-10 18:44:37 -0700 | [diff] [blame] | 64 | .register(ImmutableMap.class, new ImmutableMapSerializer()) |
| 65 | .register(ImmutableList.class, new ImmutableListSerializer()) |
Yuta HIGUCHI | 5fa3dc0 | 2014-10-15 17:08:13 -0700 | [diff] [blame] | 66 | .register(ImmutableSet.class, new ImmutableSetSerializer()) |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 67 | .register( |
| 68 | // |
| 69 | ArrayList.class, |
Madan Jampani | 53e44e6 | 2014-10-07 12:39:51 -0700 | [diff] [blame] | 70 | Arrays.asList().getClass(), |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 71 | HashMap.class, |
| 72 | // |
Yuta HIGUCHI | 47c4088 | 2014-10-10 18:44:37 -0700 | [diff] [blame] | 73 | // |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 74 | ControllerNode.State.class, |
| 75 | Device.Type.class, |
alshabib | 7911a05 | 2014-10-16 17:49:37 -0700 | [diff] [blame] | 76 | ChassisId.class, |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 77 | DefaultAnnotations.class, |
| 78 | DefaultControllerNode.class, |
| 79 | DefaultDevice.class, |
Madan Jampani | 53e44e6 | 2014-10-07 12:39:51 -0700 | [diff] [blame] | 80 | DefaultDeviceDescription.class, |
Madan Jampani | 2ff0559 | 2014-10-10 15:42:47 -0700 | [diff] [blame] | 81 | DefaultLinkDescription.class, |
Yuta HIGUCHI | d40483d | 2014-10-09 15:20:30 -0700 | [diff] [blame] | 82 | MastershipRole.class, |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 83 | Port.class, |
Madan Jampani | 53e44e6 | 2014-10-07 12:39:51 -0700 | [diff] [blame] | 84 | DefaultPortDescription.class, |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 85 | Element.class, |
Yuta HIGUCHI | 9ee60f6 | 2014-10-09 10:00:01 -0700 | [diff] [blame] | 86 | Link.Type.class, |
Yuta HIGUCHI | 5fa3dc0 | 2014-10-15 17:08:13 -0700 | [diff] [blame] | 87 | Timestamp.class, |
| 88 | HostId.class, |
| 89 | HostDescription.class, |
Madan Jampani | 38b250d | 2014-10-17 11:02:38 -0700 | [diff] [blame^] | 90 | DefaultHostDescription.class, |
| 91 | DefaultFlowRule.class |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 92 | ) |
| 93 | .register(URI.class, new URISerializer()) |
| 94 | .register(NodeId.class, new NodeIdSerializer()) |
| 95 | .register(ProviderId.class, new ProviderIdSerializer()) |
| 96 | .register(DeviceId.class, new DeviceIdSerializer()) |
| 97 | .register(PortNumber.class, new PortNumberSerializer()) |
| 98 | .register(DefaultPort.class, new DefaultPortSerializer()) |
| 99 | .register(LinkKey.class, new LinkKeySerializer()) |
| 100 | .register(ConnectPoint.class, new ConnectPointSerializer()) |
| 101 | .register(DefaultLink.class, new DefaultLinkSerializer()) |
Yuta HIGUCHI | fa891c9 | 2014-10-09 15:21:40 -0700 | [diff] [blame] | 102 | .register(MastershipTerm.class, new MastershipTermSerializer()) |
Yuta HIGUCHI | d40483d | 2014-10-09 15:20:30 -0700 | [diff] [blame] | 103 | .register(MastershipRole.class, new MastershipRoleSerializer()) |
Yuta HIGUCHI | 5fa3dc0 | 2014-10-15 17:08:13 -0700 | [diff] [blame] | 104 | .register(HostLocation.class, new HostLocationSerializer()) |
Yuta HIGUCHI | 03fec1f | 2014-10-03 09:13:50 -0700 | [diff] [blame] | 105 | |
| 106 | .build(); |
| 107 | |
| 108 | |
| 109 | // not to be instantiated |
| 110 | private KryoPoolUtil() {} |
| 111 | } |