前言

以前學(xué)android的時(shí)候就曾經(jīng)使用過APT在編譯時(shí)解析注解，然后結(jié)合Javapoet生成代碼，在flutter端其實(shí)也有這種類似的技術(shù)，這就是source_gen和code_builder，source_gen負(fù)責(zé)編譯時(shí)解析注解，code_builder負(fù)責(zé)生成代碼，本篇文章，就是在講解如何使用source_gen和code_builder編譯時(shí)生成代碼，code_builder如何使用，請(qǐng)看之前的文章, 本文 github鏈接

具體步驟

1. 新建flutter package，命名為annotations

這個(gè)包專門用來放注解，在這個(gè)包中新建一個(gè)注解類，如下所示

class ChannelHelp {
  final String channelName;
  const ChannelHelp(this.channelName); //注解類，構(gòu)造函數(shù)必須是Const的
}

2. 新建flutter package，命名為generate

這個(gè)包專門用來解析注解，生成代碼，先在pubspec.yaml中依賴上面創(chuàng)建的包annotations，然后再添加依賴source_gen和build_runner，代碼如下

name: generate
description: A new Flutter package.
version: 0.0.1
author:
homepage:

environment:
  sdk: ">=2.7.0 <3.0.0"
  flutter: ">=1.17.0 <2.0.0"

dependencies:
  flutter:
    sdk: flutter
  source_gen: ^0.9.6 #需要包含代碼自動(dòng)庫
  annotations:
    path: ../annotations

dev_dependencies:
  flutter_test:
    sdk: flutter
  build_runner: ^1.10.0

flutter:

3.在generate包中新建dart類ChannelHelpGenerator

這個(gè)類繼承自ChannelHelpGenerator，用于解析ChannelHelp注解，代碼如下：

import 'package:analyzer/dart/element/element.dart';
import 'package:annotations/channel_help.dart';
import 'package:build/src/builder/build_step.dart';
import 'package:source_gen/source_gen.dart';
import 'package:code_builder/code_builder.dart';
import 'package:dart_style/dart_style.dart';
import 'package:path/path.dart' as Path;
import 'package:build/build.dart';

class ChannelHelpGenerator extends GeneratorForAnnotation<ChannelHelp> {
  static final String channelName = "_MethodChannel";

  String className;

  @override
  generateForAnnotatedElement(
      Element element, ConstantReader annotation, BuildStep buildStep) {
    final emitter = DartEmitter();

    if (element is! ClassElement) {
      throw InvalidGenerationSourceError('ChannelHelper只能用在類上');
    }
    className = element.displayName;
    var channelName = annotation.peek("channelName").stringValue;
    ClassBuilder classBuilder;

    var channelHelper = Class((builder) {
      classBuilder = builder;

      classBuilder.constructors.add(Constructor((constructorBuild) {
        constructorBuild.name = "_internal";
      }));

      classBuilder.name = '${className}Imp';
      classBuilder.implements.add(refer(className));
      classBuilder.fields.add(Field((fieldBuild) {
        fieldBuild.name = "_MethodChannel";
        fieldBuild.type = refer("MethodChannel");
        fieldBuild.modifier = FieldModifier.final$;
        fieldBuild.assignment = Code('MethodChannel("$channelName")');
      }));

      classBuilder.fields.add(Field((fieldBuild) {
        fieldBuild.name = "_$className";
        fieldBuild.type = refer("$className");
        fieldBuild.static = true;
      }));

      classBuilder.methods.add(Method((methodBuild) {
        methodBuild.name = "getInstance";
        methodBuild.returns = refer('$className');
        methodBuild.static = true;
        methodBuild.body = _generatorSingleInstantBody();
      }));

      ClassElement classElement = element as ClassElement;
      List<MethodElement> methodElements = classElement.methods;
      if (methodElements != null && methodElements.length > 0) {
        methodElements.forEach((methodElement) {
          classBuilder.methods.add(Method((methodBuild) {
            methodBuild.name = methodElement.name;
            methodBuild.modifier = MethodModifier.async;
            methodBuild.returns =
                refer("${methodElement.returnType.getDisplayString()}");
            methodBuild.annotations.add(TypeReference((build) {
              // 給方法添加注解
              build.symbol = "override"; //注解類型是override
            }));
            var parameters = methodElement.parameters;
            methodBuild.body = _generatorBody(methodElement, parameters);
          }));
        });
      }
    });

    String channelHelperStr =
        DartFormatter().format('${channelHelper.accept(emitter)}');

    return """
        
        part of '${Path.basename(buildStep.inputId.path)}';
        
        $channelHelperStr
    """;
  }

  Code _generatorSingleInstantBody() {
    final blocks = <Code>[];
    blocks.add(Code("if(_$className == null) {"));
    blocks
        .add(Code("_$className = ${className}Imp._internal() as $className;"));
    blocks.add(Code("}"));
    blocks.add(Code("return _$className;"));
    return Block.of(blocks);
  }

  // ignore: missing_return
  Code _generatorBody(
      MethodElement methodElement, List<ParameterElement> parameters) {
    final blocks = <Code>[];
    if (parameters == null || parameters.length == 0) {
      blocks.add(Code(
          "dynamic _result = await $channelName.invokeMethod('${methodElement.name}');"));
    }
    blocks.add(_generatorResult(methodElement));
    return Block.of(blocks);
  }

  Code _generatorResult(MethodElement methodElement) {
    final blocks = <Code>[];
    return Block.of(blocks);
  }
}

4.在generate包中新建dart文件channel_help_builder

這個(gè)包用于加載ChannelHelpGenerator，指定生成的文件后綴，代碼如下：

import 'package:source_gen/source_gen.dart';
import 'package:build/build.dart';
import 'channel_help_generator.dart';

Builder nativeCallBuilder(BuilderOptions options) =>
    LibraryBuilder(ChannelHelpGenerator(), generatedExtension: '.nc.g.dart');

5.在generate包中新建文件build.yaml

代碼如下：

builders:
  channel_help_builder:
    target: ":annotations" #目標(biāo)庫
    import: 'package:generate/channel_help_builder.dart'  #build文件
    builder_factories: ['nativeCallBuilder']
    build_extensions: {'.dart': ['.nc.g.dart']}
    auto_apply: dependents #將此Builder應(yīng)用于包，直接依賴于公開構(gòu)建起的包，也可以是root_package
    build_to: source #輸出到注解的目標(biāo)類的代碼同目錄中，或者輸出轉(zhuǎn)到隱藏的構(gòu)建緩存，不會(huì)發(fā)布(cache)
    applies_builders: ["source_gen|combining_builder"] #指定是否可以延遲運(yùn)行構(gòu)建器

如何使用

1. 在主工程的pubspec.yaml添加依賴annotations，generate，build_runner

代碼如下：

name: generate_code
description: A new Flutter application.

publish_to: 'none' # Remove this line if you wish to publish to pub.dev

version: 1.0.0+1

environment:
  sdk: ">=2.7.0 <3.0.0"

dependencies:
  flutter:
    sdk: flutter
  annotations:
    path: annotations
  cupertino_icons: ^1.0.0

dev_dependencies:
  flutter_test:
    sdk: flutter
  generate:
    path: generate
  build_runner: ^1.4.0

flutter:

  uses-material-design: true

  

2. 新建ChannelTest文件，添加ChannelHelp注解

代碼如下

import 'package:annotations/channel_help.dart';
import 'package:flutter/services.dart';

part 'channel_test.nc.g.dart'; 


@ChannelHelp('test')
abstract class ChannelTest {

  void test();

}

3.執(zhí)行指令，如下

flutter packages pub run build_runner build

4.等待指令執(zhí)行完成即可看見編譯生成的代碼

代碼如下：

part of 'channel_test.dart';

class ChannelTestImp implements ChannelTest {
  ChannelTestImp._internal();

  final MethodChannel _MethodChannel = MethodChannel("test");

  static ChannelTest _ChannelTest;

  static ChannelTest getInstance() {
    if (_ChannelTest == null) {
      _ChannelTest = ChannelTestImp._internal() as ChannelTest;
    }
    return _ChannelTest;
  }

  @override
  void test() async {
    dynamic _result = await _MethodChannel.invokeMethod('test');
  }
}

QQ交流群

群號(hào)碼：770892444