Move tpu samples to seperate directory

Author: Joanna Grycz

.github/workflows/tpu.yaml

@@ -0,0 +1,54 @@
+# Copyright 2024 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+name: tpu
+on:
+  push:
+    branches:
+    - main
+    paths:
+    - 'tpu/**'
+    - '.github/workflows/tpu.yaml'
+    - '.github/workflows/test.yaml'
+  pull_request:
+    types:
+    - opened
+    - reopened
+    - synchronize
+    - labeled
+    paths:
+    - 'tpu/**'
+    - '.github/workflows/tpu.yaml'
+    - '.github/workflows/test.yaml'
+  schedule:
+  - cron:  '0 0 * * 0'
+jobs:
+  test:
+    # Ref: https://github.com/google-github-actions/auth#usage
+    permissions:
+      contents: 'read'
+      id-token: 'write'
+    if: github.event.action != 'labeled' || github.event.label.name == 'actions:force-run'
+    uses: ./.github/workflows/test.yaml
+    with:
+      name: 'tpu'
+      path: 'tpu'
+  flakybot:
+    # Ref: https://github.com/google-github-actions/auth#usage
+    permissions:
+      contents: 'read'
+      id-token: 'write'
+    if: github.event_name == 'schedule' && always() # always() submits logs even if tests fail
+    uses: ./.github/workflows/flakybot.yaml
+    needs: [test]

.github/workflows/utils/workflows.json

@@ -90,6 +90,7 @@
   "speech",
   "talent",
   "texttospeech",
+  "tpu",
   "translate",
   "video-intelligence",
   "vision/productSearch",

CODEOWNERS

@@ -24,6 +24,7 @@ recaptcha_enterprise @GoogleCloudPlatform/dee-infra @GoogleCloudPlatform/nodejs-
 recaptcha_enterprise/demosite @GoogleCloudPlatform/dee-infra @GoogleCloudPlatform/recaptcha-customer-obsession-reviewers @GoogleCloudPlatform/nodejs-samples-reviewers @GoogleCloudPlatform/cloud-samples-reviewers
 secret-manager @GoogleCloudPlatform/dee-infra @GoogleCloudPlatform/nodejs-samples-reviewers @GoogleCloudPlatform/cloud-samples-reviewers @GoogleCloudPlatform/cloud-secrets-team
 service-directory @GoogleCloudPlatform/dee-infra @GoogleCloudPlatform/nodejs-samples-reviewers @GoogleCloudPlatform/cloud-samples-reviewers
+tpu @GoogleCloudPlatform/dee-infra @GoogleCloudPlatform/nodejs-samples-reviewers @GoogleCloudPlatform/cloud-samples-reviewers
 webrisk @GoogleCloudPlatform/dee-infra @GoogleCloudPlatform/nodejs-samples-reviewers @GoogleCloudPlatform/cloud-samples-reviewers
 
 # SoDa teams

compute/package.json

@@ -15,7 +15,6 @@
   },
   "dependencies": {
     "@google-cloud/compute": "^4.0.0",
-    "@google-cloud/tpu": "^3.5.0",
     "@sendgrid/mail": "^8.0.0",
     "nodemailer": "^6.0.0",
     "nodemailer-smtp-transport": "^2.7.4",

compute/test/tpu.test.js

@@ -15,15 +15,14 @@
 
 const uuid = require('uuid');
 const compute = require('@google-cloud/compute');
-const {TpuClient} = require('@google-cloud/tpu').v2;
 
 const PREFIX = 'gcloud-test-';
 
 const instancesClient = new compute.InstancesClient();
 
 // Get a unique ID to use for test resources.
 function generateTestId() {
-  return `${PREFIX}-${uuid.v4()}`.slice(0, 31);
+  return `${PREFIX}-${uuid.v4()}`.substr(0, 30);
 }
 
 /**
@@ -33,7 +32,7 @@ async function getStaleVMInstances(prefix = PREFIX) {
   const projectId = await instancesClient.getProjectId();
   const result = [];
   const currentDate = new Date();
-  currentDate.setHours(currentDate.getHours() - 3);
+  currentDate.setHours(currentDate.getHours() - 1);
 
   const aggListRequest = instancesClient.aggregatedListAsync({
     project: projectId,
@@ -93,7 +92,7 @@ async function getStaleReservations(prefix) {
   const projectId = await reservationsClient.getProjectId();
   const result = [];
   const currentDate = new Date();
-  currentDate.setHours(currentDate.getHours() - 3);
+  currentDate.setHours(currentDate.getHours() - 1);
 
   const aggListRequest = reservationsClient.aggregatedListAsync({
     project: projectId,
@@ -153,7 +152,7 @@ async function getStaleStoragePools(prefix) {
   const projectId = await storagePoolsClient.getProjectId();
   const result = [];
   const currentDate = new Date();
-  currentDate.setHours(currentDate.getHours() - 3);
+  currentDate.setHours(currentDate.getHours() - 1);
 
   const aggListRequest = storagePoolsClient.aggregatedListAsync({
     project: projectId,
@@ -212,7 +211,7 @@ async function getStaleDisks(prefix) {
   const projectId = await disksClient.getProjectId();
   const result = [];
   const currentDate = new Date();
-  currentDate.setHours(currentDate.getHours() - 3);
+  currentDate.setHours(currentDate.getHours() - 1);
 
   const aggListRequest = disksClient.aggregatedListAsync({
     project: projectId,
@@ -263,48 +262,6 @@ async function deleteDisk(zone, diskName) {
   }
 }
 
-const tpuClient = new TpuClient();
-
-/**
- * Get nodes created more than one hour ago.
- */
-async function getStaleNodes(prefix, zone) {
-  const projectId = await tpuClient.getProjectId();
-  const result = [];
-  const currentDate = new Date();
-  currentDate.setHours(currentDate.getHours() - 3);
-
-  const listNodesAsyncRequest = tpuClient.listNodesAsync({
-    parent: `projects/${projectId}/locations/${zone}`,
-  });
-
-  for await (const tpuObject of listNodesAsyncRequest) {
-    const name = tpuObject.name.split('/').slice(-1)[0];
-    const data = new Date(tpuObject.createTime.nanos / 1000000);
-    if (data < currentDate && name.startsWith(prefix)) {
-      result.push({
-        nodeName: tpuObject.name,
-        timestamp: tpuObject.createTime,
-      });
-    }
-  }
-
-  return result;
-}
-
-async function deleteNode(zone, nodeName) {
-  const projectId = await tpuClient.getProjectId();
-
-  const request = {
-    name: `projects/${projectId}/locations/${zone}/nodes/${nodeName}`,
-  };
-
-  const [operation] = await tpuClient.deleteNode(request);
-
-  // Wait for the delete operation to complete.
-  await operation.promise();
-}
-
 module.exports = {
   generateTestId,
   getStaleVMInstances,
@@ -315,6 +272,4 @@ module.exports = {
   deleteStoragePool,
   getStaleDisks,
   deleteDisk,
-  getStaleNodes,
-  deleteNode,
 };