nemo-impl/src/main/java/org/opendaylight/nemo/user/tenantmanager/TenantManage.java

   1 /*\r
   2  * Copyright (c) 2015 Huawei, Inc. and others. All rights reserved.\r
   3  *\r
   4  * This program and the accompanying materials are made available under the\r
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,\r
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html\r
   7  */\r
   8 package org.opendaylight.nemo.user.tenantmanager;\r
   9 \r
  10 import com.google.common.base.Optional;\r
  11 import com.google.common.util.concurrent.FutureCallback;\r
  12 import com.google.common.util.concurrent.Futures;\r
  13 import com.google.common.util.concurrent.CheckedFuture;\r
  14 import com.google.common.util.concurrent.ListenableFuture;\r
  15 import org.opendaylight.controller.md.sal.binding.api.DataBroker;\r
  16 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;\r
  17 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;\r
  18 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;\r
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.common.rev151010.UserId;\r
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.intent.rev151010.RegisterUserInput;\r
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.intent.rev151010.Users;\r
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.intent.rev151010.users.User;\r
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.intent.rev151010.users.UserBuilder;\r
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.intent.rev151010.users.UserKey;\r
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.user.rev151010.UserRoles;\r
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.nemo.user.rev151010.user.roles.UserRole;\r
  27 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;\r
  28 import org.slf4j.Logger;\r
  29 import org.slf4j.LoggerFactory;\r
  30 \r
  31 import java.util.List;\r
  32 \r
  33 \r
  34 /**\r
  35  * Created by z00293636 on 2015/8/29.\r
  36  */\r
  37 \r
  38 /* maintain tenant information, including how many tenants are active, their information */\r
  39 public class TenantManage {\r
  40     private static final Logger LOG = LoggerFactory.getLogger(TenantManage.class);\r
  41     private DataBroker dataBroker;\r
  42     private List<UserRole> userRoleList;\r
  43     private List<User> usersList ;\r
  44     private User user;\r
  45 \r
  46     public TenantManage(DataBroker dataBroker)\r
  47     {\r
  48         this.dataBroker = dataBroker;\r
  49     }\r
  50 \r
  51     private void setUserRoleList(List<UserRole> userRoleList)\r
  52     {\r
  53         this.userRoleList = userRoleList;\r
  54     }\r
  55 \r
  56     private void setUserList(List<User> userList)\r
  57     {\r
  58         this.usersList = userList;\r
  59     }\r
  60 \r
  61     private void setUser(User user)\r
  62     {\r
  63         this.user = user;\r
  64     }\r
  65 \r
  66     public List<UserRole> getUserRoleList()\r
  67     {\r
  68         return userRoleList;\r
  69     }\r
  70 \r
  71     public List<User> getUsersList()\r
  72     {\r
  73         return usersList;\r
  74     }\r
  75 \r
  76     public User getUser()\r
  77     {\r
  78         return user;\r
  79     }\r
  80 \r
  81     public void fetchUserRoles(){\r
  82 \r
  83         InstanceIdentifier<UserRoles> userRolesInsId = InstanceIdentifier.builder(UserRoles.class).build();\r
  84         ListenableFuture<Optional<UserRoles>> userRolesFuture = this.dataBroker.newReadOnlyTransaction().read(LogicalDatastoreType.CONFIGURATION, userRolesInsId);\r
  85         Futures.addCallback(userRolesFuture, new FutureCallback<Optional<UserRoles>>() {\r
  86             @Override\r
  87             public void onSuccess(Optional<UserRoles> result)\r
  88             {\r
  89                 setUserRoleList(result.get().getUserRole());\r
  90                 return;\r
  91             }\r
  92 \r
  93             @Override\r
  94             public void onFailure(Throwable t)\r
  95             {\r
  96                 LOG.error("Can not read role information.", t);\r
  97 \r
  98                 return;\r
  99             }\r
 100         });\r
 101          return;\r
 102     }\r
 103 \r
 104     public void fetchUsers(){\r
 105         InstanceIdentifier<Users> usersInsId = InstanceIdentifier.builder(Users.class).build();\r
 106         ListenableFuture<Optional<Users>> usersFuture = dataBroker.newReadOnlyTransaction().read(LogicalDatastoreType.CONFIGURATION, usersInsId);\r
 107         Futures.addCallback(usersFuture, new FutureCallback<Optional<Users>>() {\r
 108             @Override\r
 109             public void onSuccess(Optional<Users> result)\r
 110             {\r
 111                 setUserList(result.get().getUser());\r
 112                 return;\r
 113             }\r
 114 \r
 115             @Override\r
 116             public void onFailure(Throwable t)\r
 117             {\r
 118                 LOG.error("Can not read users information.", t);\r
 119 \r
 120                 return;\r
 121             }\r
 122         });\r
 123         return;\r
 124     }\r
 125 \r
 126     public void fetchVNSpace(UserId userId)\r
 127     {\r
 128         fetchUsers();\r
 129         if (getUsersList() != null)\r
 130         {\r
 131             for (User user : getUsersList())\r
 132             {\r
 133                 if (user.getUserId().equals(userId))\r
 134                 {\r
 135                     setUser(user);\r
 136                     break;\r
 137                 }\r
 138             }\r
 139         }\r
 140         return;\r
 141     }\r
 142 \r
 143     public void addUser(RegisterUserInput registerUserInput){\r
 144         WriteTransaction t = dataBroker.newWriteOnlyTransaction();\r
 145         if (registerUserInput.getUserId() != null)\r
 146         {\r
 147             User user = new UserBuilder(registerUserInput).build();\r
 148 //            UserBuilder userBuilder = new UserBuilder();\r
 149 //            userBuilder.setUserId(registerUserInput.getUserId());\r
 150 //            userBuilder.setUserName(registerUserInput.getUserName());\r
 151 //            userBuilder.setUserPassword(registerUserInput.getUserPassword());\r
 152 //            userBuilder.setUserRole(registerUserInput.getUserRole());\r
 153 //\r
 154 //            User user = userBuilder.build();\r
 155             UserKey userKey = new UserKey(registerUserInput.getUserId());\r
 156 \r
 157             InstanceIdentifier<User> userid = InstanceIdentifier.builder(Users.class).child(User.class, userKey).build();\r
 158 \r
 159             t.put(LogicalDatastoreType.CONFIGURATION, userid, user,true);\r
 160             CheckedFuture<Void, TransactionCommitFailedException> f = t.submit();\r
 161             Futures.addCallback(f, new FutureCallback<Void>() {\r
 162                 @Override\r
 163                 public void onFailure(Throwable t) {\r
 164                     LOG.error("Could not write endpoint base container", t);\r
 165                 }\r
 166 \r
 167                 @Override\r
 168                 public void onSuccess(Void result) {\r
 169 \r
 170                 }\r
 171             });\r
 172         }\r
 173     }\r
 174 }\r