Data streams

src/main/java/com/wechi/adweb/bridge/google/analytics/GAAdminService.java

@@ -1,15 +1,15 @@
 package com.wechi.adweb.bridge.google.analytics;
 
-import com.google.analytics.admin.v1beta.Account;
-import com.google.analytics.admin.v1beta.AnalyticsAdminServiceClient;
-import com.google.analytics.admin.v1beta.AnalyticsAdminServiceSettings;
-import com.google.analytics.admin.v1beta.ListAccountsRequest;
+import com.google.analytics.admin.v1beta.*;
 import com.google.api.client.util.Lists;
 import com.google.api.gax.core.FixedCredentialsProvider;
 import com.google.auth.oauth2.GoogleCredentials;
 import com.google.common.base.Splitter;
+import com.google.gson.Gson;
 import com.wechi.adweb.bridge.exception.DataException;
 import com.wechi.adweb.bridge.google.analytics.dto.GAAccountDTO;
+import com.wechi.adweb.bridge.google.analytics.dto.GADataStreamDTO;
+import com.wechi.adweb.bridge.google.analytics.dto.GAPropertyDTO;
 
 import jakarta.annotation.PostConstruct;
 
@@ -28,6 +28,8 @@ import java.util.List;
 public class GAAdminService {
     private AnalyticsAdminServiceSettings adminServiceSettings;
 
+    private Gson gson = new Gson();
+
     @PostConstruct
     private void init() throws IOException {
         GoogleCredentials credentials =
@@ -51,8 +53,8 @@ public class GAAdminService {
 
             for (Account account : analyticsAdminServiceClient.listAccounts(request).iterateAll()) {
                 GAAccountDTO gaAccount = new GAAccountDTO();
-                gaAccount.setAccountId(parseId(account.getName()));
-                gaAccount.setAccountName(account.getName());
+                gaAccount.setId(parseId(account.getName()));
+                gaAccount.setName(account.getName());
                 gaAccount.setDisplayName(account.getDisplayName());
                 gaAccount.setCreateTime(account.getCreateTime().getSeconds());
                 gaAccount.setUpdateTime(account.getUpdateTime().getSeconds());
@@ -61,6 +63,7 @@ public class GAAdminService {
                 gaAccounts.add(gaAccount);
             }
 
+            log.info("listGAAccounts : {}", gson.toJson(gaAccounts));
             return gaAccounts;
         } catch (IOException e) {
             log.error(e.getMessage());
@@ -68,7 +71,78 @@ public class GAAdminService {
         }
     }
 
-    /** Parses GA account ID from name. */
+    public List<GAPropertyDTO> listGAProperties(String accountName) throws DataException {
+        List<GAPropertyDTO> gaProperties = Lists.newArrayList();
+        try (AnalyticsAdminServiceClient analyticsAdminServiceClient =
+                AnalyticsAdminServiceClient.create(adminServiceSettings)) {
+            // Initializes the list request with account name filter.
+            // See
+            // https://developers.google.com/analytics/devguides/config/admin/v1/rest/v1beta/properties/list#http-request
+            ListPropertiesRequest request =
+                    ListPropertiesRequest.newBuilder().setFilter("ancestor:" + accountName).build();
+
+            for (Property property :
+                    analyticsAdminServiceClient.listProperties(request).iterateAll()) {
+                GAPropertyDTO gaProperty = new GAPropertyDTO();
+                gaProperty.setId(parseId(property.getName()));
+                gaProperty.setName(property.getName());
+                gaProperty.setDisplayName(property.getDisplayName());
+                gaProperty.setPropertyType(property.getPropertyType().toString());
+                gaProperty.setAccount(property.getAccount());
+                gaProperty.setParent(property.getParent());
+                gaProperty.setIndustryCategory(property.getIndustryCategory().toString());
+                gaProperty.setTimeZone(property.getTimeZone());
+                gaProperty.setCurrencyCode(property.getCurrencyCode());
+                gaProperty.setServiceLevel(property.getServiceLevel().toString());
+                gaProperty.setCreateTime(property.getCreateTime().getSeconds());
+                gaProperty.setUpdateTime(property.getUpdateTime().getSeconds());
+
+                // Sets the data streams associated with this property.
+                gaProperty.setDataStreams(listGADataStreams(property.getName()));
+
+                // Adds into the list;
+                gaProperties.add(gaProperty);
+            }
+
+            log.info("listGAProperties for {} : {}", accountName, gson.toJson(gaProperties));
+            return gaProperties;
+        } catch (IOException e) {
+            log.error(e.getMessage());
+            throw new DataException(e);
+        }
+    }
+
+    public List<GADataStreamDTO> listGADataStreams(String propertyName) throws DataException {
+        List<GADataStreamDTO> gaDataStreams = Lists.newArrayList();
+        try (AnalyticsAdminServiceClient analyticsAdminServiceClient =
+                AnalyticsAdminServiceClient.create(adminServiceSettings)) {
+            // Initializes the list request with parent property.
+            ListDataStreamsRequest request =
+                    ListDataStreamsRequest.newBuilder().setParent(propertyName).build();
+
+            for (DataStream dataStream :
+                    analyticsAdminServiceClient.listDataStreams(request).iterateAll()) {
+                GADataStreamDTO gaDataStream = new GADataStreamDTO();
+                gaDataStream.setId(parseId(dataStream.getName()));
+                gaDataStream.setName(dataStream.getName());
+                gaDataStream.setDisplayName(dataStream.getDisplayName());
+                gaDataStream.setStreamMeasurementId(
+                        dataStream.getWebStreamData().getMeasurementId());
+                gaDataStream.setStreamDefaultUrl(dataStream.getWebStreamData().getDefaultUri());
+
+                // Adds into the list;
+                gaDataStreams.add(gaDataStream);
+            }
+
+            log.info("listGADataStreams for {} : {}", propertyName, gson.toJson(gaDataStreams));
+            return gaDataStreams;
+        } catch (IOException e) {
+            log.error(e.getMessage());
+            throw new DataException(e);
+        }
+    }
+
+    /** Parses GA ID from the resource name. */
     private static String parseId(String name) {
         List<String> parts = Splitter.on('/').splitToList(name);
         return parts.size() == 2 ? parts.get(1) : null;

src/main/java/com/wechi/adweb/bridge/google/analytics/dto/GAAccountDTO.java

@@ -7,13 +7,13 @@ import lombok.Data;
  */
 @Data
 public class GAAccountDTO {
-    public String accountId;
+    private String id;
 
-    public String accountName;
+    private String name;
 
-    public String displayName;
+    private String displayName;
 
-    public long createTime;
+    private long createTime;
 
-    public long updateTime;
+    private long updateTime;
 }

src/main/java/com/wechi/adweb/bridge/google/analytics/dto/GADataStreamDTO.java

@@ -0,0 +1,19 @@
+package com.wechi.adweb.bridge.google.analytics.dto;
+
+import lombok.Data;
+
+/**
+ * @author wfansh
+ */
+@Data
+public class GADataStreamDTO {
+    private String id;
+
+    private String name;
+
+    private String displayName;
+
+    private String streamMeasurementId;
+
+    private String streamDefaultUrl;
+}

src/main/java/com/wechi/adweb/bridge/google/analytics/dto/GAPropertyDTO.java

@@ -0,0 +1,38 @@
+package com.wechi.adweb.bridge.google.analytics.dto;
+
+import lombok.Data;
+
+import java.util.List;
+
+/**
+ * @author wfansh
+ */
+@Data
+public class GAPropertyDTO {
+
+    private String id;
+
+    private String name;
+
+    private String displayName;
+
+    private String propertyType;
+
+    private String account;
+
+    private String parent;
+
+    private String industryCategory;
+
+    private String timeZone;
+
+    private String currencyCode;
+
+    private String serviceLevel;
+
+    private long createTime;
+
+    private long updateTime;
+
+    private List<GADataStreamDTO> dataStreams;
+}