View Javadoc
1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one or more
3    * contributor license agreements.  See the NOTICE file distributed with
4    * this work for additional information regarding copyright ownership.
5    * The ASF licenses this file to You under the Apache License, Version 2.0
6    * (the "License"); you may not use this file except in compliance with
7    * the License.  You may obtain a copy of the License at
8    *
9    *     http://www.apache.org/licenses/LICENSE-2.0
10   *
11   * Unless required by applicable law or agreed to in writing, software
12   * distributed under the License is distributed on an "AS IS" BASIS,
13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14   * See the License for the specific language governing permissions and
15   * limitations under the License.
16   */
17  
18  package org.apache.shardingsphere.infra.metadata.user;
19  
20  import com.google.common.base.Objects;
21  import com.google.common.base.Strings;
22  import lombok.Getter;
23  import org.apache.shardingsphere.infra.annotation.HighFrequencyInvocation;
24  
25  /**
26   * Grantee.
27   */
28  public final class Grantee {
29      
30      private static final String DELIMITER = "@";
31      
32      private static final String HOST_WILDCARD = "%";
33      
34      @Getter
35      private final String username;
36      
37      @Getter
38      private final String hostname;
39      
40      private final boolean isUnlimitedHost;
41      
42      private final int hashCode;
43      
44      private final String toString;
45      
46      public Grantee(final String value) {
47          this(value.contains("@") ? value.substring(0, value.indexOf('@')) : value, value.contains("@") ? value.substring(value.indexOf('@') + 1) : "");
48      }
49      
50      public Grantee(final String username, final String hostname) {
51          this.username = username;
52          this.hostname = Strings.isNullOrEmpty(hostname) ? HOST_WILDCARD : hostname;
53          isUnlimitedHost = HOST_WILDCARD.equals(this.hostname);
54          hashCode = Objects.hashCode(username.toUpperCase(), this.hostname.toUpperCase());
55          toString = username + DELIMITER + this.hostname;
56      }
57      
58      /**
59       * Check if the grantee is acceptable.
60       *
61       * @param grantee grantee
62       * @return if the grantee is acceptable
63       */
64      @HighFrequencyInvocation
65      public boolean accept(final Grantee grantee) {
66          return grantee.username.equalsIgnoreCase(username) && isPermittedHost(grantee);
67      }
68      
69      private boolean isPermittedHost(final Grantee grantee) {
70          return isUnlimitedHost || grantee.hostname.equalsIgnoreCase(hostname);
71      }
72      
73      @Override
74      public boolean equals(final Object obj) {
75          if (obj instanceof Grantee) {
76              Grantee grantee = (Grantee) obj;
77              return grantee.username.equalsIgnoreCase(username) && grantee.hostname.equalsIgnoreCase(hostname);
78          }
79          return false;
80      }
81      
82      @Override
83      public int hashCode() {
84          return hashCode;
85      }
86      
87      @Override
88      public String toString() {
89          return toString;
90      }
91  }