migrate code from googleapis/nodejs-datacatalog

.github/workflows/datacatalog-snippets.yaml

@@ -0,0 +1,68 @@
+name: datacatalog-snippets
+on:
+  push:
+    branches:
+    - main
+    paths:
+    - 'datacatalog/snippets/**'
+  pull_request:
+    paths:
+    - 'datacatalog/snippets/**'
+  pull_request_target:
+    types: [labeled]
+    paths:
+    - 'datacatalog/snippets/**'
+  schedule:
+  - cron:  '0 0 * * 0'
+jobs:
+  test:
+    if: ${{ github.event.action != 'labeled' || github.event.label.name == 'actions:force-run' }}
+    runs-on: ubuntu-latest
+    timeout-minutes: 60
+    permissions:
+      contents: 'write'
+      pull-requests: 'write'
+      id-token: 'write'
+    steps:
+    - uses: actions/checkout@v3.1.0
+      with:
+        ref: ${{github.event.pull_request.head.sha}}
+    - uses: 'google-github-actions/auth@v1.0.0'
+      with:
+        workload_identity_provider: 'projects/1046198160504/locations/global/workloadIdentityPools/github-actions-pool/providers/github-actions-provider'
+        service_account: 'kokoro-system-test@long-door-651.iam.gserviceaccount.com'
+        create_credentials_file: 'true'
+        access_token_lifetime: 600s
+    - uses: actions/setup-node@v3.5.1
+      with:
+        node-version: 16
+    - run: npm install
+      working-directory: datacatalog/snippets
+    - run: npm test
+      working-directory: datacatalog/snippets
+      env:
+        MOCHA_REPORTER_SUITENAME: datacatalog_snippets
+        MOCHA_REPORTER_OUTPUT: datacatalog_snippets_sponge_log.xml
+        MOCHA_REPORTER: xunit
+    - if: ${{ github.event.action == 'labeled' && github.event.label.name == 'actions:force-run' }}
+      uses: actions/github-script@v6
+      with:
+        github-token: ${{ secrets.GITHUB_TOKEN }}
+        script: |
+          try {
+            await github.rest.issues.removeLabel({
+              name: 'actions:force-run',
+              owner: 'GoogleCloudPlatform',
+              repo: 'nodejs-docs-samples',
+              issue_number: context.payload.pull_request.number
+            });
+          } catch (e) {
+            if (!e.message.includes('Label does not exist')) {
+              throw e;
+            }
+          }
+    - if: ${{ github.event_name == 'schedule'}}
+      run: |
+        curl https://github.com/googleapis/repo-automation-bots/releases/download/flakybot-1.1.0/flakybot -o flakybot -s -L
+        chmod +x ./flakybot
+        ./flakybot --repo GoogleCloudPlatform/nodejs-docs-samples --commit_hash ${{github.sha}} --build_url https://github.com/${{github.repository}}/actions/runs/${{github.run_id}}

.github/workflows/workflows.json

@@ -28,6 +28,7 @@
   "containerengine/hello-world",
   "container/snippets",
   "datacatalog/cloud-client",
+  "datacatalog/snippets",
   "datalabeling",
   "datastore/functions",
   "datacatalog/quickstart",

datacatalog/snippets/.eslintrc.yml

@@ -0,0 +1,5 @@
+---
+rules:
+  no-console: off
+  node/no-missing-require: off
+  node/no-extraneous-require: off

datacatalog/snippets/createCustomEntry.js

@@ -0,0 +1,192 @@
+// Copyright 2020 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.
+
+'use strict';
+
+async function main(projectId, entryGroupId, entryId, tagTemplateId) {
+  // [START data_catalog_create_custom_entry]
+  // Import the Google Cloud client library.
+  const {DataCatalogClient} = require('@google-cloud/datacatalog').v1;
+  const datacatalog = new DataCatalogClient();
+
+  async function createCustomEntry() {
+    // Create a custom entry within an entry group.
+
+    /**
+     * TODO(developer): Uncomment the following lines before running the sample.
+     */
+    // const projectId = 'my_project';
+    // const entryGroupId = 'my_entry_group';
+    // const entryId =  'my_entry';
+    // const tagTemplateId = 'my_tag_template';
+
+    // Currently, Data Catalog stores metadata in the us-central1 region.
+    const location = 'us-central1';
+
+    // Delete any pre-existing Entry with the same name
+    // that will be used to create the new Entry.
+    try {
+      const entryName = datacatalog.entryPath(
+        projectId,
+        location,
+        entryGroupId,
+        entryId
+      );
+      await datacatalog.deleteEntry({name: entryName});
+      console.log(`Deleted Entry: ${entryName}`);
+    } catch (err) {
+      console.log('Entry does not exist.');
+    }
+
+    // Delete any pre-existing Entry Group with the same name
+    // that will be used to construct the new EntryGroup.
+    try {
+      const entryGroupName = datacatalog.entryGroupPath(
+        projectId,
+        location,
+        entryGroupId
+      );
+      await datacatalog.deleteEntryGroup({name: entryGroupName});
+      console.log(`Deleted Entry Group: ${entryGroupName}`);
+    } catch (err) {
+      console.log('Entry Group does not exist.');
+    }
+
+    // Delete any pre-existing Template with the same name
+    // that will be used to create a new Template.
+    const tagTemplateName = datacatalog.tagTemplatePath(
+      projectId,
+      location,
+      tagTemplateId
+    );
+
+    let tagTemplateRequest = {
+      name: tagTemplateName,
+      force: true,
+    };
+
+    try {
+      await datacatalog.deleteTagTemplate(tagTemplateRequest);
+      console.log(`Deleted template: ${tagTemplateName}`);
+    } catch (error) {
+      console.log(`Cannot delete template: ${tagTemplateName}`);
+    }
+
+    // Construct the EntryGroup for the EntryGroup request.
+    const entryGroup = {
+      displayName: 'My awesome Entry Group',
+      description: 'This Entry Group represents an external system',
+    };
+
+    // Construct the EntryGroup request to be sent by the client.
+    const entryGroupRequest = {
+      parent: datacatalog.locationPath(projectId, location),
+      entryGroupId: entryGroupId,
+      entryGroup: entryGroup,
+    };
+
+    // Use the client to send the API request.
+    const [createdEntryGroup] = await datacatalog.createEntryGroup(
+      entryGroupRequest
+    );
+    console.log(`Created entry group: ${createdEntryGroup.name}`);
+
+    // Construct the Entry for the Entry request.
+    const entry = {
+      userSpecifiedSystem: 'onprem_data_system',
+      userSpecifiedType: 'onprem_data_asset',
+      displayName: 'My awesome data asset',
+      description: 'This data asset is managed by an external system.',
+      linkedResource: '//my-onprem-server.com/dataAssets/my-awesome-data-asset',
+      schema: {
+        columns: [
+          {
+            column: 'first_column',
+            description: 'This columns consists of ....',
+            mode: 'NULLABLE',
+            type: 'STRING',
+          },
+          {
+            column: 'second_column',
+            description: 'This columns consists of ....',
+            mode: 'NULLABLE',
+            type: 'DOUBLE',
+          },
+        ],
+      },
+    };
+
+    // Construct the Entry request to be sent by the client.
+    const entryRequest = {
+      parent: datacatalog.entryGroupPath(projectId, location, entryGroupId),
+      entryId: entryId,
+      entry: entry,
+    };
+
+    // Use the client to send the API request.
+    const [createdEntry] = await datacatalog.createEntry(entryRequest);
+    console.log(`Created entry: ${createdEntry.name}`);
+
+    // Create a Tag Template.
+    // For more field types, including ENUM, please refer to
+    // https://cloud.google.com/data-catalog/docs/quickstarts/quickstart-search-tag#data-catalog-quickstart-nodejs.
+    const fieldSource = {
+      displayName: 'Source of data asset',
+      type: {
+        primitiveType: 'STRING',
+      },
+    };
+
+    const tagTemplate = {
+      displayName: 'Demo Tag Template',
+      fields: {
+        source: fieldSource,
+      },
+    };
+
+    tagTemplateRequest = {
+      parent: datacatalog.locationPath(projectId, location),
+      tagTemplateId: tagTemplateId,
+      tagTemplate: tagTemplate,
+    };
+
+    // Use the client to send the API request.
+    const [createdTagTemplate] = await datacatalog.createTagTemplate(
+      tagTemplateRequest
+    );
+    console.log(`Created template: ${createdTagTemplate.name}`);
+
+    // Attach a Tag to the custom Entry.
+    const tag = {
+      template: createdTagTemplate.name,
+      fields: {
+        source: {
+          stringValue: 'On-premises system name',
+        },
+      },
+    };
+
+    const tagRequest = {
+      parent: createdEntry.name,
+      tag: tag,
+    };
+
+    // Use the client to send the API request.
+    const [createdTag] = await datacatalog.createTag(tagRequest);
+    console.log(`Created tag: ${createdTag.name}`);
+  }
+  createCustomEntry();
+  // [END data_catalog_create_custom_entry]
+}
+main(...process.argv.slice(2));