Refactor Additional header for netconf hello message.
[controller.git] / opendaylight / netconf / netconf-impl / src / main / java / org / opendaylight / controller / netconf / impl / NetconfServerSession.java
1 /*
2  * Copyright (c) 2013 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.netconf.impl;
10
11 import io.netty.channel.Channel;
12
13 import java.text.SimpleDateFormat;
14 import java.util.Date;
15 import java.util.regex.Matcher;
16 import java.util.regex.Pattern;
17
18 import org.opendaylight.controller.netconf.api.AbstractNetconfSession;
19 import org.opendaylight.controller.netconf.api.monitoring.NetconfManagementSession;
20 import org.opendaylight.controller.netconf.util.messages.NetconfHelloMessageAdditionalHeader;
21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.DomainName;
22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Host;
23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.extension.rev131210.NetconfTcp;
24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.extension.rev131210.Session1;
25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.extension.rev131210.Session1Builder;
26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.NetconfSsh;
27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.Transport;
28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.sessions.Session;
29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.sessions.SessionBuilder;
30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.sessions.SessionKey;
31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev100924.DateAndTime;
32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev100924.ZeroBasedCounter32;
33 import org.slf4j.Logger;
34 import org.slf4j.LoggerFactory;
35
36 import com.google.common.base.Preconditions;
37
38 public final class NetconfServerSession extends AbstractNetconfSession<NetconfServerSession, NetconfServerSessionListener> implements NetconfManagementSession {
39
40     private static final Logger logger = LoggerFactory.getLogger(NetconfServerSession.class);
41
42     private final NetconfHelloMessageAdditionalHeader header;
43
44     private Date loginTime;
45     private long inRpcSuccess, inRpcFail, outRpcError;
46
47     public NetconfServerSession(NetconfServerSessionListener sessionListener, Channel channel, long sessionId,
48             NetconfHelloMessageAdditionalHeader header) {
49         super(sessionListener, channel, sessionId);
50         this.header = header;
51         logger.debug("Session {} created", toString());
52     }
53
54     @Override
55     protected void sessionUp() {
56         super.sessionUp();
57         Preconditions.checkState(loginTime == null, "Session is already up");
58         this.loginTime = new Date();
59     }
60
61     public void onIncommingRpcSuccess() {
62         inRpcSuccess++;
63     }
64
65     public void onIncommingRpcFail() {
66         inRpcFail++;
67     }
68
69     public void onOutgoingRpcError() {
70         outRpcError++;
71     }
72
73     public static final String ISO_DATE_FORMAT = "yyyy-MM-dd'T'HH:mm:ss.SSSXXX";
74
75     private static final String dateTimePatternString = DateAndTime.PATTERN_CONSTANTS.get(0);
76     private static final Pattern dateTimePattern = Pattern.compile(dateTimePatternString);
77
78     @Override
79     public Session toManagementSession() {
80         SessionBuilder builder = new SessionBuilder();
81
82         builder.setSessionId(getSessionId());
83         builder.setSourceHost(new Host(new DomainName(header.getAddress())));
84
85         Preconditions.checkState(DateAndTime.PATTERN_CONSTANTS.size() == 1);
86         String formattedDateTime = formatDateTime(loginTime);
87
88         Matcher matcher = dateTimePattern.matcher(formattedDateTime);
89         Preconditions.checkState(matcher.matches(), "Formatted datetime %s does not match pattern %s", formattedDateTime, dateTimePattern);
90         builder.setLoginTime(new DateAndTime(formattedDateTime));
91
92         builder.setInBadRpcs(new ZeroBasedCounter32(inRpcFail));
93         builder.setInRpcs(new ZeroBasedCounter32(inRpcSuccess));
94         builder.setOutRpcErrors(new ZeroBasedCounter32(outRpcError));
95
96         builder.setUsername(header.getUserName());
97         builder.setTransport(getTransportForString(header.getTransport()));
98
99         builder.setOutNotifications(new ZeroBasedCounter32(0L));
100
101         builder.setKey(new SessionKey(getSessionId()));
102
103         Session1Builder builder1 = new Session1Builder();
104         builder1.setSessionIdentifier(header.getSessionIdentifier());
105         builder.addAugmentation(Session1.class, builder1.build());
106
107         return builder.build();
108     }
109
110     private Class<? extends Transport> getTransportForString(String transport) {
111         switch(transport) {
112         case "ssh" : return NetconfSsh.class;
113         case "tcp" : return NetconfTcp.class;
114         default: throw new IllegalArgumentException("Unknown transport type " + transport);
115         }
116     }
117
118     private String formatDateTime(Date loginTime) {
119         SimpleDateFormat dateFormat = new SimpleDateFormat(ISO_DATE_FORMAT);
120         return dateFormat.format(loginTime);
121     }
122
123     @Override
124     protected NetconfServerSession thisInstance() {
125         return this;
126     }
127 }